본문 바로가기
DC 2

[자료구조] 주사위 게임 파이썬 (product)

by YGSEO 2021. 4. 23.
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

댓글