[Lv.0/Python] 주사위 게임1

🔗 문제

https://school.programmers.co.kr/learn/courses/30/lessons/181839

 

💻 Code

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