프로그래머스 - 베스트앨범(C++)

 난 해시 문제를 별로 안좋아하지만 map의 사용법을 익힐 겸 문제를 추천받아 풀어봤다.

이런 생구현 문제는 확실히 먼저 구하는게 뭔지 확실히 해두고 차근차근 풀어가는게 빠르고 좋다. 처음 풀 때 중구난방하게 막 풀다가 결국 내가 내 코드를 못알아봐서 결국 다시 풀었다.

 

1. 장르별로 뭐가 가장 많이 재생됐고, 그 목록을 벡터로 만듦.

2. 가장 많이 재생된 장르부터 검사하며 고유번호, 재생수 형태로 매핑

3. 매핑한 컨테이너를 정렬. compare 함수 만들기

4. 반복

 

#include <string>
#include <vector>
#include <algorithm>
#include <map>
using namespace std;

bool comp(pair<string, int>& a, pair<string, int>& b) {
    return a.second > b.second;
}

bool compare(pair<int, int>& a, pair<int, int>& b) {
    if (a.second > b.second)
        return true;
    else if (a.second < b.second)
        return false;
    else if (a.second == b.second) {
        if (a.first < b.first)
            return true;
        else
            return false;
    }
}

vector<int> solution(vector<string> genres, vector<int> plays) {
    vector<int> answer;
    map<string, int> temp;
    int N = genres.size();


    for (int i = 0; i < N; i++) {
        if (temp.empty() || temp.find(genres[i]) == temp.end()) // 못찾거나 처음이면 인설트
            temp.insert(make_pair(genres[i], plays[i]));
        else
            temp[genres[i]] += plays[i];
    }

    vector<pair<string, int>> tmp(temp.begin(), temp.end());

    sort(tmp.begin(), tmp.end(), comp);
    vector<string> prior;
    for (int i = 0; i < tmp.size(); i++) {
        prior.push_back(tmp[i].first);
    }

    for (auto pri : prior) {
        map<int, int> index;

        for (int i = 0; i < N; i++)
            if (pri == genres[i]) index.insert(make_pair(i, plays[i]));

        vector<pair<int, int>> result(index.begin(), index.end());
        sort(result.begin(), result.end(), compare);
        if (result.size() == 1)
            answer.push_back(result[0].first);
        else {
            for (int j = 0; j < 2; j++) {
                answer.push_back(result[j].first);
            }
        }

    }
    return answer;
}