[알고리즘] 정다면체 (Python)

문제 설명

• 인프런 파이썬 알고리즘 문제풀이(코딩 테스트 대비)

파이썬 알고리즘 문제풀이 (코딩테스트 대비) - 인프런 | 강의

N, M이 주어지고, 정 N면체, 정 M면체로 이루어진 주사위가 있다고 가정하고, 두 개의 주사위를 던져 나올 수 있는 합 중 확률이 가장 큰 수를 출력합니다. 정답이 여러 개면 오름차순으로 출력합니다.

 

나의 풀이

• 처음에 permutations를 이용해 푸려고 했지만 잘못된 접근이었습니다.
• 2중 for문을 통해 해결했습니다.

 

Code

import sys
# sys.stdin=open("input.txt", "r")
n, m=map(int, input().split())

cnt = [0] * (n + m + 1)

for dice1 in range(1,n+1):
    for dice2 in range(1, m+1):
        cnt[dice1+dice2] += 1

most_cnt = max(cnt)
answer = []

for i in range(1,len(cnt)):
    if most_cnt == cnt[i]:
        print(i,end=" ")