date posted: 2020-04-19
문제 설명은 아래 프로그래머스 링크에서 보세요.
import numpy as np
def solution(clothes):
hashmap = {}
for c in clothes:
if c[1] in hashmap:
hashmap[c[1]].append(c[0])
else:
hashmap[c[1]] = [c[0]]
count_per_group = [len(value) for value in hashmap.values()]
multiplied = np.prod(count_per_group) if len(count_per_group) > 1 else 0
return sum(count_per_group) + multiplied
import numpy as np
def solution(clothes):
hashmap = {}
for c in clothes:
if c[1] in hashmap:
hashmap[c[1]].append(c[0])
else:
hashmap[c[1]] = [c[0]]
count_per_group = [len(value) for value in hashmap.values()]
return np.prod([c+1 for c in count_per_group]) -1
import numpy as np
def solution(clothes):
hashmap = {}
for c in clothes:
if c[1] in hashmap:
hashmap[c[1]] = hashmap[c[1]] + 1
else:
hashmap[c[1]] = 1
count_per_group = list(hashmap.values())
return np.prod([c+1 for c in count_per_group]) -1
import numpy as np
from collections import Counter
def solution(clothes):
hashmap = Counter([cat for name,cat in clothes])
count_per_group = list(hashmap.values())
return np.prod([c+1 for c in count_per_group]) -1