11728번. 배열 합치기

 

배열 합치기

문제

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

입력

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

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

출력

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

예제 입력

2 2

3 5

2 9

예제 출력

2 3

5 9

예제 입력

2 1

4 7

1

예제 출력

1

4 7

예제 입력

4 3

2 3 5 9

1 4 7

예제 출력

1 2

3

4 5 7 9

 

#include<iostream>
#include<vector>
#include<algorithm>

using namespace std;

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	vector<int> arr1;
	vector<int> arr2;

	int N, M;
	int tmp;

	cin >> N >> M;
	for (int i = 0; i < N; i++) {
		cin >> tmp;
		arr1.push_back(tmp);
	}
	for (int i = 0; i < M; i++) {
		cin >> tmp;
		arr2.push_back(tmp);
	}

	for (int i = 0; i < M; i++) {
		arr1.push_back(arr2[i]);
	}

	sort(arr1.begin(), arr1.end());

	for (int i = 0; i < N + M; i++)
		cout << arr1[i] << " ";

	return 0;
}

배열을 합치는 문제인데, STL을 사용해보았다

아무래도 코딩테스트를 하게되면 STL로 많이 하고있고, 직접 하나하나 짜는 것보다 문제를 풀어야하기 때문에..ㅠㅠ

구현 속도 또한 중요해서 ㅋㅋ..

 

문제는 생각보다 간단하다

벡터 컨테이너 arr1, arr2를 초기화한 뒤,

arr1, arr2에 입력값을 받아 모두 넣고

arr1 뒤에 arr2를 붙인 다음, sort 함수를 통해 정렬해주고 출력하면 끗