문제 링크

정답 :

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