[Python/파이썬] 백준 2480번 - 주사위 세 개

문제

1에서부터 6까지의 눈을 가진 3개의 주사위를 던져서 다음과 같은 규칙에 따라 상금을 받는 게임이 있다.

1. 같은 눈이 3개가 나오면 10,000원+(같은 눈)×1,000원의 상금을 받게 된다.
2. 같은 눈이 2개만 나오는 경우에는 1,000원+(같은 눈)×100원의 상금을 받게 된다.
3. 모두 다른 눈이 나오는 경우에는 (그 중 가장 큰 눈)×100원의 상금을 받게 된다.

예를 들어, 3개의 눈 3, 3, 6이 주어지면 상금은 1,000+3×100으로 계산되어 1,300원을 받게 된다. 또 3개의 눈이 2, 2, 2로 주어지면 10,000+2×1,000 으로 계산되어 12,000원을 받게 된다. 3개의 눈이 6, 2, 5로 주어지면 그중 가장 큰 값이 6이므로 6×100으로 계산되어 600원을 상금으로 받게 된다.

3개 주사위의 나온 눈이 주어질 때, 상금을 계산하는 프로그램을 작성 하시오.

입력

첫째 줄에 3개의 눈이 빈칸을 사이에 두고 각각 주어진다.

출력

첫째 줄에 게임의 상금을 출력 한다.

풀이

• 직관적 풀이
  • if문을 통해, 문제에 주어진 조건을 판단하여 출력해준다.
    
    
• combination 라이브러리 & list 길이 활용
  • 같은 눈이 1개, 2개, 3개가 나온다 → numbers 리스트의 길이가 1, 2, 3이다
    • 같은 눈이 1개, 3개이다 → 문제의 조건대로 출력 / set 자료형으로 받아 중복값 제거
    • 같은 눈이 2개이다 → itertools의 combination 라이브러리를 이용하여, 세 수 중 같은 값이 있는 두 수를 확인해 출력한다.
      • 유의할 점 - itertools의 라이브러리를 통해 출력한 자료형은 iterator이다 → 리스트와 다름 → next 메서드 활용하여 상금 출력

코드

• 직관적인 풀이의 코드

from itertools import combinations

a, b, c = map(int, input().split())
numbers = [a, b, c]

if a==b==c:
    print(10000 + a * 1000)
elif a == b:
    print(1000 + a * 100)
elif b == c:
    print(1000 + b * 100)
elif a == c:
    print(1000 + a * 100)
else:
    print(max(a, b, c)*100)

• combination 라이브러리 활용한 코드

from itertools import combinations

a, b, c = map(int, input().split())
numbers = [a, b, c]

if len(set(numbers)) == 1:
    print(10000 + a * 1000)
elif len(set(numbers)) == 3:
    print(max(numbers) * 100)
else:
    same_pair_number = next(x for x, y in combinations(numbers, 2) if x == y)
    print(1000 + same_pair_number * 100)