[level 2] 의상
1. 문제
코니는 매일 다른 옷을 조합하여 입는것을 좋아합니다.
예를 들어 코니가 가진 옷이 아래와 같고, 오늘 코니가 동그란 안경, 긴 코트, 파란색 티셔츠를 입었다면 다음날은 청바지를 추가로 입거나 동그란 안경 대신 검정 선글라스를 착용하거나 해야합니다.
- 코니는 각 종류별로 최대 1가지 의상만 착용할 수 있습니다. 예를 들어 위 예시의 경우 동그란 안경과 검정 선글라스를 동시에 착용할 수는 없습니다.
- 착용한 의상의 일부가 겹치더라도, 다른 의상이 겹치지 않거나, 혹은 의상을 추가로 더 착용한 경우에는 서로 다른 방법으로 옷을 착용한 것으로 계산합니다.
- 코니는 하루에 최소 한 개의 의상은 입습니다.
코니가 가진 의상들이 담긴 2차원 배열 clothes가 주어질 때 서로 다른 옷의 조합의 수를 return 하도록 solution 함수를 작성해주세요.
제한사항
- clothes의 각 행은 [의상의 이름, 의상의 종류]로 이루어져 있습니다.
- 코니가 가진 의상의 수는 1개 이상 30개 이하입니다.
- 같은 이름을 가진 의상은 존재하지 않습니다.
- clothes의 모든 원소는 문자열로 이루어져 있습니다.
- 모든 문자열의 길이는 1 이상 20 이하인 자연수이고 알파벳 소문자 또는 '_' 로만 이루어져 있습니다.
2. 문제 풀이 방법
문제 풀이 접근법1
1) 주어진 2차원 배열 clothes를 딕셔너리로 변경한다.
- key = clothes[i][1] // 옷 종류
- value = clothes[i][0] // 옷 이름
2) 생성한 딕셔너리의 카테고리 별로 만들 수 있는 경우의 수를 계산, 반환한다.
def solution(clothes):
answer = 1
hash ={}
for i in range(len(clothes)):
if clothes[i][1] in hash.keys():
hash[clothes[i][1]] += [clothes[i][0]]
else:
hash[clothes[i][1]] = [clothes[i][0]]
for kind in hash.values():
answer *= (len(kind) + 1)
return answer -1
🎀알게 된 점
A의 종류가 N개, B의 종류가 M개일 때, 가능한 모든 경우의 수를 구하는 방식은
(N+1)(M+1) = NM + N + M + 1
과 같이 구할 수 있다.
- N과 M에 1을 더해주는 이유는 각각 A를 사용하지 않을 경우, B를 사용하지 않을 경우를 포함해주기 위함이다.
- NM: N과 M을 모두 사용하는 경우
- N: N만 사용하는 경우
- M: M만 사용하는 경우
- 1: 모두 사용하지 않는 경우
해당 경우에서는 하루에 최소 한 개 이상의 의상을 입는다고 했으므로, 아무 것도 입지 않는 경우의 수 1을 빼줘야 한다.
3. 다른 분들 문제 풀이
def solution(clothes):
clothes_type = {}
for c, t in clothes:
if t not in clothes_type:
clothes_type[t] = 2
else:
clothes_type[t] += 1
cnt = 1
for num in clothes_type.values():
cnt *= num
return cnt - 1
비슷한 풀이인데, 위에서는 카테고리에 속하는 옷들을 리스트로 만들었다면 여기는 그냥 길이를 처음부터 구하는 것과 같다.
처음에 2를 더하는 이유는(즉, 처음 들어온 카테고리일 때) 안 입는 경우를 더해주기 위함이다!
딕셔너리로 바꾸는 것 까지는 생각했는데 (아무래도 해시니까..)
그 다음 경우의 수를 구하는 방법은 깊게 고민해보지 못했음...^_^
그래도 이번에 새로 경우의 수 구하는 방법 알았으니
담번에 더 잘 풀어보는 걸로..