[프로그래머스/C++] Lv.1 추억 점수

문제

문제

---

코드

#include <iostream>
#include <unordered_map>
#include <vector>

using namespace std;

vector<int> solution(vector<string> name, vector<int> yearning, vector<vector<string>> photo) {
    vector<int> answer;
    unordered_map<string, int> score;

    for (int i = 0; i < name.size(); ++i)
    {
        score.emplace(name[i], yearning[i]);
    }

    for (const auto& p : photo)
    {
        int temp = 0;
        for (const auto& s : p)
        {
            if (auto search = score.find(s) != score.end())
            {
                temp += score[s];
            }
        }
        answer.emplace_back(temp);
    }
    return answer;
}

---

문제 해설

- unordered_map 을 이용해서 순서에 상관없이 key:name, value:yearning을 저장
- photo 안에 있는 vector<string>의 개수만큼 foreach문 사용
- 만약 unordered_map 에 담겨 있는 string 이 나오면 임시 변수 temp 에 yearning 더해준다.
- 그리고 배열이 끝나면 answer에 temp를 담는다.