Python - 두 리스트 합치기

문제 : 

오름차순으로 정렬이 된 두 리스트가 주어지면 두 리스트를 오름차순으로 합쳐 출력하는 프로
그램을 작성하세요.

 

입력

첫 번째 줄에 첫 번째 리스트의 크기 N(1<=N<=100)이 주어집니다.
두 번째 줄에 N개의 리스트 원소가 오름차순으로 주어집니다.
세 번째 줄에 두 번째 리스트의 크기 M(1<=M<=100)이 주어집니다.
네 번째 줄에 M개의 리스트 원소가 오름차순으로 주어집니다.
각 리스트의 원소는 int형 변수의 크기를 넘지 않습니다.

 

출력

오름차순으로 정렬된 리스트를 출력합니다.

입력 예시

3
1 3 5
5
2 3 6 7 9

출력 예시

1 2 3 3 5 6 7 9

 

 

코드 : 

# 두 리스트 합치기

# 오름차순으로 정렬 된 두 리스트가 주어지면 두 리스트를 오름차순으로 합쳐 출력하는 프로그램을 작성

import sys

def input_arr(num):
    
    li = list(map(int, sys.stdin.readline().split()))

    if len(li) > num:
        print('Error')
    else:
        return li

N = int(sys.stdin.readline())
arr1 = input_arr(N)
M = int(sys.stdin.readline())
arr2 = input_arr(M)

arr1.extend(arr2)
print(sorted(arr1))