C++ / 프로그래머스 / 추억 점수

문제

https://school.programmers.co.kr/learn/courses/30/lessons/176963

프로그래머스

문제풀이

map을 이용하면 간단한 문제이다.

name과 yearning을 매칭해서 map에 담아준다.

photo를 탐색하면서 map[이름]의 점수가 있다면 더해서 answer에 합산하여 저장한다.

 

코드

#include <string>
#include <vector>
#include <unordered_map>
using namespace std;

vector<int> solution(vector<string> name, vector<int> yearning, vector<vector<string>> photo) {
    vector<int> answer(photo.size(), 0);
    int idx(0), idx2(0);
    unordered_map<string, int>score;
    for (idx = 0; idx < name.size(); ++idx) {
        score[name[idx]] = yearning[idx];
    }
    for (idx = 0; idx < photo.size(); ++idx) {
        for (idx2 = 0; idx2 < photo[idx].size(); ++idx2) {
            answer[idx] += score[photo[idx][idx2]];
        }
    }
    return answer;
}