[이코테 Python] 정렬 알고리즘 문제

정렬 알고리즘 문제

정렬 알고리즘

선택 정렬

# 선택 정렬
arr = [7, 5, 9, 0, 3, 1, 6, 2, 4, 8]
for i in range(len(arr)):
    min_index = i
    for j in range(i + 1, len(arr)):
        if arr[min_index] > arr[j]:
            min_index = j
    arr[i], arr[min_index] = arr[min_index], arr[i]

print("선택 정렬 :", arr)

삽입 정렬

# 삽입 정렬
arr = [7, 5, 9, 0, 3, 1, 6, 2, 4, 8]
for i in range(1, len(arr)): # i-1까지는 정렬되어 있다고 가정
    for j in range(i, 0, -1):
        if arr[j] < arr[j - 1]:
            arr[j], arr[j - 1] = arr[j - 1], arr[j]
        else: # 자신보다 작은 데이터를 만나면 그 위치에서 멈춤
            break

print("삽입 정렬 :", arr)

퀵 정렬

# 퀵 정렬
arr = [5, 7, 9, 0, 3, 1, 6, 2, 4, 8]
def quick_sort(arr, stt, end):
    if stt >= end:
        return
    pivot = stt
    l = stt + 1
    r = end
    while l <= r:
        # 기준보다 큰 데이터를 찾을 때까지 반복
        while l <= end and arr[l] <= arr[pivot]:
            l += 1
        # 기준보다 작은 데이터를 찾을 때까지 반복
        while r > stt and arr[r] >= arr[pivot]:
            r -= 1
        if l > r: # 엇갈렸다면 작은 데이터 <-> 피벗
            arr[r], arr[pivot] = arr[pivot], arr[r]
        else: # 엇갈리지 않았다면 큰 데이터 <-> 작은 데이터
            arr[l], arr[r] = arr[r], arr[l]
    quick_sort(arr, stt, r - 1)
    quick_sort(arr, r + 1, end)
quick_sort(arr, 0, len(arr) - 1) 
print("퀵 정렬 :", arr)

퀵 정렬 (리스트 컴프리헨션 이용)

# 퀵 정렬
arr = [5, 7, 9, 0, 3, 1, 6, 2, 4, 8]
def quick_sort_python(arr):
    if len(arr) <= 1:
        return arr
    pivot = arr[0]
    tail = arr[1:] # 기준값을 제외한 리스트
    
    left_side = [x for x in tail if x <= pivot]
    right_side = [x for x in tail if x > pivot]
    # 분할 이후 왼쪽 부분과 오른쪽 부분에서 각각 정렬 수행하고 전체 리스트 반환
    return quick_sort_python(left_side) + [pivot] + quick_sort_python(right_side)
print("퀵 정렬 :", quick_sort_python(arr))

계수 정렬

# 계수 정렬
print("계수 정렬 : ", end="")
arr = [7, 5, 9, 0, 3, 1, 6, 2, 9, 1, 4, 8, 0, 5, 2]
count = [0] * (max(arr) + 1)

for i in range(len(arr)):
    count[arr[i]] += 1 # 각 데이터에 해당하는 인덱스의 값 증가
for i in range(len(count)):
    for j in range(count[i]):
        print(i, end=" ")

 

두 배열의 원소 교체

[문제]

동빈이는 두 개의 배열 A와 B를 가지고 있다. 두 배열은 N개의 원소로 구성되어 있으며, 배열의 원소는 모두 자연수이다. 동빈이는 최대 K 번의 바꿔치기 연산을 수행할 수 있는데, 바꿔치기 연산이란 배열 A에 있는 원소 하나와 배열 B에 있는 원소 하나를 골라서 두 원소를 서로 바꾸는 것을 말한다.

 

동빈이의 최종 목표는 배열 A의 모든 원소의 합이 최대가 되도록 하는 것이며, 여러분은 동빈이를 도와야한다.

 

N, K, 그리고 배열 A와 B의 정보가 주어졌을 때, 최대 K 번의 바꿔치기 연산을 수행하여 만들 수 있는 배열 A의 모든 원소의 합의 최댓값을 출력하는 프로그램을 작성하라.

 

예를 들어 N = 5, K = 3이고, 배열 A와 B가 다음과 같다고 해보자

배열 A = [1, 2, 5, 4, 3]   

배열 B = [5, 5, 6, 6, 5]

 

이 경우, 다음과 같이 세 번의 연산을 수행할 수 있다

연산 1) 배열 A의 원소 '1'과 배열 B의 원소 '6'을 바꾸기

연산 2) 배열 A의 원소 '2'와 배열 B의 원소 '6'을 바꾸기

연산 3) 배열 A의 원소 '3'과 배열 B의 원소 '5'를 바꾸기

 

세 번의 연산 이후 배열 A와 배열 B의 상태는 다음과 같이 구성될 것이다

배열 A = [6, 6, 5, 4, 5]

배열 B = [3, 5, 1, 2, 5]

 

이때 배열 A의 모든 원소의 합은 26이 되며, 이보다 더 합을 크게 만들 수는 없다.

 

[입력]

첫 번째 줄에 N, K가 공백으로 구분되어 입력된다. (1 ≤ N ≤ 100,000, 0 ≤ K ≤ N) 

두 번째 줄에 배열 A의 원소들이 공백으로 구분되어 입력된다. 모든 원소는 10,000,000보다 작은 자연수이다. 

세 번째 줄에 배열 B의 원소들이 공백으로 구분되어 입력된다. 모든 원소는 10,000,000보다 작은 자연수이다.

 

입력 예시

5 3

1 2 5 4 3

5 5 6 6 5

26

 

[출력] 

최대 K번의 바꿔치기 연산을 수행하여 만들 수 있는 배열 A의 모든 원소의 합의 최댓값을 출력한다.

 

출력 예시

26

개인 풀이

n, k = map(int, input().split())
a1 = list(map(int, input().split()))
b1 = list(map(int, input().split()))

def qs(a):
    if len(a) <= 1:
        return a
    pivot = a[0]
    tail = a[1:]

    left = [i for i in tail if i <= pivot]
    right = [i for i in tail if i > pivot]

    return qs(left) + [pivot] + qs(right)

a1 = qs(a1)
b1 = qs(b1)[::-1]

for i in range(k):
    if a1[i] < b1[i]:
        a1[i], b1[i] = b1[i], a1[i]
print(sum(a1))

정답 풀이

# 정답
n, k = map(int, input().split())
a = list(map(int, input().split()))
b = list(map(int, input().split()))    

a.sort()
b.sort(reverse=True)
for i in range(k):
    if a[i] < b[i]:
        a[i], b[i] = b[i], a[i]
    else: # a의 원소가 b의 원소보다 크거나 같을 때, 반복문을 탈출
        break
print(sum(a))

 

 

 

※ 문제 출처 : 이것이 취업을 위한 코딩 테스트다 with 파이썬 (나동빈 저).

※ 개인 풀이는 오류가 있을 수 있음.