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()

Compiling 'stove.py'...

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

=======