[프로그래머스] Lv.0 /flag에 따라 다른 값 반환하기 (파이썬/Python)

1. 문제 설명

2. 풀이 과정

flag는 불린 정수이기 때문에 1일때 true 0일때 false이다.
따라서 flag가 1이면 a+b를 반환하고, 0이면 a-b를 반환한다.

3. 코드

def solution(a, b, flag):
    if flag == 1:
        return a+b
    else:
        return a-b