본문 바로가기
알고리즘

[알고리즘] 위장

by keel_im 2021. 3. 11.
반응형

포인트

  • 해시를 사용하여 종류의 수를 계산해서 조합을 구하는 함수를 구현하는 것입니다. 
  • 먼저 종류의 개수를 세어서 저장을 합니다

🧶문서는 항상 수정될 수 있습니다. 비판은 환영합니다. 

c++/cpp

#include <bits/stdc++.h>

using namespace std;

int solution(vector<vector<string>> clothes)
{
    int answer = 1;
    unordered_map<string, int> data;
    for (auto clothe : clothes)
    {
        data[clothe[1]] += 1;
    }

    for (auto it = data.begin(); it != data.end(); it++)
    {
        answer *= it->second + 1;
    }
    answer-=1;
    return answer;
}

python


def solution(clothes):
    data = dict()

    for clothe in clothes:
        data[clothe[1]] = data.get(clothe[1], 0) + 1

    answer = 1
    for ele in data.values():
        answer *= (ele+1)
    print(answer)
    answer -= 1
    return answer

 

반응형

'알고리즘' 카테고리의 다른 글

[알고리즘] 인구 이동  (0) 2021.03.14
[알고리즘] 나무 재테크  (0) 2021.03.14
[알고리즘] 뱀  (0) 2021.03.11
[알고리즘] 특이한 자석 + 톱니바퀴  (0) 2021.03.10
[알고리즘] 가장 먼 노드  (0) 2021.03.09

댓글