[백준][Python3] #25304. 영수증

---

코드 입력

class var:
    def __init__(self, price, cnt):
        self.price = price
        self.cnt = cnt

total = int(input())
total_cnt = int(input())


def val_price():
    
    for i in range(1, total_cnt+1):
        p, c = map(int, input().split())
        globals()[f'stuff_{i}'] = var(p, c)

    val_total = 0

    for g in globals():
        if 'stuff' in g:
            val_total += globals()[g].price * globals()[g].cnt

    if val_total != total:
        return print('No')
    else:
        return print('Yes')
    
val_price()

 

해당 문제는 N개만큼의 변수를 생성하는 로직에서 살짝 막혔다.

이전에 비슷한 느낌으로 로직을 구성해야 할 일이 있어서, globals() 로직을 써봤는데 잘 먹히지 않았다.

input을 2개씩 받아서 그런지, 나중에 변수값을 불러올 때 KeyError가 발생했다.

그래서 클래스를 만들어 각각 변수를 저장해주고 불러왔더니 성공했다.

아직 클래스에 대한 이론을 잘 알지 못 해서, 일 할 때나 이런 문제를 풀 때 이해가 조금 느린데

해당 부분을 잘 파악하면 나중에 로직을 구성할 때도 다양한 방법으로 구상할 수 있지 않을까 생각해본다

 

---

출처 : 백준 Online Judge

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

25304번: 영수증