[Python] 프로그래머스 - 코딩테스트입문 : 두 수의 합 구하기 / lambda와 가변인자 *

 

 

코딩테스트 입문 | 프로그래머스 스쿨

 

 

 

---

 

 

1. 문제 설명

정수 num1과 num2가 주어질 때,
num1과 num2의 합을 return하도록 soltuion 함수를 완성해주세요.

 

 

 

2. 입출력 예

#1
num1이 2이고 num2가 3이므로 2 + 3 = 5를 return합니다.


#2
num1이 100이고 num2가 2이므로 100 + 2 = 102를 return합니다.

 

 

 

3. 문제풀이 

내가 푼 풀이

def solution(num1, num2):
    answer = -1

    answer = num1 + num2
    return answer

다른 사람이 푼 풀이

solution = lambda *x:sum(x)

다른 사람이 푼 풀이를 들어가서 확인해보니 처음보는 형태의 코드가 있었다. 물론 lambda는 많이 봤는데 *x가 뭔지 몰라서, 알아본 김에 블로그에 정리해두면 도움이 될 것 같아 이렇게 적어보려 한다. 

 

 

 

 

4. lambda와 가변 인자

 

lambda

• 익명 함수(이름 없는 함수)를 만들 때 사용한다.
• 일반 함수는 def를 사용하는데, 이와 달리 lambda는 한 줄로 간단히 정의 가능하다.

 

*x

• 함수의 매개변수 앞에 *가 붙으면 가변인자를 받는다는 의미가 된다.
• 몇 개의 인자가 들어올지 정해져 있지 않고, 들어온 모든 인자가 튜플로 묶인다.

 

예시코드

def f(*x):
    print(x)

f(1, 2, 3)

f = lambda *x: x

print(f(1, 2, 3))
print(f(10, 20, 30, 40))

# 출력
# (1, 2, 3)
# (10, 20, 30, 40)

첫번째 코드를 보자. 1, 2, 3이라는 값들이 함수 안으로 들어갈 때 자동으로 (1, 2, 3)이라는 튜플이 된다. 이때 튜플은 파이썬 자료형 중 하나로, 순서가 있는 여러 값을 한 덩어리로 저장한다. 튜플은 소괄호를 쓰니 알아두자!

 

 

solution = lambda *x:sum(x)

sum은 파이썬의 내장함수로 알아서 x 값을 더해준다.

내가 푼 풀이는 딱 두 개의 값만 들어왔을 때 풀 수 있었다. 하지만 이 풀이는 여러 개의 인자가 들어와도 튜플로 받아 처리하니, 인자의 갯수가 일정하지 않은 경우 사용하면 아주 효율적일 것 같다.

 

 

def solution(*x):
    return sum(x)

일반 함수와 가변 인자를 함께 써봤다.