카테고리 없음

프로그래머스 - 해시/위장/C++

bell22 2021. 7. 26. 14:05

띵곡

 

문제 설명

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

예를 들어 스파이가 가진 옷이 아래와 같고

오늘 스파이가 동그란 안경, 긴 코트, 파란색 티셔츠를 입었다면

다음날은 청바지를 추가로 입거나 동그란 안경 대신 검정 선글라스를 착용하거나 해야 합니다.

 

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

스파이가 가진 의상들이 담긴 2차원 배열 clothes가 주어질 때

서로 다른 옷의 조합의 수를 return 하도록 solution 함수를 작성해주세요.

 

제한사항

  • clothes의 각 행은 [의상의 이름, 의상의 종류]로 이루어져 있습니다.
  • 스파이가 가진 의상의 수는 1개 이상 30개 이하입니다.
  • 같은 이름을 가진 의상은 존재하지 않습니다.
  • clothes의 모든 원소는 문자열로 이루어져 있습니다.
  • 모든 문자열의 길이는 1 이상 20 이하인 자연수이고 알파벳 소문자 또는 '_' 로만 이루어져 있습니다.
  • 스파이는 하루에 최소 한 개의 의상은 입습니다.

입출력 예

 

clothes return
[["yellowhat", "headgear"], ["bluesunglasses", "eyewear"], ["green_turban", "headgear"]] 5
[["crowmask", "face"], ["bluesunglasses", "face"], ["smoky_makeup", "face"]] 3

 

입출력 예 설명

 

예제 #1
headgear에 해당하는 의상이 yellow_hat, green_turban이고

eyewear에 해당하는 의상이 blue_sunglasses이므로 아래와 같이 5개의 조합이 가능합니다.

 

1. yellow_hat

2. blue_sunglasses

3. green_turban

4. yellow_hat + blue_sunglasses

5. green_turban + blue_sunglasses

 

예제 #2
face에 해당하는 의상이 crow_mask, blue_sunglasses, smoky_makeup이므로

아래와 같이 3개의 조합이 가능합니다.

 

1. crow_mask

2. blue_sunglasses

3. smoky_makeup

 

 


코드

 

#include <string>
#include <vector>
#include <unordered_map>
#include <iostream>

using namespace std;

int solution(vector<vector<string>> clothes)
{
    int answer = 1;
    unordered_map<string, int> clotheMap;
    for( auto elem : clothes )
    {
        cout << "elem[1]: " << elem[1] << endl;
        clotheMap[elem[1]] += 1;
    }
    for( auto elem: clotheMap)
    {
        answer *= (elem.second+1);
        cout << "answer: " << answer << endl;
    }
    return answer-1;
}

int main(void)
{
    vector<vector<string>> vec;

    vector<string> vec1;
    vec1.push_back("yellowhat");
    vec1.push_back("headgear");

    vector<string> vec2;
    vec2.push_back("bluesunglasses");
    vec2.push_back("eyewear");

    vector<string> vec3;
    vec3.push_back("green_turban");
    vec3.push_back("headgear");

    vec.push_back(vec1);
    vec.push_back(vec2);
    vec.push_back(vec3);

    cout << solution(vec) << endl;

    return 0;
}

 

* warning 확인차 main 함수 추가한거라 solution 함수만 보면 됩니다 *

 

요거도 해시 맵을 사용해서 풀면 되는데...!

나는 글을 잘 못 읽어서 문제 이해하는데 오래걸렸다 ㅜㅜ

책을 읽어야 한다 이래서

 

이 문제의 포인트는 같은 의상 종류에 대한 갯수이다

예를 들어 첫 번째 예제의 경우 headgear 의상이 두 가지(yellowhat과 green_turban)이니까

해쉬 맵에 headgear의 key를 가지는 value 값을 증가시켜주면 된다

 

그러면 요소의 개수를 통해 의상을 고르는 경우의 수를 구할 수 있다

첫 번째 예제의 경우 headgear 두 개, eyewear 한 개 이므로

 

headgear 의 경우의 수 → yellowhat 입을 경우 / green_turban 입을 경우/ 둘 다 안입을 경우

eyewear의 경우의 수 → bluesunglasses 입을 경우 / 안 입을 경우

 

∴ (요소의 개수 +1)*(요소의 개수 +1) 이렇게 되고!

 

또한 하루에 한 개의 의상은 입어야하니까

전체 다 안 입을 경우의 수인 1을 빼기 해주면 공식은 완성된당 

(return answer-1 해주는 이유)