백준(boj) 10825 파이썬 - 국영수

문제링크 : https://www.acmicpc.net/problem/10825

10825번: 국영수

 

리스트의 요소 인덱스르 기준으로 sort하는 방법을 알아야 했다.

arr.sort(key=lambda x: x[0] )    -> x[0]번 기준으로 sort

*내 코드

N = int(input()) ; arr = []

for _ in range(N):
    arr.append(list(map(str,input().split())))
arr.sort(key=lambda x : str(x[0])) ##4
arr.sort(key=lambda x:int(x[3]), reverse=True) ##3
arr.sort(key = lambda x : int(x[2])) ## 2
arr.sort(key = lambda x: int(x[1]), reverse = True)##1


for i in arr :
    print(i[0])

 

arr.sort를 4번이나 사용했다. 이렇게 일일이 사용했을때는 문제에서 제시한 정렬순서와, 실제로 소팅하는 순서를 반대로 해야했다. (문제에서 제시한 1순위 기준을 마지막 순서로 코딩해야, 1순위로 적용되기 때문)

다른사람의 코드를 보니, 이런식으로 좀 더 깔끔하게 사용했다.

*모범 코드

N = int(input()) ; arr = []

for _ in range(N):
    a,b,c,d = list(map(str,input().split()))
    arr.append( [a, int(b),int(c),int(d)])

arr.sort(key = lambda x : (-x[1] , x[2],-x[3],x[0]) )

for i in arr :
    print(i[0])

 

 

sort(reverse=True)를 람다식에서, 마이너스를 붙이는 방법으로 짧게 사용할 수 있었다. 

이렇게 하니 문제에서 제시하는 정렬순서와, 코딩하는 정렬순서도 일치하고 좀 더 가독성도 좋았다.

arr.sort(key = lambda x : x[1], reverse=True) == arr.sort(key=lambda x : -x[1])