[프로그래머스 / 파이썬] flag에 따라 다른 값 반환하기

<내가 푼 풀이>

def solution(a, b, flag):

    
    if flag == 1:
        answer = a + b
    else:
        answer = a - b
    return answer