백준 11728번 파이썬 풀이: 배열 합치기

백준 11728번 배열 합치기

 

알고리즘 분류: 정렬

 

링크: www.acmicpc.net/problem/11728

 

11728번: 배열 합치기

문제

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

입력

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

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

출력

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

 

 

문제 파악

 

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

 

풀이를 따로 할 필요가없을 정도로 간단해보이지만 조금 코드를 짧고 간단하게 쓰는 방법을

 

알게되어서 포스팅을 해놓으려 한다.

 

 

코드

# 11728번

import sys

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

일단 혹시 시간 초과가 나올 수 있으니 sys를 import 해줘서

 

input 대신에 sys.stdin.readline을 사용해준다.

 

a_list와 b_list를 입력받은 다음에 더해서 answer_list를 만들어준다.

 

answer_list를 문자열로 바꿔줄 때 리스트 안에있는 요소들이 int형이라서

 

반복문을 통해 str형으로 바꿔서 join을 사용해야된다고 생각했는데

 

map을 통해서 그럴필요없이 바로 str형으로 바꿔서 join 해줄수있었다.