ABC121 C - Energy Drink Collector を解いた

3768 ワード
AtCoder Python AtCoderBeginnerContest Python テキストリンク




貪欲的な感じで Ai,Bi のリストをソートし、
最安値のモノから買い占めればよろしいのでは？以下で通った。

EnergyDrinkCollector.py
N,M = map(int,input().split())
lis = []
for _ in range(N):
    a,b = map(int,input().split())
    lis.append([a,b])
lis = sorted(lis,key=lambda t:t[0])
score = 0
for a,b in lis:
    if M == b:
        score += a*b
        break
    elif M-b > 0:
        M -= b
        score += a*b
    else:
        score += a*M
        break
print(score)