[프로그래머스] Python - Lv.0 flag에 따라 다른 값 반환하기

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

프로그래머스

 

문제

 

코드

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