[파이썬 알고리즘] 정다면체

정다면체

📄 문제

두 개의 정 N면체와 정 M면체의 두 개의 주사위를 던져서
나올 수 있는 눈의 합 중 가장 확률이 높은 숫자를 출력하는 프로그램을 작성하세요.
정답이 여러 개일 경우 오름차순으로 출력합니다.

▣ 입력설명
첫 번째 줄에는 자연수 N과 M이 주어집니다.
N과 M은 4, 6, 8, 12, 20 중의 하나입니다.

▣ 출력설명
첫 번째 줄에 답을 출력합니다.

▣ 입력예제
4 6

▣ 출력예제
5 6 7

🧞 풀이

풀이

# 입력: 첫째 줄 - n, m (4, 6, 8, 12, 20 중 하나)

# 출력: 답 출력

n, m = map(int, input().split())        # 정수 n, m 입력하여 할당

list = []                               # 빈 리스트 list 할당

for i in range(1, n+1):                 # 1부터 n까지 반복
    for j in range(1, m+1):             # 1부터 m까지 이중 반복
        list.append(i+j)                # i+j를 리스트에 추가

cnt_list = []                           # 갯수 담는 빈 리스트 cnt_list 할당

for i in range(2, n+m):                 # 2부터 n+m-1까지 반복
    cnt_list.append(list.count(i))      # cnt_list에 각 i에 해당하는 list의 요소 갯수 카운트하여 추가

max_value = max(cnt_list)               # cnt_list의 최댓값 max_value에 할당

a_list = []                             # 정답을 담을 빈 리스트 a_list 할당

for i in range(len(cnt_list)):          # cnt_list의 길이만큼 반복
    if cnt_list[i] == max_value:        # cnt_list의 각 요소가 최댓값과 일치한다면
        a_list.append(i+2)              # i+2를 a_list에 추가 (i+2를 하는 이유는 cnt_list의 인덱스 0이 합 2의 갯수를 의미하기 때문)


print(*a_list, sep = ' ')               # a_list 언패킹하여, 공백 하나 간격으로 출력

 

다른 사람 풀이

n, m = map(int, input().split())

# 넉넉하게 잡아서
# 리스트의 크기가 n+m+3 만큼 생김
cnt = [0]*(n+m+3)

max = -2147000000

# 1부터 n까지 돈다.
# 1부터 m까지 돈다.
for i in range(1, n+1):
    for j in range(1, m+1):
        cnt[i+j] += 1

# n + m까지 돈다.
for i in range(n+m+1):
    if cnt[i] > max:
        max = cnt[i]

# n + m까지 돈다.
# 최대값과 같은 칼럼들 찾아내기
# i 출력
# end = ' '로 한 칸 띄워서 옆에 출력
for i in range(n+m+1):
    if cnt[i] == max:
        print(i, end = ' ')