[python] 주사위 게임 3

문제 설명

 

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

• 네 주사위에서 나온 숫자가 모두 p로 같다면 1111 × p점을 얻습니다.
• 세 주사위에서 나온 숫자가 p로 같고 나머지 다른 주사위에서 나온 숫자가 q(p ≠ q)라면 (10 × p + q)² 점을 얻습니다.
• 주사위가 두 개씩 같은 값이 나오고, 나온 숫자를 각각 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	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

 

제출 내역

def solution(a, b, c, d):
    li = [a,b,c,d]
    li.sort()
    dice = set(li)
    
    if len(dice) == 1:
        return 1111*a
    
    elif len(dice) == 4:
        return li[0]
    
    elif len(dice) == 2:
        if li.count(li[0]) == 3:
            return (10*li[0]+li[3])**2
        elif li.count(li[0]) == 2:
            return (li[0]+li[2])*abs(li[0]-li[2])
        else:
            return (10*li[1]+li[0])**2
        
    else:
        for i in li:
            if li.count(i) == 2:
                li.remove(i)
                li.remove(i)
                return li[0]*li[1]

set: 집합 자료형 - 중복X, 순서X

>>> set1 = set([1, 1, 3, 2])

>>> set1

{1, 2, 3}

 

sort(): 리스트 요소를 순서대로 정렬

>>> a = [1, 3, 4, 2]

>>> a.sort()

>>> a

[1, 2, 3, 4]

 

count(x): 리스트 안에 x의 개수를 리턴함

>>> a = [1, 2, 3, 1]

>>> a.count(1)

2

 

remove(x): 리스트에서 첫 번째로 나오는 x를 제거함

>>> a = [1, 2, 3, 1, 2, 3]

>>> a.remove(3)

>>> a

[1, 2, 1, 2, 3]

 

 

프로그래머스 코딩 기초 트레이닝 Day8 조건문, 문자열

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