주사위 게임 1

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

a와 b가 모두 홀수라면 a^2 + b^2 점을 얻습니다.

a와 b 중 하나만 홀수라면 2 × (a + b) 점을 얻습니다

a와 b 모두 홀수가 아니라면 |a - b| 점을 얻습니다.

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

내 코드

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

예시 코드

def solution(a, b):
        return a*a+b*b if a & b & 1 else (a + b) << 1 if (a | b) & 1 else abs(a - b)