PythonでJOI難易度5を埋める #14

A - ストーブ (Stove)

考えたこと
それぞれのを計算して，大きい順からマッチを使った方がストーブが燃料を節約できます．最初の訪問者が来たときは，必ずマッチを消費することに注意します．

n, k = map(int, input().split())
visitors = [int(input()) for _ in range(n)]

diff = [visitors[i+1] - visitors[i] for i in range(n-1)]
diff.sort(reverse = True)

k -= 1 #最初の一本
ans = 0
for i in range(n-1):
    if k > 0: #マッチを使う
        ans += 1
        k -= 1
    else:
        ans += diff[i]

ans += 1 #最後の一分
print(ans)