Lv.3 ( 해시 ) 베스트앨범 - C++

https://school.programmers.co.kr/learn/courses/30/lessons/42579?language=cpp 

프로그래머스

#include <string>
#include <vector>
#include <unordered_map>
#include <algorithm>
#include <iostream>

using namespace std;

bool cmp(const pair<string, int>& a, const pair<string, int>& b)
{
    return a.first > b.first;
}

bool cmp2(const tuple <int, int>& a, const tuple <int, int>& b)
{
    return get<1>(a) > get<1>(b);
}

vector<int> solution(vector<string> genres, vector<int> plays) {

    vector<int> answer;

    unordered_map<string, int> firstmap;
    unordered_map<string, vector<tuple <int, int>>> secondmap;

    for (int i = 0; i < genres.size(); i++)
    {
        firstmap[genres[i]] += plays[i];
        secondmap[genres[i]].push_back(make_tuple(i, plays[i]));
    }

    vector<pair<string, int>> sortgenres(firstmap.begin(), firstmap.end());

    sort(sortgenres.begin(), sortgenres.end(), cmp);

    for (auto i : sortgenres)
    {
        sort(secondmap[i.first].begin(), secondmap[i.first].end(), cmp2);
        for (int j = 0; j < 2; j++)
        {
            answer.push_back(get<0>(secondmap[i.first][j]));
            if (secondmap.size() < 2)
            {
                break;
            }
        }
    }



    return answer;
}

TC는 맞는데 왜 제출하면 틀린지 솔직히 모르겠다. 

시간이 빠듯해서 코테 시험을 본 이후 다시 살펴볼 예정이다.