flag에 따라 다른 값 반환하기.py

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