[코딩테스트 Lv.0] 주사위 게임 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 함수를 작성해 주세요.

제한사항

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
예제 1번에서 네 주사위 숫자가 모두 2로 같으므로 1111 × 2 = 2222점을 얻습니다. 따라서 2222를 return 합니다.

입출력 예 #2
예제 2번에서 세 주사위에서 나온 숫자가 4로 같고 나머지 다른 주사위에서 나온 숫자가 1이므로 (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 합니다.

---

내 풀이

Java

import java.util.*;
class Solution {
    public int solution(int a, int b, int c, int d) {
        int answer = 0;
        
        int arr[] = {a, b, c, d};
        Arrays.sort(arr);
        
        
        if(arr[0] == arr[3]){//4개 일치
            answer = 1111 * arr[0];
        }else if(arr[0] == arr[2] || arr[1] == arr[3]){//3개 일치
            answer = (int)Math.pow((10 * arr[1]) + (arr[0] + arr[3] - arr[1]), 2);
        }else if(arr[0] == arr[1] && arr[2] == arr[3]){//2개, 2개 일치
            answer = (arr[2] + arr[0]) * Math.abs(arr[2] - arr[0]);
        }else if(arr[0] == arr[1]){//2개 일치
            answer = arr[2] * arr[3];
        }else if(arr[1] == arr[2]){//2개 일치
            answer = arr[0] * arr[3];
        }else if(arr[2] == arr[3]){//2개 일치
            answer = arr[0] * arr[1];
        }else{//일치하지 않음
            answer = arr[0];
        }        
        
        return answer;
    }
}

C

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>
#include <math.h>
int solution(int a, int b, int c, int d) {
    int answer = 0;
    int arr[4] = {a, b, c, d};
    int i, j;
    for(i = 0; i < 3; i++){
        for(j = i + 1; j < 4; j++){
            int tmp = 0;
            if(arr[i] > arr[j]){
                tmp = arr[j];
                arr[j] = arr[i];
                arr[i] = tmp;
            }
        }
    }
    if(arr[0] == arr[3]){
        answer = 1111 * arr[0];
    }else if(arr[0] == arr[2] || arr[1] == arr[3]){
        answer = pow((10 * arr[1]) + (arr[0] + arr[3] - arr[1]), 2);        
    }else if(arr[0] == arr[1] && arr[2] == arr[3]){
        answer = (arr[2] + arr[0]) * abs(arr[2] - arr[0]);
    }else if(arr[0] == arr[1]){
        answer = arr[2] * arr[3];
    }else if(arr[1] == arr[2]){
        answer = arr[0] * arr[3];
    }else if(arr[2] == arr[3]){
        answer = arr[0] * arr[1];
    }else{
        answer = arr[0];
    }
    return answer;
}

Python

def solution(a, b, c, d):
    answer = 0
    arr = [a, b, c, d]
    arr = sorted(arr)
    
    if arr[0] == arr[3]:
        answer = 1111 * arr[0]
    elif arr[0] == arr[2] or arr[1] == arr[3]:
        answer = pow((10 * arr[1]) + (arr[0] + arr[3] - arr[1]), 2)
    elif arr[0] == arr[1] and arr[2] == arr[3]:
        answer = (arr[2] + arr[0]) * abs(arr[2] - arr[0]) 
    elif arr[0] == arr[1]:
        answer = arr[2] * arr[3]
    elif arr[1] == arr[2]:
        answer = arr[0] * arr[3]
    elif arr[2] == arr[3]:
        answer = arr[0] * arr[1]
    else:
        answer = arr[0]
        
    return answer

---

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

프로그래머스

---

[코딩테스트 Lv.0] 주사위 게임 1