[Python/파이썬] 백준 11728번 배열 합치기

문제

정렬되어있는 두 배열 A와 B가 주어진다. 두 배열을 합친 다음 정렬해서 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 배열 A의 크기 N, 배열 B의 크기 M이 주어진다. (1 ≤ N, M ≤ 1,000,000)

둘째 줄에는 배열 A의 내용이, 셋째 줄에는 배열 B의 내용이 주어진다. 배열에 들어있는 수는 절댓값이 109보다 작거나 같은 정수이다.

출력

첫째 줄에 두 배열을 합친 후 정렬한 결과를 출력한다.

코드

• 정렬 함수 이용

n, m = map(int, input().split())
a = list(map(int, input().split()))
b = list(map(int, input().split()))

for i in sorted(a + b):
    print(i, end=' ')

• Quick Sort

n, m = map(int, input().split())
a = list(map(int, input().split()))
b = list(map(int, input().split()))

arr = a + b

def quickSort(arr):
    if len(arr) <= 1:
        return arr
    pivot, tail = arr[0], arr[1:]
    
    left = [x for x in tail if x <= pivot]
    right = [x for x in tail if x > pivot]
    
    return quickSort(left) + [pivot] + quickSort(right)

for i in quickSort(arr):
    print(i, end=' ')

• 투 포인터

n, m = map(int, input().split())
a = list(map(int, input().split()))
b = list(map(int, input().split()))

p1 = p2 = 0

while p1 < n and p2 < m:
    if a[p1] <= b[p2]:
        print(a[p1], end=' ')
        p1 += 1
    else:
        print(b[p2], end=' ')
        p2 += 1
        
if p1 < n:
    print(*a[p1:])
if p2 < m:
    print(*b[p2:])

입력으로 받은 두 배열이 이미 정렬된 배열이기 때문에 이렇게 할 수 있다.