[C++][Programmers][해시] 의상 (feat. hash map 반복문)

https://school.programmers.co.kr/learn/courses/30/lessons/42578?language=cpp 

프로그래머스

 

*풀이

의상조합개수를 찾는 문제로, 의상의 종류를 key로 의상의 종류개수를 value로 hash map으로 표현해서

의상종류개수의 조합을 계산하면된다(곱셈  연산)

 

*알아둘 것

[hash map 반복문 탐색방법]

hash map에 반복문을 적용할 때 iterator를 이용하거나 foreach 를 이용하는 방법이 있다.

 

1) iterator을 이용한 hash map 반복문

unordered_map<string, int>::iterator it;
for(it = map.begin(); it != map.end(); it++){
	cout << it->first << " " << it->second << "\n";
}

iterator 를 만들고 ->로 접근한다.

 

2) foreach를 이용한 hash_map 반복문

for (auto& i : map) {
    cout << i.first << " " << i.second << endl;
}

auto& 문으로 자동으로 데이터형을 지정하거나 pair(string, int> i 로 형을 지정해주고

. 으로 접근해서 사용한다.

 

*코드

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

int solution(vector<vector<string>> clothes) {
    int answer = 1;
    unordered_map<string, int> map;
    
    //1. 옷 종류를 key로 두고 옷 종류를 센다.
    for(auto& i: clothes){
        map[i[1]]++;
    }
    
    //2. 옷 종류 개수만큼 곱해서 조합 개수를 찾는다.
    unordered_map<string, int>::iterator it;
    for(it = map.begin(); it != map.end(); it++){
        //옷 종류 개수에 착용안하는 경우를 추가해서 +1한 것.
        answer *= (it->second +1);
    }
    
    //3. 전부 착용 안하는 경우 1가지를 제외한다.
    answer--;
    
    return answer;
}

iterator 로 반복문을 돌렸었는데 검색해보고 아래처럼 고쳤다.

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

int solution(vector<vector<string>> clothes) {
    int answer = 1;
    unordered_map<string, int> map;
    
    //1. 옷 종류를 key로 두고 옷 종류를 센다.
    for(auto& i: clothes){
        map[i[1]]++;
    }
    
    //2. 옷 종류 개수만큼 곱해서 조합 개수를 찾는다.
    for(pair<string, int> i: map){
         //옷 종류 개수에 착용안하는 경우를 추가해서 +1한 것.
        answer *= (i.second + 1);
    }
    
    //3. 전부 착용 안하는 경우 1가지를 제외한다.
    answer--;
    
    return answer;
}

pair<string, int> i 를 auto& i 로 변경가능함 !