[프로그래머스] 코딩 기초 트레이닝 - flag에 따라 다른 값 반환하기

문제

두 정수 a, b 와 boolean 변수 flag가 매개변수로 주어질 때, flag가 true면 a + b 를 false면 a -b를 return 하는 solution 함수를 작성해주세요.

입출력 예

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

조건문(if - else) 사용하여 조건이 flag 참이면 a+b 리턴하고, flag가 거짓이면 a-b 리턴합니다

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

파이썬에서 삼항 연산자는 참일 때 값 if 조건 else 거짓일 때 값 형태로 사용하여 다른 값을 반환합니다.

def solution(a, b, flag):
    operations = {
        True: a + b,
        False: a - b
    }
    return operations[flag]

딕셔너리의 키로 참과 거짓을 사용하며, 이 키에 각각 해당하는 연산 결과를 값으로 할당합니다. 대괄호 [ ]를 사용하여 딕셔너리의 특정 키에 해당하는 값을 접근하고, 이값을 반환합니다.

출처