[백준 11728번] [🥈5] 배열 합치기 (python)

백준 사이트 링크

문제

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

입력

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

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

출력

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

예제 입력 1

2 2
3 5
2 9

예제 출력 1

2 3 5 9

예제 입력 2

2 1
4 7
1

예제 출력 2

1 4 7

예제 입력 3

4 3
2 3 5 9
1 4 7

예제 출력 3

1 2 3 4 5 7 9

---

문제를 풀면서 막혔던 부분

• 두 배열을 합쳐서 정렬해주는 간단한 문제이다.

1. join 함수

join 함수는 매개변수로 들어온 리스트에 있는 요소 하나하나를 합쳐서 하나의 문자열로 바꾸어 반환하는 함수이다.

a = ['a', 'b', 'c', 'd']
print(a)

# join 이용
result1 = " ".join(a)
print(result1)

['a', 'b', 'c', 'd']
a b c d

1. unpacking (*list) 사용

인수를 순서대로 넣을 때 리스트나 튜플을 사용할 수 있는데, 리스트와 튜플 앞에 *을 붙여 인자들을 일일히 넘기는 효과와 같은 효과를 준다.

a = ['a', 'b', 'c', 'd']
print(*a)

a b c d

코드

ver(1) - join 사용

import sys

a, b = map(int, sys.stdin.readline().split())
A = list(map(int, sys.stdin.readline().split()))
B = list(map(int, sys.stdin.readline().split()))
list = A + B
answer = ' '.join(map(str, sorted(list)))
print(answer)

ver(2) - (*list) 사용

n, m = list(map(int,input().split()))
n_list = list(map(int,input().split()))
m_list = list(map(int,input().split()))

l = n_list+m_list
l.sort(reverse=False)
print(*l)