[프로그래머스 / 파이썬] Lv.0 주사위 게임 3

https://school.programmers.co.kr/learn/courses/30/lessons/181916

 

문제

 

문제 설명

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

네 주사위에서 나온 숫자가 모두 p로 같다면 1111 × p점을 얻습니다.
세 주사위에서 나온 숫자가 p로 같고 나머지 다른 주사위에서 나온 숫자가 q(p ≠ q)라면 (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 이하의 정수입니다.

 

입출력 예

a	b	c	d	result
2	2	2	2	2222
4	1	4	4	1681
6	3	3	6	27
2	5	2	6	30
6	4	2	5	2

 

 

소스 코드

# 1
def solution(a, b, c, d):
    answer = 0
    origin = [a,b,c,d]
    dice = list(set([a,b,c,d]))    
    
    # 모두 같은 경우
    if len(dice) == 1:    
        answer = dice[0] * 1111
        
    # 3개가 같거나 2개씩 같은 경우
    elif len(dice) == 2:    
        # 3개가 같은 경우
        if max([origin.count(num) for num in dice]) == 3:
            p = max(origin, key=origin.count)
            q = min(origin, key=origin.count)
            answer = (10 * p + q)**2
        # 2개씩 같은 경우
        else:
            answer = ((dice[0] + dice[1]) * abs(dice[0] - dice[1])) 
            
    # 2개가 같고 나머지는 다른 경우
    elif len(dice) == 3:
        for num in dice:
            if origin.count(num) == 2:
                dice.remove(num)
        answer = dice[0]*dice[1]
        
    # 모두 다른 경우
    else:
        answer = min(dice)                
    
    return answer

 

풀이 # 1

• a, b, c, d가 모두 존재하는 리스트 = origin / 중복을 제거한 리스트 = dice
• dice에 원소가 1개 존재하는 경우 a, b, c, d가 모두 같다는 의미
• dice에 원소가 2개 존재하는 경우는 3개가 같거나 2개가 같은 경우
  • dice의 원소가 origin에 3개 있으면 origin.count 기준으로 최댓값=p / 최솟값=q
  • 위의 경우가 아니라면 2개씩 같은 경우
• dice에 원소가 3개 존재하는 경우는 2개는 같고 나머지는 다른 경우
  • 같은 2개의 원소를 뺴고 나머지의 곱
• 위의 경우에 포함되지 않으면 a, b, c, d가 모두 다른 경우

 

# 2
def solution(a, b, c, d):
    answer = 0
    dice = [a,b,c,d]
    cnt = [dice.count(num) for num in dice]
    
    # 모두 같은 경우
    if max(cnt) == 4:
        answer = a * 1111
        
    # 3개가 같고 나머지 1개가 다른 경우
    elif max(cnt) == 3:
        answer = (10 * dice[cnt.index(3)] + dice[cnt.index(1)])**2
        
    # 2개가 같은 경우
    elif max(cnt) == 2:
        # 2개가 같고 나머지 2개는 다른 경우
        if min(cnt) == 1:
            answer = 1
            answer = eval('*'.join([str(dice[i]) for i, c in enumerate(cnt) if c == 1])) 
        # 2개씩 같은 경우
        else:
            answer = (max(dice) + min(dice)) * abs(max(dice) - min(dice))
    
    # 모두 다른 경우
    else:
        answer = min(dice)
    
    return answer

 

풀이 #2

• 주사위에서 나온 숫자 리스트 = dice / 각 숫자의 갯수 리스트 = cnt
• 갯수의 최댓값이 4인 경우는 모두 같다는 의미
• 갯수의 최댓값이 3인 경우는 3개가 같고 나머지 1개가 다르다는 의미
• 갯수의 최댓값이 2인 경우에는 2가지 상황 존재
  • 2개가 같고 나머지 2개가 다른 경우에는 cnt가 1인 값들의 곱
  • 2개씩 같은 경우에는 dice의 최댓값 최솟값으로 각 숫자를 구분
• 위의 경우에 포함되지 않으면 a, b, c, d가 모두 다른 경우

 

 

머리 싸매다가 첫 번째 풀이도 겨우 생각했는데 두 번째 풀이 너무 신박하다. 푸신 분 너무 대단해요..같은 숫자의 갯수를 어떻게 구분하는지 생각하기가 어려웠던 문제였다.🎲🎲