DC 2
[자료구조] 주사위 게임 파이썬 (product)
YGSEO
2021. 4. 23. 01:17
728x90
from itertools import product
def solution(monster, S1, S2, S3):
l1 = list(range(1, S1+1))
l2 = list(range(1, S2+1))
l3 = list(range(1, S3+1))
prod = list(map(sum,product(l1, l2, l3)))
print(prod)
cnt = 0
for p in prod:
if p + 1 in monster:
cnt += 1
return int((len(prod)-cnt)/len(prod)*1000)
출처: velog.io/@rapsby/%ED%94%84%EB%A1%9C%EA%B7%B8%EB%9E%98%EB%A8%B8%EC%8A%A4-%EC%A3%BC%EC%82%AC%EC%9C%84-%EA%B2%8C%EC%9E%84-python-edmg8xrg
product만 생각했어도 쉽게 풀리는 문제
728x90