프로그래머스 Lv2 의상

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

프로그래머스

나는 이 문제를 조합 문제로 접근했다.

문제가 선글라스 하나만 입어도 옷을 입은 것으로 쳐줘 이상하지만

만약 파란색 티셔츠와 검정 선글라스가 있다면

 

파란색 티셔츠를 입는 경우의 수 1

검정 선글라스를 입는 경우의 수 1

둘다 입지 않는 경우의 수 1

 

얼굴, 상의, 하의, 겉옷에 위와 같은 식으로 경우의 수를 더해주면 된다.

모든 옷을 입지 않는 경우의 수는 존재하지 않으므로 -1을 해준다.

 

#include <string>
#include <vector>
#include <unordered_map>

using namespace std;

int solution(vector<vector<string>> clothes) {
    int answer = 1;
    unordered_map<string, int> pit;
    
    for(int i = 0; i < clothes.size(); ++i)
    {
        pit[clothes[i][1]]++;
    }
    
    for(auto iter : pit)
    {
        answer *= iter.second + 1;
    }
    return answer - 1;
}