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

문제 설명

1부터 6까지 숫자가 적힌 주사위가 세 개 있습니다. 세 주사위를 굴렸을 때 나온 숫자를 각각 a, b, c라고 했을 때 얻는 점수는 다음과 같습니다.

• 세 숫자가 모두 다르다면 a + b + c 점을 얻습니다.
• 세 숫자 중 어느 두 숫자는 같고 나머지 다른 숫자는 다르다면 (a + b + c) × (a2 + b2 + c2 )점을 얻습니다.
• 세 숫자가 모두 같다면 (a + b + c) × (a2 + b2 + c2 ) × (a3 + b3 + c3 )점을 얻습니다.

세 정수 a, b, c가 매개변수로 주어질 때, 얻는 점수를 return 하는 solution 함수를 작성해 주세요.

제한사항

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

입출력 예

a	b	c	result
2	6	1	9
5	3	3	473
4	4	4	110592

입출력 설명

입출력 예 #1
예제 1번에서 세 주사위 숫자가 모두 다르므로 2 + 6 + 1 = 9점을 얻습니다. 따라서 9를 return 합니다.

입출력 예 #2
예제 2번에서 두 주사위 숫자만 같으므로 (5 + 3 + 3) × (52 + 32 + 32 ) = 11 × 43 = 473점을 얻습니다. 따라서 473을 return 합니다.

입출력 예 #3
예제 3번에서 세 주사위 숫자가 모두 같으므로 (4 + 4 + 4) × (42 + 42 + 42 ) × (43 + 43 + 43 ) = 12 × 48 × 192 = 110,592점을 얻습니다. 따라서 110592를 return 합니다.

---

내 풀이

Java

import java.util.*;
class Solution {
    public int solution(int a, int b, int c) {
        int answer = 0;
        
        int[] arr = {a, b, c};
        Arrays.sort(arr);
        
        if(arr[0] == arr[2]){
            answer = (arr[0] + arr[1] + arr[2]) 
                * ((int)Math.pow(arr[0], 2) + (int)Math.pow(arr[1], 2) + (int)Math.pow(arr[2], 2))
                * ((int)Math.pow(arr[0], 3) + (int)Math.pow(arr[1], 3) + (int)Math.pow(arr[2], 3));
        }else if(arr[0] == arr[1] || arr[1] == arr[2]){
            answer = (arr[0] + arr[1] + arr[2]) 
                * ((int)Math.pow(arr[0], 2) + (int)Math.pow(arr[1], 2) + (int)Math.pow(arr[2], 2));
        }else{
            answer = arr[0] + arr[1] + arr[2];
        }
        
        return answer;
    }
}

C

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>
#include <math.h>

int solution(int a, int b, int c) {
    int answer = 0;    
    int arr[3] = {a, b, c};
    int i, j;
    for(i = 0; i < 2; i++){
        for(j = i + 1; j < 3; j++){
            int tmp = 0;
            if(arr[i] > arr[j]){
                tmp = arr[j];
                arr[j] = arr[i];
                arr[i] = tmp;
            }
        }
    }
    
    if(arr[0] == arr[2]){
        answer = (arr[0] + arr[1] + arr[2]) 
            * (pow(arr[0], 2) + pow(arr[1], 2) + pow(arr[2], 2))
            * (pow(arr[0], 3) + pow(arr[1], 3) + pow(arr[2], 3));
    }else if(arr[0] == arr[1] || arr[1] == arr[2]){
        answer = (arr[0] + arr[1] + arr[2]) 
            * (pow(arr[0], 2) + pow(arr[1], 2) + pow(arr[2], 2));            
    }else{
        answer = (arr[0] + arr[1] + arr[2]);
    }
    
    return answer;
}

Python

def solution(a, b, c):
    answer = 0
    arr = [a, b, c]
    arr = sorted(arr)
    
    if arr[0] == arr[2]:
        answer = sum(arr) * (pow(arr[0], 2) + pow(arr[1], 2) + pow(arr[2], 2)) * (pow(arr[0], 3) + pow(arr[1], 3) + pow(arr[2], 3))
    elif arr[0] == arr[1] or arr[1] == arr[2]:
        answer = sum(arr) * (pow(arr[0], 2) + pow(arr[1], 2) + pow(arr[2], 2))
    else:
        answer = sum(arr)
    
    return answer

---

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

프로그래머스

---

 

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

 

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