[iOS / Swift] flag에 따라 다른 값 반환하기

• 출처 : 프로그래머스 (https://school.programmers.co.kr/learn/courses/30/lessons/181933)
• 난이도 : Lv0
• 정답률 : 89%
• 알고리즘 유형 : 연산, 조건문

---

문제

풀이

import Foundation

func solution(_ a:Int, _ b:Int, _ flag:Bool) -> Int {
    flag == true ? a + b : a-b
}

최적화

import Foundation

func solution(_ a:Int, _ b:Int, _ flag:Bool) -> Int {
    flag ? a + b : a-b
}

요약

• 삼항연산자 사용할 것