코딩테스트 - 주사위 게임 2

SSO·2024년 11월 14일

주사위 게임 2

문제 설명

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

세 숫자가 모두 다르다면 a + b + c 점을 얻습니다.
세 숫자 중 어느 두 숫자는 같고 나머지 다른 숫자는 다르다면 (a + b + c) × (a의 제곱 + b의 제곱 + c의 제곱 )점을 얻습니다.
세 숫자가 모두 같다면 (a + b + c) × (a의 제곱 + b의 제곱 + c의 제곱 ) × (a의 세제곱 + b의 세제곱 + c의 세제곱 )점을 얻습니다.
세 정수 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

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

입출력 예 #3

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

풀이(개인)

def solution(a, b, c):
    numbers = [a,b,c]
    score_1 = sum(numbers)
    score_2 = sum(n**2 for n in numbers)
    score_3 = sum(n**3 for n in numbers)
    
    if len(set(numbers)) == 3:
        answer = score_1
    elif len(set(numbers)) == 2:
        answer = score_1*score_2
    else:
        answer = score_1*score_2*score_3
    return answer

set() 함수란?

집합을 생성하는 함수다.
집합은 중복을 허용하지 않으며, 순서가 없다.
리스트, 튜플, 문자열 등을 인수로 전달하면 중복된 요소들이 자동으로 제거된다.

다른 사람들의 풀이

def solution(a, b, c):
check=len(set([a,b,c]))
if check == 1:
return 3a3(a**2)3(a**3)
elif check == 2:
return (a+b+c)(a2 + b2+c**2)
else:
return (a+b+c)
- set()이 클래스이기 때문에, set(a,b,c)일 경우 a는 set의 첫번째 파라미터, b는 두번째 파라미터, c는 세번째 파라미터로 인식됩니다. 따라서, set([a,b,c])처럼 list로 감싸서 넘겨줍니다.

def solution(a, b, c):
answer=a+b+c
if a==b or b==c or a==c: answer=a2+b2+c**2
if a==b==c:answer=a3+b3+c**3
return answer
solution = lambda a, b, c: a+b+c if a!=b and b!=c and c!=a else ((a+b+c)(a2+b2+c**2)(a3+b3+c3) if a==b and b==c else (a+b+c)*(a2+b2+c2))

def solution(a, b, c):
list = [a, b, c]
answer = 1
for i in range(4-len(set(list))):
answer *= a(i+1)+b(i+1)+c**(i+1)
return answer
def solution(a, b, c):
answer = 0
if a == b == c:
answer = (a+b+c) (a2 + b2 + c**2) (a3 + b3 + c3)
elif a==b or a==c or b==c:
answer = (a+b+c) * (a2 + b2 + c2)
else:
answer = (a+b+c)
return answer
def solution(a, b, c):
answer = 0
if a==b and a==c:
answer=(a+b+c)(a2+b2+c**2)(a3+b3+c3)
elif a==b or b==c or a==c:
answer=(a+b+c)*(a2+b2+c2)
else:
answer=a+b+c
return answer
def solution(a, b, c):
if a == b == c:
return (a+b+c)(a2+b2+c**2)(a3+b3+c3)
elif a==b or b==c or a==c:
return (a+b+c)*(a2+b2+c2)
else:
return a+b+c