[python] 백준 4716 풍선

https://www.acmicpc.net/problem/4716

4716번: 풍선

 

문제 해결

• 거리를 비교해서 A아니면 B를 선택하는 문제로 보면 된다.
• A와 B와 거리가 같으면 풍선 남는 쪽으로 가면 되는데 풍선이 모자랄 일이 없으므로 그냥 계산하면 된다.

 

CODE

while True:
    n, a, b = map(int, input().split())
    if n==0 and a==0 and b==0:break
    X = [list(map(int, input().split())) for _ in range(n)]
    X.sort(key=lambda x: abs(x[1]-x[2]), reverse = True)
    answer = 0
    E = []
    for ballon, distanceA, distanceB in X:
        if distanceA > distanceB:
            if b>= ballon:
                b-= ballon
                answer += distanceB*ballon
            else:
                ballon -=b
                answer += distanceB*b
                answer += distanceA * ballon
                b= 0
        elif distanceA < distanceB:
            if a>= ballon:
                a -= ballon
                answer += distanceA*ballon
            else:
                ballon -= a
                answer += distanceA*a
                answer += distanceB*ballon
                a=0
        else:
            E.append([ballon, distanceA,distanceB])
    for i in range(len(E)):
        answer += E[i][0]*E[i][1]
    print(answer)