[프로그래머스 / C++] 위장

 

문제 링크: programmers.co.kr/learn/courses/30/lessons/42578

코딩테스트 연습 - 위장

 

풀이

 

상의의 종류가 A, B, C 3가지일 경우

1. A를 입는다.

2. B를 입는다.

3. C를 입는다.

4. 상의를 입지 않는다. 

위와 같이 4가지 경우가 있을 수 있다. 

 

따라서 옷의 종류에 따라 (해당 종류의 옷 개수 + 1) 가지 경우의 수가 생긴다. 하지만 하루에 한 가지 이상의 옷을 꼭 착용해야 하므로 "모든 옷을 입지 않는다"라는 경우의 수를 빼줘야 한다. (answer - 1을 하는 이유)

 

[ 의상의 종류: 해당 종류 의상의 개수 ]는 unordered_map을 이용해 저장했다. 

 

 

 

 

소스코드

 

#include <string>
#include <vector>
#include <unordered_map>
 
using namespace std;
 
int solution(vector<vector<string>> clothes) {
    int answer = 1;
    unordered_map<string, int> m;
 
    for(auto it: clothes)
        m[it[1]]++;
    
    for(pair<string, int> atom: m)
        answer *= (atom.second + 1);
    
    return answer - 1;
}

 

테스트 결과

 

 

 

공부한 것을 정리한 내용입니다. 수정할 부분이 있다면 알려주시면 감사하겠습니다 :)