「みんなのプロコン」: B - オークション

solution

開始時点での値段でsortして安いのから順に$K$個買う。買う順序は重要でないが、安い方から取ればよい。$O(N \log N)$。

implementation

#!/usr/bin/env python3
n, k = map(int, input().split())
a = sorted(map(int, input().split()))
acc = 0
for i in range(k):
    acc += a[i] + i
print(acc)