Submission #1351834

#	Time	Username	Problem	Language	Result	Execution time	Memory
1351834		gaeguldev	Stove (JOI18_stove)	Pypy 3	100 / 100	160 ms	168452 KiB

stove

import sys
input=lambda: sys.stdin.readline().rstrip()
def solve():
    n,k=map(int,input().split())
    t=[int(input()) for _ in range(n)]
    t.sort()
    diff=[]
    for i in range(n-1):
        diff.append(t[i+1]-t[i]-1)
    diff.sort()
    print(n+sum(diff[:n-k]))
solve()

Compilation message (stdout)

Compiling 'stove.py'...

=======
  adding: __main__.pyc (deflated 38%)

=======

• Subtask 1: 20 / 20
• Subtask 2: 30 / 30
• Subtask 3: 50 / 50