Your Name

I am an amazing person.

2 분 소요

문제 설명

프로그래머스 181930번 주사위 게임 2

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

  • 세 숫자가 모두 다르다면 a + b + c 점을 얻습니다.
  • 세 숫자 중 어느 두 숫자는 같고 나머지 다른 숫자는 다르다면 (a + b + c) × (a^2 + b^2 + c^2 )점을 얻습니다.
  • 세 숫자가 모두 같다면 (a + b + c) × (a^2 + b^2 + c^2 ) × (a^3 + b^3 + c^3 )점을 얻습니다.

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

제한 사항

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

입출력 예시

예시 1

입출력 1

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) × (5^2 + 3^2 + 3^2 ) = 11 × 43 = 473점을 얻습니다. 따라서 473을 return 합니다.

입출력 예 설명 3

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

문제 해결 전략

1. 접근 방법

  • 세 수가 모두 같은 경우, 세 수가 모두 다른 경우를 분류하기 빠르므로 두 경우를 먼저 분류한다.

코드 구현

Python

def solution(a, b, c):
    if a != b and a != c and b != c:
        return a + b + c
    elif a == b and a == c and b == c:
        return (a + b + c) * (a ** 2 + b ** 2 + c** 2) * (a ** 3 + b ** 3 + c ** 3)
    else:
        return (a + b + c) * (a ** 2 + b ** 2 + c** 2)

Java

class Solution {
    public int solution(int a, int b, int c) {
        if (a != b && a != c && b != c) {
            return a + b + c;
        } else if (a == b && a == c && b == c) {
            return (a + b + c) * (a * a + b * b + c * c) * (a * a * a + b * b * b + c * c * c);
        } else {
            return (a + b + c) * (a * a + b * b + c * c);
        }
    }
}

개선할 점 - Java

  • Math.pow() 메서드를 이용하여 거듭제곱을 표현할 수 있다.
  • else if 문에서 b == c는 a와 b가 같고 a와 c가 같으면 자연스럽게 b와 c도 같은 것이 되므로 생략할 수 있다.
class Solution {
    public int solution(int a, int b, int c) {
        if (a != b && a != c && b != c) {
            return a + b + c;
        } else if (a == b && a == c) {
            return (int) ((a + b + c) * (Math.pow(a, 2) + Math.pow(b, 2) + Math.pow(c, 2)) * (Math.pow(a, 3) + Math.pow(b, 3) + Math.pow(c, 3)));
        } else {
            return (int) ((a + b + c) * (Math.pow(a, 2) + Math.pow(b, 2) + Math.pow(c, 2)));
        }
    }
}

참고