Submission #1313764

#	Time	Username	Problem	Language	Result	Execution time	Memory
1313764		tahasin_hossain	Stove (JOI18_stove)	Pypy 3	100 / 100	199 ms	59036 KiB

stove

import sys
def solve():
    line1 = sys.stdin.readline().split()
    if not line1: return
    n, k = map(int, line1)
    
    t = []
    for _ in range(n):
        t.append(int(sys.stdin.readline()))
    if n == 0:
        print(0)
        return
    total_time = (t[-1] + 1) - t[0]
    gaps = []
    for i in range(n - 1):
        gap_size = t[i+1] - (t[i] + 1)
        gaps.append(gap_size)
    gaps.sort(reverse=True)
    for i in range(min(k - 1, len(gaps))):
        total_time -= gaps[i]
        
    print(total_time)

if __name__ == "__main__":
    solve()

Compilation message (stdout)

Compiling 'stove.py'...

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

=======

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