只能通过4/6的测试样例,网上答案貌似也都这样。
import sys
k = int(sys.stdin.readline().strip())
arr = []
arr = list(map(int, sys.stdin.readline().strip().split()))
arr.sort(reverse=True)
sum = 0
p = arr[0]
sum+=p
for i in range(1, len(arr)):
if p > arr[i]:
p = arr[i]
else:
p = p - 1
if p<=0:
break
sum += p
print(sum)