[프로그래머스] 주사위 게임 1

 

<내가 쓴 코드>

def solution(a, b):
    answer = 0
    if a%2==1 and b%2==1:   #a와 b가 모두 홀수
        answer += a**2+b**2   #a^2+b^2를 answer에 추가
    elif a%2==1 or b%2==1:   #a와 b 중 하나만 홀수
        answer += (a+b)*2   #(a+b)*2를 answer에 추가
    else:   #a와 b 모두 홀수가 아니면
        if a > b:   #절대값을 표현하기 위해...
            answer += a-b
        else:
            answer += b-a
    return answer