[프로그래머스 C++] 의상

[프로그래머스 C++] 의상

 

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

프로그래머스

---

 

해결전략

 

map 안의 벡터 배열 

key와 여러개의 value

 

 

 

---

 

코드

 

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

int solution(vector<vector<string>> clothes) {

    map<string, vector<string>> myMap; //맵 생성, key: 의상의 종류, value: 의상의 이름들을 값

    for (int i=0; i<clothes.size(); i++){

        auto it = myMap.find(clothes[i][1]); // 의상의 종류에 해당하는 키를 찾음
        if(it != myMap.end()) // key가 있다면 (=해당 의상의 종류가 이미 존재하면)
        {
            it->second.push_back(clothes[i][0]); // 해당 key의 값 벡터에 의상 이름 추가
        }
        else // key가 없으면 새로운 벡터를 생성하고 맵에 추가
        {
            vector<string> inputTemp = { clothes[i][0] };
            myMap.insert({ clothes[i][1], inputTemp});	        
        }
    }

    vector<int> Items(myMap.size()); 
	int x = 0;

    // 각 카테고리별 아이템 개수를 저장하기 위한 벡터 생성
    for(auto iter = myMap.begin(); iter != myMap.end(); iter++)
    {
        Items[x++] = iter->second.size();
    }

    int answer = 1;
    for (int i = 0; i < Items.size(); i++)
    {
        answer *= (Items[i] + 1); //'아무것도 선택 안 함' 경우 +1
    }

    // 최소 한 개 이상 의상을 입으므로 전체에서 정말 아무것도 안입은 상태(-1) 제외
    answer -= 1;

    return answer;
}

 

 

---