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
'DC 2' 카테고리의 다른 글
[자료구조] 배달 파이썬 (BFS, queue, Dijkstra, heap) (0) | 2021.04.23 |
---|---|
[자료구조] 문자열 압축 사본 파이썬 (👍) (0) | 2021.04.23 |
[자료구조] 사전순 부분문자열 파이썬 (0) | 2021.04.23 |
[자료구조] 짝지어 제거하기 파이썬 (stack, valid pair) (0) | 2021.04.22 |
[자료구조] 배상 비용 최소화 파이썬 (heap) (0) | 2021.04.22 |
댓글