[프로그래머스] Lv0_주사위 게임 3

박선영·2023년 10월 7일
0
post-thumbnail

Lv0_주사위 게임 3

📄Description

1부터 6까지 숫자가 적힌 주사위가 네 개 있습니다. 네 주사위를 굴렸을 때 나온 숫자에 따라 다음과 같은 점수를 얻습니다.

네 주사위에서 나온 숫자가 모두 p로 같다면 1111 × p점을 얻습니다.
세 주사위에서 나온 숫자가 p로 같고 나머지 다른 주사위에서 나온 숫자가 q(p ≠ q)라면 (10×p+q)2(10 × p + q)^2점을 얻습니다.
주사위가 두 개씩 같은 값이 나오고, 나온 숫자를 각각 p, q(p ≠ q)라고 한다면 (p + q) × |p - q|점을 얻습니다.
어느 두 주사위에서 나온 숫자가 p로 같고 나머지 두 주사위에서 나온 숫자가 각각 p와 다른 q, r(q ≠ r)이라면 q × r점을 얻습니다.
네 주사위에 적힌 숫자가 모두 다르다면 나온 숫자 중 가장 작은 숫자 만큼의 점수를 얻습니다.
네 주사위를 굴렸을 때 나온 숫자가 정수 매개변수 a, b, c, d로 주어질 때, 얻는 점수를 return 하는 solution 함수를 작성해 주세요.

제한 조건

  • a, b, c, d는 1 이상 6 이하의 정수입니다.

입출력 예시

abcdresult
22222222
41441681
633627
252630
64252

입출력 예 설명

  1. 예제 1번에서 네 주사위 숫자가 모두 2로 같으므로 1111 × 2 = 2222점을 얻습니다. 따라서 2222를 return 합니다.
  2. 예제 2번에서 세 주사위에서 나온 숫자가 4로 같고 나머지 다른 주사위에서 나온 숫자가 1이므로 (10×4+1)2(10 × 4 + 1)^2 = 412 = 1681점을 얻습니다. 따라서 1681을 return 합니다.
  3. 예제 3번에서 a, d는 6으로, b, c는 3으로 각각 같으므로 (6 + 3) × |6 - 3| = 9 × 3 = 27점을 얻습니다. 따라서 27을 return 합니다.
  4. 예제 4번에서 두 주사위에서 2가 나오고 나머지 다른 두 주사위에서 각각 5, 6이 나왔으므로 5 × 6 = 30점을 얻습니다. 따라서 30을 return 합니다.
  5. 예제 5번에서 네 주사위 숫자가 모두 다르고 나온 숫자 중 가장 작은 숫자가 2이므로 2점을 얻습니다. 따라서 2를 return 합니다.

🤔생각 정리

  1. 몇 개의 숫자가 중복되는지 확인해야겠네
    collections 모듈의 Counter 클래스를 활용해 볼 수 있겠네
  2. 조건을 잘 정의해주면 되겠네
    (1) 모두 같은 숫자 p -> 1111 x p
    (2) 세 개의 같은 숫자 p & q -> (10×p+q)2(10 × p + q)^2
    (3) 두 개의 같은 숫자 p & q -> (p + q) × |p - q|
    (4) 두 개의 같은 숫자 p와 q & r -> q × r
    (5) 모두 다른 숫자 p, q, r, s -> min(p, q, r, s)
  3. 인덱싱해서 조건에 맞는 값을 return 하도록 구성해봐야지

💡Pseudo Code💡

1. cnt = Counter([a,b,c,d])
2. if 모두 같은 숫자 -> return 1111 * p
3. elif 세 개의 같은 숫자 -> return (10 x p + q)**2
4. elif 두 개의 같은 숫자 
5.		if 두 개의 같은 숫자 -> return (p + q) × |p - q|
6. 		else -> return q × r
7. else -> return min(a,b,c,d)

🖥️코드화

조건에 맞는 값을 인덱싱을 통해 return 할 수 있도록 구성하였다.
그래서 인덱싱 에러를 피하기 위해 nums 리스트에 요소를 추가적으로 넣어주었다.

from collections import Counter

def solution(a, b, c, d):
    cnt = Counter([a,b,c,d])
    cnt = sorted([(p, n) for p, n in list(cnt.items())], key=lambda x: -x[1])
    nums = [int(i[0]) for i in cnt] + [a] # 인덱싱을 위해 추가
    p, n = nums[0], cnt[0][1]
    if n == 2:
        return (p + nums[1]) * abs(p - nums[1]) if len(nums)==3 else nums[1] * nums[2]
#     print([min(nums), 0, (10*p+nums[1])**2, 1111 * p])
    return [min(nums), 0, (10*p+nums[1])**2, 1111 * p][n-1]   
profile
데이터를 만지는 사람

0개의 댓글