[Python] 2480번: 주사위 세개

주사위 세개 

 

시간 제한메모리 제한제출정답맞힌 사람정답 비율

1 초

128 MB

277171

129804

108899

46.719%

문제

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개의 눈이 빈칸을 사이에 두고 각각 주어진다.

출력

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

예제 입력 1 복사

3 3 6

예제 출력 1 복사

1300

예제 입력 2 복사

2 2 2

예제 출력 2 복사

12000

예제 입력 3 복사

6 2 5

예제 출력 3 복사

600

 

 

---

데이터 시각화하느라 오랜만에 백준..

 

첫번째 정답

#주사위 3개 입력
a,b,c = map(int, input().split())

#주사위 알고리즘 (if문)
if a == b == c: #같은 눈이 3개가 나오면 10,000원+(같은 눈)×1,000원의 상금을 받게 된다.
  print(10000+a*1000)
elif a == b: #같은 눈이 2개만 나오는 경우에는 1,000원+(같은 눈)×100원의 상금을 받게 된다.
  print(1000+a*100)
elif b == c: 
  print(1000+b*100)
elif a == c: 
  print(1000+a*100)
else: #모두 다른 눈이 나오는 경우에는 (그 중 가장 큰 눈)×100원의 상금을 받게 된다.
  print(max(a,b,c) * 100)

 

가장 큰 수를 탐색하는걸 max 함수를 사용했다.

 

결과는

 

 

 

더 좋은 코드가 있는지 GPT에게 물어봤다.

 

개선된 코드

# 주사위 3개 입력
dice = list(map(int, input().split()))

# 같은 눈의 개수와 값 확인
unique = set(dice)
if len(unique) == 1:  # 같은 눈이 3개
    print(10000 + dice[0] * 1000)
elif len(unique) == 2:  # 같은 눈이 2개
    for num in unique:
        if dice.count(num) == 2:
            print(1000 + num * 100)
            break
else:  # 모두 다른 눈
    print(max(dice) * 100)

 

 

개선 내용

1. 리스트와 집합을 활용:

• set(dice): 중복을 제거해 고유값의 개수를 계산.

• 같은 눈 3개 → 집합 크기 = 1.

• 같은 눈 2개 → 집합 크기 = 2.

• 모두 다른 눈 → 집합 크기 = 3.

 

2. 조건 간소화:

• 중복 조건인 a == b, b == c, a == c 대신, 리스트에서 특정 숫자의 개수를 계산(count())하여 더 간단히 처리.

 

3. 가독성 향상:

• unique와 len(unique)로 조건을 분명히 표현.

 

4. 주요 함수:

1. set():

• 리스트에서 중복을 제거하여 고유값의 집합을 생성.

• 예: set([3, 3, 6]) → {3, 6}.

2. list.count():

• 리스트에서 특정 값이 등장하는 횟수를 반환.

• 예: [3, 3, 6].count(3) → 2.