프로그래머스 - '주사위 게임 1' (PYTHON 풀이)

문제:

 

두 개의 주사위를 굴렸을 때 나온 결과를 바탕으로 점수를 계산해서 출력하는 문제이다.

두 개의 주사위 모두 홀수라면 각 수의 제곱을 더한 값을, 하나만 홀수이면 두 수를 더한 다음 2를 곱한 수를, 둘 다 홀수가 아니라면 절댓값 기준 하나의 주사위에서 다른 하나의 주사위의 수를 뺀 수를 출력하면 된다.

 

풀이:

def solution(a, b):
    score = 0
    
    if a % 2 != 0 and b % 2 != 0:
        score += a**2 + b**2
    elif a % 2 != 0 or b % 2 != 0:
        score += 2 * (a + b)
    else:
        score += abs(a - b)
    
    return score

두 개의 수를 입력받고, 두 개의 값 모두 2로 나눴을 때 0이 아닌 경우(홀수인 경우) 앞서 말한 조건대로 각 수의 제곱값을 더한 수를 출력하고, 둘 중 하나의 수만 홀수인 경우 각 수를 더한 다음 2로 곱한 수를, 두 가지 조건이 모두 아니라면 절댓값 기준 a에서 b를 뺀 수를 score에 담고 마지막으로 score 값을 출력하면 된다.