프로그래머스
[프로그래머스/c++] Lv2 의상
돈 많은 백수가 되고싶다
2024. 10. 8. 11:56
https://school.programmers.co.kr/learn/courses/30/lessons/42578
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr
map을 생각해서 풀 수 있으면 어려운 문제는 아닌 것 같다.
map에 clothes 종류 갯수를 넣고 (A종류 갯수+1) * (B종류 갯수+1) *... 다 곱한 후 마지막에 -1 해주면 된다.
-1은 다 입지 않은 경우를 고려한 것이다.
#include <string>
#include <vector>
#include <map>
using namespace std;
int solution(vector<vector<string>> clothes) {
int answer = 1;
map<string,int> m;
// 옷의 종류별로 개수를 셈
for(auto pair : clothes){
m[pair[1]]++;
}
// map의 모든 값에 대해 계산
for(auto it = m.begin(); it != m.end(); it++){
answer *= (it->second + 1); // it->second로 각 의류 종류의 개수에 접근
}
return answer - 1;
}