개요
람다 표현식을 사용하면 함수를 매우 간단하게 작성할 수 있다.
예를 들어 add 메소드를 사용한다고 치면
def add(a,b):
return a+b
print(add(3,7))이 코드를
print((lambda a,b:a+b)(3,7))로 줄일 수 있다.
lambda에서 : 왼쪽에 오는 값이 인자, 오른쪽에 오는 값이 반환값이다. 람다식을 괄호로 묶고, 그 옆에 또 괄호를 붙여서 인자를 넣어서 호출할 수 있다.
이 람다식은 sort 함수를 쓸때 key를 설정하기 위해 자주 사용된다. 백준 1931번을 보자.
문제
한 개의 회의실이 있는데 이를 사용하고자 하는 N개의 회의에 대하여 회의실 사용표를 만들려고 한다. 각 회의 I에 대해 시작시간과 끝나는 시간이 주어져 있고, 각 회의가 겹치지 않게 하면서 회의실을 사용할 수 있는 회의의 최대 개수를 찾아보자. 단, 회의는 한번 시작하면 중간에 중단될 수 없으며 한 회의가 끝나는 것과 동시에 다음 회의가 시작될 수 있다. 회의의 시작시간과 끝나는 시간이 같을 수도 있다. 이 경우에는 시작하자마자 끝나는 것으로 생각하면 된다.
입력
첫째 줄에 회의의 수 N(1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N+1 줄까지 각 회의의 정보가 주어지는데 이것은 공백을 사이에 두고 회의의 시작시간과 끝나는 시간이 주어진다. 시작 시간과 끝나는 시간은 231-1보다 작거나 같은 자연수 또는 0이다.
출력
첫째 줄에 최대 사용할 수 있는 회의의 최대 개수를 출력한다.
예제
풀이법
회의 종료시간을 기준으로 먼저 정렬한 다음, 회의 시작시간을 기준으로 정렬한다. 이후 정렬된 회의들을 하나씩 꺼내 보면서 이전 회의의 종료 시간이 다음 회의의 시작 시간과 같거나 작으면 카운트한다.
람다표현식 활용
여기서 정렬기준1-종료시간 정렬기준2-시작시간 을 구현할때 람다식을 쓸 수 있다.
n=int(input())
meeting=[]
for _ in range(n):
start,end=map(int,input().split())
meeting.append((start,end))
print(meeting)
시작시간,종료시간을 입력받아 튜플로 구성된 리스트를 만든다.
튜플의 두번째 원소를 정렬기준1, 첫번째 원소를 정렬기준2로 정렬하기 위해..
#정렬 조건1- 종료 시간 ,정렬조건2- 시작 시간
meeting.sort(key=lambda x:(x[1],x[0]))
print(meeting)
번외
내림차순으로 정렬하고 싶다면 -를 붙여서 사용하면 된다.
예를 들어 회의종료시간 내림차순 시작시간 오름차순으로 하고 싶으면
meeting.sort(key=lambda x:(-x[1],x[0]))
print(meeting)이런 식으로 사용할 수 있다.
참고
