28-janghw0126

[janghw0126]

🔗 문제 링크

백준 | 수 정렬하기 2

✔️ 소요된 시간

1시간

✨ 수도 코드

알고리즘 시험 대비 겸 정렬 문제를 풀어보았습니다. 주어진 수를 오름차순으로 정렬하는 것이므로 sort 자료형을 쓰면

import sys

n = int(input())
num = []
for _ in range(n):
    num.append(int(sys.stdin.readline()))

num.sort()

for i in num:
    print(i)

이렇게 굉장히 쉽게 해결되는 것을 알 수 있지만, 정렬 부분을 마스터하기 위해 이번에는 자료형을 쓰지 않고 합병정렬을 이용해서 풀어보았습니다.

Note

합병정렬이란?
데이터가 저장된 배열을 절반으로 나누고 각각을 순환적으로 정렬해서 정렬된 두개의 배열을 합쳐서 정렬을 하는 것이다.
더 자세하게 로직을 설명하자면,

1. 분할 : 해결하고자 하는 문제를 작은 크기의 동일한 문제들로 분할한다.
2. 정복 : 각각의 작은 문제를 순환적으로 해결한다.
3. 합병 : 작은 문제의 해를 합하여 원래 문제에 대한 해를 구한다.

이를 바탕으로 문제를 풀어보았습니다.

1. 분할/정복

# 분할정복을 이용한 합병정렬 구현
def sort(arr):
    if len(arr)<2:
        return arr
    
    mid=len(arr)//2
    left=sort(arr[:mid])
    right=sort(arr[mid:])

    return merge(left,right)

분할/정복 부분은 주어진 배열을 반으로 계속 나누다가 길이가 1이 되는 배열까지 나눈다. 길이가 1이 될 때는 다시 arr를 반환하게 되는데 이때의 arr는 길이가 1인 배열로 각각 분할되어 있는 상태이고 분할된 배열 그 자체로 정렬되어 있는 상태이므로 그 다음 차례로 반환하는 부분인 return merge(left,right)을 차례대로 레벨 별로 반복하여 수행하면서 순환적으로 왼쪽과 오른쪽 배열을 비교하고 정렬해서 합병해나간다.

2. 합병

def merge(left,right):
    new_list=[]
    i=0
    j=0
    
    while (i<len(left)) & (j<len(right)):
        if left[i]>right[j]:
            new_list.append(right[j])
            j+=1
        else:
            new_list.append(left[i])
            i+=1
    while (j<len(right)):
            new_list.append(right[j])
            j+=1
    while (i<len(left)):
            new_list.append(left[i])
            i+=1
    return new_list

분할해서 정렬했던 각각의 왼쪽과 오른쪽 배열을 비교해서 하나로 정렬하는 과정을 반복하는 부분이다.

⬇️최종 코드

# 합병정렬 이용
import sys
input=sys.stdin.readline

n=int(input())
list=[]

for i in range(n):
    list.append(int(input()))

def sort(arr):
    if len(arr)<2:
        return arr
    
    mid=len(arr)//2
    left=sort(arr[:mid])
    right=sort(arr[mid:])

    return merge(left,right)
    
def merge(left,right):
    new_list=[]
    i=0
    j=0
    
    while (i<len(left)) & (j<len(right)):
        if left[i]>right[j]:
            new_list.append(right[j])
            j+=1
        else:
            new_list.append(left[i])
            i+=1
    while (j<len(right)):
            new_list.append(right[j])
            j+=1
    while (i<len(left)):
            new_list.append(left[i])
            i+=1
    return new_list
    
for i in sort(list):
    print(i)

📚 새롭게 알게된 내용

[9kyo-hwang]

헉 혜원상 누구 알고리즘 수업 들으시나요 ?.?

[janghw0126]

@9kyo-hwang 저 권오흠 교수님 수업 듣습니당!

[9kyo-hwang]

@9kyo-hwang 저 권오흠 교수님 수업 듣습니당!

역시 오흠좌인가...

22년도에 제가 들었을 땐 중간고사로

1. merge sort/quick sort 직접 구현하기
2. 리스트에서 k개의 수 뽑아서 합이 0에 가장 가까운 조합 찾기
3. preorder 주어졌을 때 postorder 출력하기(5639 백준)

이렇게 나왔었는데 이번엔 뭐가 나올려나요... 3번 문제는 23년도에도 내셨다고 들었는데...

[janghw0126]

@9kyo-hwang 헐🥹 귀한 정보 정말정말 감사합니다! 저도 정렬 알고리즘 구현하는 부분은 꼭 나올 것 같아서 차근차근 도장깨기 하고 있는 중입니다..파이팅 해봐야죳!👊👊

[9kyo-hwang]

혹시 모르니 퀵 정렬 코드도...

def quicksort(arr):
    if len(arr) <= 1:
        return arr

    pivot = arr[len(arr) // 2]  # 가운데 요소를 pivot으로 선택합니다.
    left = [x for x in arr if x < pivot]
    middle = [x for x in arr if x == pivot]
    right = [x for x in arr if x > pivot]

    return quicksort(left) + middle + quicksort(right)

print(quicksort([3,6,8,10,1,2,1]))

[LJEDD2]

교황님 간결하고 깔끔한 코드덕분에 퀵 정렬을 아주 확실하게 머릿속에 집어넣어버렸습니다 최고 ☘️

가운데에 있는 데이터를 피벗으로 하여 분할 정복과 비슷한 방식으로 접근을 할 수 있군요 .. !
이렇게도 구현해볼 수 있다는 것을 배워갑니다 !!

[LJEDD2]

교황님의 간결하고 깔끔한 코드덕분에 퀵 정렬을 아주 확실하게 머릿속에 집어넣어버렸습니다 최고 ☘️
가운데에 있는 데이터를 피벗으로 하여 분할 정복과 비슷한 방식으로 접근을 할 수 있군요 .. ! 이렇게도 구현해볼 수 있다는 것을 배워갑니다 !!

분할 -> 정복 -> 합병을 통해 가장 작은 단위까지 쪼개어 문제를 합친다는 접근 방식을 혜원님께서 그림으로 깔끔하게 정리해주셔서 이해가 잘 되었습니다 ! 감사합니다 !!

이번차시도 고생많으셨습니다 ✨