[python] List(2) list sorting : 다중 조건 정렬 & 백준 1181번[단어 정렬]

Hello👋 I'm Dona!:)

지난 포스팅에 이어서 알고리즘 문제에서 자주 쓰이는 다중 조건 정렬에 대해 알아보겠습니다.

sort와 sorted의 documentation 소개는 지난 포스팅 참고해주세요! ⬇️⬇️⬇️⬇️⬇️

https://imdona.tistory.com/12

[python] List(1) list sorting : sort & sorted 소개

---

백준 알고리즘 문제를 예시로 배워봅시다!

백준 1181번 [단어 정렬]

문제

알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬하는 프로그램을 작성하시오.
1. 길이가 짧은 것부터
2. 길이가 같으면 사전 순으로

입력

첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000) 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다. 주어지는 문자열의 길이는 50을 넘지 않는다.

출력

조건에 따라 정렬하여 단어들을 출력한다. 단, 같은 단어가 여러 번 입력된 경우에는 한 번씩만 출력한다.

입력되는 예제

13
but
i
wont
hesitate
no
more
no
more
it
cannot
wait
im
yours

---

문제를 보시면 정렬의 기준이 두 가지입니다.

1. 길이

2. 사전 순

 

물론 아래와 같이 "key=len" 이라는 파라미터만 넣어주어도 사전 순으로 정렬되기는 했지만, 오늘 배울 다중 조건 정렬로도 적용해서 풀어볼께요!

'''첫번째 방법
-> sort(key=len) : 길이 순으로 정렬
sort(key=func) : func는 하나의 값을 취하고 정렬을 안내하는 프록시 값을 반환하는 함수'''

N = int(input())
word_list = [input() for _ in range(N)]
word_list = sorted(list(set(word_list))) # 중복 제거
word_list.sort(key=len) # str 길이 순 정렬
for word in word_list:
    print(word)

두 조건을 적용하기 위해서 (word, len(word))가 들어 있는 튜플을 만들었어요! 

그리고 key 조건에 함수로 lamda를 넘겨주고  길이 순인 x[1] -> 사전순인 x[0] 순으로 정렬을 해줍니다!

(여기서 x는 하나의 튜플을 예로 들면, ('word', 4) 라는 튜플에서 인덱스 번호가 0번이 단어이고 1번이 길이겠죠!)

결론은 lambda 함수를 이용해서 원하는 함수로 순서로 다중 조건 정렬이 가능하다는 것입니다☺️ 

'''두번째 방법
1. word, word의 길이인 튜플을 원소로 갖는 리스트로 만든다
2. 정렬 메소드인 sort에 key optional parameter 사용(길이(x[1]) -> 알파벳 순(x[0]) 정렬)'''
N = int(input())
word_list = []
for i in range(N):
    word = str(input())
    word_list.append((word, len(word))) # word, word의 길이인 튜플을 원소로 갖는 리스트로 만든다
word_list = list(set(word_list))
word_list.sort(key = lambda x: (x[1], x[0])) 
# 정렬 메소드인 sort에 key optional parameter 사용(길이(x[1]) -> 알파벳 순(x[0]) 정렬)
for i in word_list:
    print(i[0])

세번째 방법은 제가 해당 문제를 고민하면서 뒤늦게 생각이 난 방법인데, 튜플의 순서를 len, word 순으로 바꾸면 lambda 함수 사용하지 않고 바로 정렬 할 수 있습니다!

'''세번째 방법
-> 두번째 방법에서 튜플의 순서를 len, word 순으로 바꾸면 lambda 함수 사용하지 않고 바로 정렬 할 수 있다'''
N = int(input())
word_list = []
for i in range(N):
    word = str(input())
    word_list.append((len(word), word)) # len, word인 튜플을 원소로 갖는 리스트로 만든다
word_list = list(set(word_list))
word_list.sort() # 튜플의 첫번째 값인 len먼저 정렬 -> word 다음으로 알아서 정렬함!
for word_len, word in word_list:
    print(word)

---

알고리즘 말고 예시로도 한번 만나볼까요?! (아래 REFERENCE 블로그 참고했습니다 ☺️)

(이름 , 나이 , 재산) 으로 예시를 들어보겠습니다. 

아래의 다중 조건으로 정렬을 해봅시다!

1. 나이 순으로 오름차순

2. 같은 나이라면 재산으로 내림차순

# (<이름> , <나이> , <재산>)
array = [("Dona", 27, 300) , ("Mr.Salmon", 27, 10000), ("KingHee", 20, 200000),("Seoki",13,5000000), ("Teco", 70, 5000000)]

array.sort(key = lambda x: (x[1], -x[2]))
print(array)

# print out : [('Seoki', 13, 5000000), ('KingHee', 20, 200000), ('Mr.Salmon', 27, 10000), ('Dona', 27, 300), ('Teco', 70, 5000000)]

1. 나이 순으로 오름차순 -> 튜플의 인덱스 번호 1(나이), +(오름차순)

2. 같은 나이라면 재산으로 내림차순 -> 튜플의 인덱스번호2(재산), -(내림차순)

---

이상으로 포스팅을 마치겠습니다. 감사합니다. 틀린 내용이 있다면 댓글로 알려주세요 :) GO FOR IT!! BYE!! 👋

---

REFERENCE

• 참고한 블로그 : https://infinitt.tistory.com/122
• https://imdona.tistory.com/12