728x90
난 해시 문제를 별로 안좋아하지만 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;
}
'프로그래머스 풀이 > Lv 3' 카테고리의 다른 글
프로그래머스 - 최고의 집합(C++) (0) | 2023.05.07 |
---|---|
프로그래머스 - 등굣길(C++) (0) | 2023.05.02 |
프로그래머스 - 징검다리 건너기(C++) (0) | 2023.05.01 |
프로그래머스 - 단속카메라(C++) (0) | 2023.04.29 |
프로그래머스 - 이중우선순위 큐(C++) (0) | 2023.03.22 |