코드 입력
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')
해당 문제는 N개만큼의 변수를 생성하는 로직에서 살짝 막혔다.
이전에 비슷한 느낌으로 로직을 구성해야 할 일이 있어서, globals() 로직을 써봤는데 잘 먹히지 않았다.
input을 2개씩 받아서 그런지, 나중에 변수값을 불러올 때 KeyError가 발생했다.
그래서 클래스를 만들어 각각 변수를 저장해주고 불러왔더니 성공했다.
아직 클래스에 대한 이론을 잘 알지 못 해서, 일 할 때나 이런 문제를 풀 때 이해가 조금 느린데
해당 부분을 잘 파악하면 나중에 로직을 구성할 때도 다양한 방법으로 구상할 수 있지 않을까 생각해본다