[C++] 추억 점수

 

추억 점수

그리워하는 사람의 이름을 담은 문자열 배열 name, 각 사람별 그리움 점수를 담은 정수 배열 yearning, 각 사진에 찍힌 인물의 이름을 담은 이차원 문자열 배열 photo가 매개변수로 주어질 때, 사진들의 추억 점수를 photo에 주어진 순서대로 배열에 담아 return하는 solution 함수를 완성해주세요.

 

 

 

입출력 예

["may", "kein", "kain", "radi"]	[5, 10, 1, 3]	[["may", "kein", "kain", "radi"],["may", "kein", "brin", "deny"], ["kon", "kain", "may", "coni"]]	[19, 15, 6]
["kali", "mari", "don"]	[11, 1, 55]	[["kali", "mari", "don"], ["pony", "tom", "teddy"], ["con", "mona", "don"]]	[67, 0, 55]
["may", "kein", "kain", "radi"]	[5, 10, 1, 3]	[["may"],["kein", "deny", "may"], ["kon", "coni"]]	[5, 15, 0]

 

첫 번째 사진 속 "may", "kein", "kain", "radi"의 그리움 점수를 합치면 19(5 + 10 + 1 + 3)점 입니다. 두 번째 사진 속 그리워하는 사람들인 "may"와 "kein"의 그리움 점수를 합치면 15(5 + 10)점입니다. 세 번째 사진의 경우 "kain"과 "may"만 그리워하므로 둘의 그리움 점수를 합한 6(1 + 5)점이 사진의 추억 점수입니다. 따라서 [19, 15, 6]을 반환합니다.

 

 

이름에 해당하는 추억 점수를 map 자료구조에 담아 빠르게 가져올 수 있도록 하여 문제를 풀이합니다. 이후 주어진 사진배열을 반복하면서 해당 이름이 map에 있을 경우 해당하는 점수를 더하고 이를 answer 배열에 삽입합니다.

 

 

 

풀이

#include <string>
#include <vector>
#include <map>

using namespace std;

vector<int> solution(vector<string> name, vector<int> yearning, vector<vector<string>> photo) {
    vector<int> answer;
    map<string, int> m;
    for(int i =0; i<name.size();i++){
        m.insert({name[i], yearning[i]});
    }
    for(int i =0; i < photo.size(); i++){
        int sum = 0;
        for(int j=0; j < photo[i].size(); j++){
            if(m.find(photo[i][j]) != m.end()){
                sum += m[photo[i][j]];
            }
        }
        answer.push_back(sum);
    }
    return answer;
}