백준 31821 학식 사주기 python / c++

문제 링크 : https://www.acmicpc.net/problem/25191

N = int(input())
menus = [int(input()) for _ in range(N)]

result = 0
M = int(input())

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

print(result)

 

#include <iostream>
#include <vector>

using namespace std;

int main() {
    int N, M, result = 0;
    cin >> N;
    vector<int> menus(N);
    

    for (int i = 0; i < N; ++i) {
        cin >> menus[i];
    }


    cin >> M;

    for (int i = 0; i < M; ++i) {
        int order;
        cin >> order;
        result += menus[order - 1];
    }

    cout << result << endl;

    return 0;
}

 

 

N개의 메뉴 개수, M명의 학생의 수를 입력받아, 학생이 희망하는 번호의 음식값을 누적하여 출력합니다