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

문제 설명
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 함수를 작성해 주세요.

def solution(a, b, c, d):
    answer = 0
    list = [a,b,c,d]
    num = [1,2,3,4,5,6]
    check = [0,0,0,0,0,0]
    check_2 = []
    check_1 = []

    for i in list:
        for j in range(len(num)):
            if i == j+1 :
                check[j] = check[j]+1
    print(check)
    
    if 4 in check :
        answer = (check.index(4)+1) * 1111
        return answer
    
    elif 3 in check :
        answer = pow(10*(check.index(3)+1)+(check.index(1)+1),2)
        return answer
    
    elif 2 in check and 1 not in check :
        for i in range(len(check)) :
            if check[i]==2:
                check_2.append(i+1)
        answer = (check_2[0]+check_2[1]) * abs(check_2[0]-check_2[1])        
        return answer
    elif 2 in check and 1 in check :
        for i in range(len(check)) :
            if check[i]==2:
                check_2.append(i+1)
            elif check[i]==1:
                check_1.append(i+1)
        print(check_1)
        answer = check_1[0]*check_1[1]       
        return answer
    
    elif a != b != c != d:
        answer = min(list)
        return answer

 

주사위의 눈 수를 체크 할 수 있는 빈 배열을 생성하고 문제에서 주어진 것 처럼 케이스를 5가지로 나누었다.