22-H0ngJu

[H0ngJu]

🔗 문제 링크

의상

✔️ 소요된 시간

시간 재는걸 까먹음 .. 아마 1.5H?

✨ 수도 코드

문제 요약
서로 다른 옷의 조합의 수 구하라! (단, 적어도 하나 이상의 의상을 선택해야 함)

문제를 보고 바로 떠올린건,
일단 입력받은 clothes에 대하여 dictionary화 하고,
각 종류별로 조합을 구하고, 해당 조합(선택된 종류)에서 각 의상 수를 곱하면 된다고 생각했습니다.

간단히 예시를 들자면, 아래와 같습니다.

위 로직을 코드로 구현하면 아래와 같습니다.
c_dict는 clothes를 딕셔너리로 바꾼 것입니다.

for r in range(len(c_dict)): # nCr에서 r은 dictionary 길이만큼 가능
    nCr = list(itertools.combinations(c_dict,r+1)) # 조합 찾기 r이 1이면, nCr의 반환값의 예는 (얼굴, 상의)
    for n in range(len(nCr)): # 찾은 조합에 대하여 경우의 수 구하기
        check = 1
        for k in range(len(nCr[n])): # 각 종류에 대하여 의상 수 곱하기
            check *= c_dict[nCr[n][k]] # r = 1, 이고 nCr의 첫번째 경우의 수가 (얼굴, 상의)이면, check의 최종값은 5*4임
        answer += check # 최종 답인 answer에 더하기

어째저째 풀긴했지만 굉~~~~장히 뺑뺑 돌아간걸 느낄 수 있습니다.
두번째 for문을 제외하더라도 시간복잡도가 O(n^2)이니까요 ,,,

아니나 다를까 제출하니까 테스트 케이스 1에 대해서 시간초과가 발생하더라고요

그래서 모든 경우의 수를 구해서 더하기 보다, 여집합의 개념을 사용하기로 했습니다.

즉,
(전체 경우의 수) - (어떤 종류의 의상도 선택하지 않는 경우의 수)
가 여집합을 사용한 식입니다.

위에서 든 예시로 설명하면 아래와 같습니다.

다시 코드로 변환하면 이렇게 됩니다!

for c in c_dict.values():
    answer *= (c+1)

answer -= 1

2달만에 알고 문제푸니까 새롭네요 ..ㅎㅎ

아래는 전체코드입니다.

import itertools

clothes = [["yellow_hat", "headgear"], 
           ["blue_sunglasses", "eyewear"], 
           ["green_turban", "headgear"],
           ["test1", "sample"],
           ["test2", "sample"],
           ]
c_dict = {}
answer = 1

for info in range(len(clothes)):
    if not clothes[info][1] in c_dict:
        c_dict[clothes[info][1]] = 1
    else:
        c_dict[clothes[info][1]] += 1

for c in c_dict.values():
    answer *= (c+1)

answer -= 1

print(answer)

📚 새롭게 알게된 내용

[tgyuuAn]

이 문제 알고리즘 처음 공부할때 되게 헤메던 문제였는데

지금 다시봐도 어떤 문제인지 기억이 나네요 하하.

문제 딱 봤을때 풀이가 바로 떠올라서 코드를 다시 치진 않았습니다!!

from collections import defaultdict

def solution(clothes):
    closet = defaultdict(list)
    answer = 1
    
    for cloth in clothes:
        cloth_detail, cloth_type = cloth[0], cloth[1]
    
        closet[cloth_type].append(cloth_detail)
        
    for cloth_type in closet:
        answer *=(len(closet[cloth_type])+1)
        
    return answer-1

고생했어유 홍쥬르상 웰컴백알고리드미~~~~~~~~~

[tgyuuAn]

로그까지 남겨주시는 ㅎㄷㄷㄷㄷ

[tgyuuAn]

최신기술 흔적 ㅎㄷㄷㄷㄷㄷㄷㄷㄷㄷ

[InSange]

저도 각 종류별로 맵에다가 집어 넣은 후 각각의 조합을 구해서 일일이 다 더하는 식으로 풀었더니 타임 초과같은 오류가 발생하더군요... 조합의 경우의 수로 곱했더니 저도 똑같은 방식으로 해결할 수 있었던 것 같습니다.

#include <string>
#include <vector>
#include <map>
#include <algorithm>
#include <iostream>

using namespace std;

int solution(vector<vector<string>> clothes) {
    map<string, vector<string>> m;
    vector<string> v;
    int answer = 1;
    
    for(auto cloth : clothes)
    {
        m[cloth[1]].push_back(cloth[0]);
        if(m[cloth[1]].size() == 1) v.push_back(cloth[1]);
    }
    
    for(int i = 0; i < v.size(); i++)
    {
        answer *= (m[v[i]].size()+1);
    }

    return answer - 1;
}

[seongwon030]

사용하지 않는 경우의 수도 더해줘서 전체경우의 수를 구한 다음에 아무것도 입지 않은 경우의 수를 뺀 풀이 잘 봤습니다! 문제를 읽다가 이해하는데 좀 걸렸어요. 오랜만에 조합문제를 보니 머리가 띵했습니다.. 프로그래머스 문제는 백준과 다르게 테스트케이스 통과율도 잘 보여주네요 ! 이젠 프로그래머스를 좀 풀어봐야겠슴니다ㅎㅎ