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

✏️ 문제

https://school.programmers.co.kr/learn/courses/30/lessons/181933?language=python3

 

Python 

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

 

다른 사람의 문제 풀이 

lambda를 활용하는 방법에 대하여 검색해봐야겠다...

solution=lambda a,b,f:[a-b,a+b][f]

 

Java

class Solution {
    public int solution(int a, int b, boolean flag) {
        return flag ? a + b : a-b;
    }
}