
정렬 관련 문제에서 자주 기용되는 유형과 기본형 코드
1. 버블 정렬
- 원리: 인접한 두 원소를 비교해가며 정렬함.
def bubbleSort(arr): n = len(arr) for i in range(n): for j in range(0, n-i-1): if arr[j] > arr[j+1]: arr[j], arr[j+1] = arr[j+1], arr[j]
2. 선택 정렬
- 원리: 리스트에서 최솟값 또는 최댓값을 하나씩 이동하면서 정렬함.
def selectionSort(arr): for i in range(len(arr)): min_idx = i for j in range(i+1, len(arr)): if arr[min_idx] > arr[j]: min_idx = j arr[i], arr[min_idx] = arr[min_idx], arr[i]
3. 삽입 정렬
- 원리: 원소를 적절한 위치에 삽입하며 배열 확장.
def insertionSort(arr): for i in range(1, len(arr)): key = arr[i] j = i-1 while j >=0 and key < arr[j]: arr[j+1] = arr[j] j -= 1 arr[j+1] = key
4. 퀵 정렬
- 원리: pivot을 선택하고 작은 원소는 왼쪽, 큰 원소는 오른쪽으로 이동.
def quickSort(arr): if len(arr) <= 1: return arr pivot = arr[len(arr) // 2] 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)
5. 병합 정렬
- 원리: 리스트를 반으로 나누고 각 부분을 정렬한 후 병합.
def mergeSort(arr): if len(arr) > 1: mid = len(arr)//2 L = arr[:mid] R = arr[mid:] mergeSort(L) mergeSort(R) i = j = k = 0 while i < len(L) and j < len(R): if L[i] < R[j]: arr[k] = L[i] i += 1 else: arr[k] = R[j] j += 1 k += 1 while i < len(L): arr[k] = L[i] i += 1 k += 1 while j < len(R): arr[k] = R[j] j += 1 k += 1