본문 바로가기

코딩테스트 with PYTHON

[python] 코딩테스트 대비 - 정다면체

 

 

문제)

 

답안 코드)

1
2
3
4
5
6
7
8
9
10
11
12
13
import sys
#sys.stdin=open("input.txt","rt")
N,M=map(int,input().split())
 
cnt=[0]*(N+M+2)#덧셈 결과 카운팅할 리스트 cnt[2]에는 두 눈의 합이 2인 경우 카운트가 삽임됨.
 
for i in range(1,N+1):#N면체 주사위
    for j in range(1,M+1):#M면체 주사위
        cnt[i+j]+=1#두 눈의 합 카운트 증가
 
for i in range(2,len(cnt)):
    if cnt[i]==max(cnt):#최대 값 이라면 출력
        print(i,end=' ')
cs