프로그래머스 : 위장

https://programmers.co.kr/learn/courses/30/lessons/42578

코딩테스트 연습 - 위장

 

#include <string>
#include <vector>
#include <unordered_map>
using namespace std;

int solution(vector<vector<string>> clothes) {
    int answer = 1;
    unordered_map<string, vector<string>> umap;
    for(int i=0; i<clothes.size(); i++) {
        auto find = umap.find(clothes[i][1]);
        if(find != umap.end()) {
            find->second.push_back(clothes[i][0]);
        } else {
            vector<string> temp;
            temp.push_back(clothes[i][0]);
            umap.insert({clothes[i][1], temp});
        }
    }
    for(auto iter = umap.begin(); iter != umap.end(); iter++) {
        answer *= iter->second.size() + 1; // 해당 종류에서 고르지 않는 경우 + 1
    }
    return answer - 1; // 아무 것도 입지 않는 경우는 없으므로 -1
}