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

,

http://yahoo-procon2017-qual.contest.atcoder.jp/tasks/yahoo_procon2017_qual_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)