[C++/PGS] Lv.2 : 의상/위장 (해시)

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

(원래 문제 제목이 위장이었는데 의상으로 바뀌었다! 구현 내용은 같음)

 

문제 설명(위장)

스파이들은 매일 다른 옷을 조합하여 입어 자신을 위장합니다.

예를 들어 스파이가 가진 옷이 아래와 같고 오늘 스파이가 동그란 안경, 긴 코트, 파란색 티셔츠를 입었다면 다음날은 청바지를 추가로 입거나 동그란 안경 대신 검정 선글라스를 착용하거나 해야 합니다.

 

종류이름

얼굴	동그란 안경, 검정 선글라스
상의	파란색 티셔츠
하의	청바지
겉옷	긴 코트

스파이가 가진 의상들이 담긴 2차원 배열 clothes가 주어질 때 서로 다른 옷의 조합의 수를 return 하도록 solution 함수를 작성해주세요.

---

 

만만하게 봤다가 틀린 문제!

경우의 수를 구하는 식을 잘 생각해보아야 한다. 

 

나는 처음에 일단 answer에다가 옷의 갯수를 더하고, 종류의 수가 2개 이상일 때 경우의 수를 곱하는 과정을 거쳤는데..

히든 테케에서 많이 틀렸다.

그래서 두 번째 방법으로, 각 종류마다 안 입는 경우까지 포함해서 곱해준 다음에

모든 옷을 입지 않는 1가지 경우를 빼주었더니 맞았다!

 

역시 수학은 싫어,,,

 

 

나의 풀이

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

int solution(vector<vector<string>> clothes) {
    int answer = 1;
    map<string, int> clist;
    
    for(int i=0; i<clothes.size(); ++i){
        clist[clothes[i][1]]++;
    }
    for(auto i : clist){
        answer *= (i.second+1); // 안입는경우, 1번입는경우, 2번입는경우, ...
    }
    answer --; // 모두 안입는 경우 제외
    
    return answer;
}