문제
지민이는 주사위 던지기 게임을 좋아하여 어느 날 옆에 있는 동호를 설득하여 주사위 던지기 게임을 하자고 하였다.
총 3개의 주사위가 있다.
그리고 이 주사위는 각각 S1(2 ≤ S1 ≤ 20), S2(2 ≤ S2 ≤ 20), S3(2 ≤ S3 ≤ 40)개의 면이 있다.
(실제로는 주사위가 6개의 면이 있는 것이 정상이지만 특별한 주사위라 생각하자.)
문제는 세 개의 주사위를 동시에 던졌을 때 가장 높은 빈도로 나오는 세 주사위의 합을 구하는 것이다.
예를 들어, S1 = 3, S2 = 2, S3 = 3으로 주어질 때, 주사위1은 S1(3)개의 면이 있으므로 1, 2, 3의 눈을 가지고,
주사위2는 S2(2)개의 면이 있으므로 1, 2의 눈을 가지며,
주사위3은 S3(3)개의 면이 있으므로 1, 2, 3의 눈을 가진다.
이 때, 이 3개의 주사위를 던져서 눈의 합을 구하면,
(1, 1, 1) = 3, (1, 1, 2) = 4, (1, 1, 3) = 5, ... , (3, 2, 1) = 6, (3, 2, 2) = 7, (3, 2, 3) = 8과 같은 합들을 얻을 수 있다.
이 때, 가장 많이 발생하는 합을 구하는 것이다.
https://www.acmicpc.net/problem/1233
입력
입력 파일의 첫째 줄에 정수 S1, S2, S3가 주어진다.
출력
출력 파일의 첫째 줄에 가장 높은 빈도로 나오는 세 주사위 합을 구하는 것이다. 단 답이 여러개라면 가장 합이 작은 것을 출력한다.
예제
# input
3 2 3
# output
5
문제 풀이
# 1 딕셔너리 활용
S1, S2, S3 = map(int, input().split())
ans = dict()
for i in range(1, S1+1):
for j in range(1, S2+1):
for k in range(1, S3+1):
if i+j+k not in ans:
ans[i+j+k] = 1
else:
ans[i+j+k] += 1
ans = sorted(ans.items(), key=lambda x:x[0])
print(max(ans, key=lambda x:x[1])[0])
*답이 여러개인 경우 가장 합이 작은 것을 출력하기 위해 sort하였습니다.
# 2 리스트 활용
S1, S2, S3 = map(int, input().split())
ans = [0] * (S1 + S2 + S3 + 1)
for i in range(1, S1+1):
for j in range(1, S2+1):
for k in range(1, S3+1):
ans[i+j+k] += 1
print(ans.index(max(ans)))
파이썬을 독학하시는 분들에게 도움이 되길 바라며,
혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.