백준 BAEKJOON 31821번 학식 사주기 [PYTHON/파이썬]

 

 

 

 

 

백준 BAEKJOON 31821번 학식 사주기 [PYTHON/파이썬]

---

<문제 출처> (BRONZE Ⅳ)

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

 

 

 

 

 

 

 

<풀이>

입력 순서대로 학식 가격을 리스트에 저장한 뒤

새내기들이 먹고 싶어 하는 메뉴의 번호의 가격을

결괏값에 더해줬다.

 

 

 

 

 

 

<코드>

N = int(input())
menu = [int(input()) for _ in range(N)]
result = 0

M = int(input())
for _ in range(M):
    eatMenu = int(input())
    result += menu[eatMenu - 1]

print(result)