프로그래머스 코딩테스트 연습 [주사위 게임2]

문제 설명

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

• 세 문자가 모두 다르다면  a + b + c 점을 얻습니다.
• 세 숫자 중 어느 두 숫자는 같고 나머지 다른 숫자는 다르다면 (a+b+c)*(a^2+b^2+c^) 점을 얻습니다.
• 세 숫자가 모두 같다면 (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 이하의 정수입니다.

입출력 예

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

입출력 예 #3

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

 

해결

내 풀이

def solution(a, b, c):
    answer = 0
    if a != b and b != c and a != c:
        answer = a+b+c
    elif a==b and b==c:
        answer = (a+b+c)*(a*a+b*b+c*c)*(a*a*a+b*b*b+c*c*c)
    elif a==b and b!=c:
        answer = (a+b+c)*(a*a+b*b+c*c)
    return answer

 

다른 풀이

다른 풀이 #1

def solution(a, b, c):
	check = len(set([a,b,c]))
    if check == 1:
    	return 3*a * 3*(a**2) * 3*(a**3)
    elif check == 2:
    	return (a+b+c)*(a**2 + b**2 + c**2)
    else:
    	return (a+b+c)

 

set으로 동일값을 추출 -> 중복을 없애서 길이로 체크

set()은 클래스이기 때문에 set(a, b, c)일 경우 a, b, c가 각각 set의 첫번째, 두번째, 세번째 파라미터가 된다.

따라서 set([a, b, c])처럼 list로 감싸서 넘겨야 함.

 

다른 풀이 #2

def solution(a, b, c):
    answer = a+b+c
    if a==b or b==c or a==c:
    	answer *= a**2 + b**2 + c**2
    if a==b==c:
    	answer *= a**3 + b**3 + c**3
    return answer

 

answer 식을 먼저 세우고, 이 후에 경우에 따른 식 변경

 

새로 알게 된 점

** -> 제곱 연산자

*= 곱해서 할당