[python] 리스트 정렬, 람다식, list 응용

아래의 출처를 이용해 python 리스트에 대해 자세히 알아보자.

 

출처: https://www.acmicpc.net/problem/2456

 

이 문제를 통해 파이썬 리스트와 정렬, 그리고 람다식을 이용해 리스트르 응용할 수 있다.

해답 코드는 아래와 같다.

 

import sys
 
n = int(input())
Sum = [0 for _ in range(3)]
num = [[0 for _ in range(5)] for _ in range(3)]
num2 = [[0,0,0,0]]*4
chk = []
#print(sum)
#num[0][1]=3
#print(num)
for _ in range(n):
    a,b,c = map(int, sys.stdin.readline().split())
    Sum[0] = Sum[0] + a
    num[0][a-1] += 1
    Sum[1] = Sum[1]+b
    num[1][b-1] += 1
    Sum[2] = Sum[2]+c
    num[2][c-1] += 1
num[0][3] ,num[1][3],num[2][3] = Sum[0],Sum[1],Sum[2]
num[0][4] , num[1][4], num[2][4] = 1,2,3
#print(num)
num = sorted(num,key = lambda x:(x[3],x[2],x[1]))
ans,ans2 = num[2][4],num[2][3]
for i in num:
    if i[3] == ans2: chk.append(i[4])
print(len(chk))
if len(chk) > 1:
    num[0].pop()
    num[1].pop()
    num[2].pop()
    if num[2] == num[1]:
        print(0, ans2)
    elif num[0] == num[1]:
        print(ans,ans2)
    else:print(ans,ans2)
else: print(ans,ans2)
print(num)
#print(num)
 
'''
arrrr = [3,4,2,5,1]
arrrr.sort()
print(arrrr)
'''

 

리스트를 보면 [] 내부에 for문을 사용하거나 []내부에 []가 있는 이중 리스트가 존재한다.

 

num 경우에는 [[0,0,0,0,0], [0,..0],[0...0]] 와 같이 값이 저장될 것이다. []내부에 for문을 사용해 초기화 할 수 있다는것을 명심하면 된다.

 

접근은 8,9,10 번 라인의 출력을 확인하면 어떻게 이루어져 있는지 쉽게 알 수 있다.

 

그리고 num2 리스트와 sum 리스트에 더한값들을 넣는다.

 

python에서 list 정렬은 sort()와 sorted()로 할 수 있으며 sort는 '리스트이름'.sort() 로 사용하는 리스트 메서드이다. sorted는 위와 같이 사용하면 된다, 2개다 람다함수를 사용가능하며 위처럼 람다함수를 사용하게 되면 순서대로 정렬을 실시한다.

정렬후 출력결과를 확인해보면 이해할 수 있다.

 

리스트 pop함수를 이용해 이중 리스트에서 어떻게 값이 pop 되는지 확인할 수 있다.   

 

파이썬으로 더 쉽게 풀 수 있으나.. C++이 익숙하여 C++방식대로 하다보니 코드가 길고 지저분해 보일 수있다. 다른 방법이 있으니 list에 대해 잘 이해하고 활용할 수 있는 연습이 되었다.