백준 11728번- '배열 합치기' (PYTHON 풀이)

문제:

 

두 배열 A, B에 대한 길이를 먼저 받고, 이후 순차적으로 각 배열에 속한 수를 입력받는다.

값을 입력받고 두 배열을 합친 뒤 정렬해서 출력하면 되는 문제이다.

 

풀이:

N, M = map(int,input().split())
a = list(map(int,input().split()))
b = list(map(int,input().split()))

result = a.extend(b)
result_sort = sorted(a)
print(' '.join(map(str, result_sort)))

앞서 말한대로 입력을 순차적으로 받고, 파이썬의 아주 좋은 함수인 extend를 써서 result라는 변수에 저장한다.

여기까지 하면 a, b 순서대로 a에 저장이 될 것이다. 이후에 sorted 함수를 써서 오름차순 정렬을 한 뒤 출력 예제에 맞춰 띄어쓰기로 출력하면 된다.

 

이 문제를 풀면서 주의할 점은 처음에 extend를 잘 안쓰다보니 정렬을 할 때 순간 sorted(result)로 진행했다. 근데 a에 b가 합쳐지는 것이기 때문에 우리가 원하는 합쳐진 데이터는 result가 아닌 a이므로 sotred(a) 형태로 써야한다. (result는 None이 저장된다.) 그리고 join은 문자열에 대한 규칙이므로 map과 결합해서 문자열로 바꾼 뒤 띄어쓰기로 이어붙였다.