[Python-CodingTest] 2-5. 정다면체

정다면체

문제 정리

입력

4 6 // n m

처리 과정

1. 0으로 초기화된 카운트 리스트 생성
2. 리스트의 길이는 n+m+3 (두 주사위의 눈의 합이 카운트 리스트의 인덱스가 되기 때문에 n+m을 넘지 않지만 +3은 여유롭게!)
3. cnt에서 max 찾기
4. cnt에서 값이 max인 인덱스들 출력

출력

5 6 7

풀이

import sys
sys.stdin = open("./input/in5.txt", "rt")

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

# 0으로 초기화된 cnt 리스트
# 예를들어 n과 m이 4와 6이라면, 두 주사위의 눈의 합은 10을 넘지 않음
cnt = [0]*(n+m+3) # +3은 여유,,
max=0

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

# cnt 리스트에서 최댓값 찾기
for i in range(n+m+1): # 0부터 n+m까지
    if (cnt[i]>max):
        max=cnt[i]

# cnt 리스트에서 값이 max인 인덱스 모두 출력
for i in range(n+m+1):
    if(cnt[i]==max):
        print(i, end=' ') # 옆으로 나란히 출력

정리

• 리스트의 생성과 초기화: lst = [0]*5
• n면체와 m면체 주사위의 눈의 합은 m+n을 넘지 않음

---

💛 개인 공부 기록용 블로그입니다. 👻

맨 위로 이동하기