본문 바로가기

백앤드 개발/코딩테스트

[파이썬 문제풀이] 정다면체

1. 문제

# 두 개의 정N면체와 정M면체의 주사위를 던져 나올 수 있는 눈의 합 중
# 가장 확률 높은 숫자를 출력
# 정답이 여러개일 경우 오름차순 출력

 

2. 풀이

import sys
# sys.stdin=open("input.txt", "r")

n, m = map(int,input().split()) # N, M 이 주어짐

# 나올 수 있는 합을 전부 나열
hap_ls = [0] * (n+m+1)  # 합 리스트 선언

# [0,1,2,3,,,n+m]

for i in range(1,n+1):
    for j in range(1, m+1):
        hap_ls[i+j]+=1


max_num = max(hap_ls) # 최대값 선언

# hap_ls 에서 눈의 합에 해당하는 값 출력
for k in range(0, n+m+1):
    if hap_ls[k] == max_num:
        print(k, end=' ')

 

 

참조: https://www.inflearn.com/course/%ED%8C%8C%EC%9D%B4%EC%8D%AC-%EC%95%8C%EA%B3%A0%EB%A6%AC%EC%A6%98-%EB%AC%B8%EC%A0%9C%ED%92%80%EC%9D%B4-%EC%BD%94%EB%94%A9%ED%85%8C%EC%8A%A4%ED%8A%B8