[Baekjoon/C++] 11728번 - 배열 합치기

[백준] Baekjoon Online Judge

문제로 이동

 

문제

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

 

입력

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

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

 

출력

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

 

---

풀이

#include <iostream>
using namespace std;

#define SIZE 1000000

int N, M;
int A[SIZE], B[SIZE];

int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
    
    	// 입력
	cin >> N >> M;
	for (int i = 0; i < N; i++) cin >> A[i];
	for (int i = 0; i < M; i++) cin >> B[i];

	int a = 0, b = 0; // 포인터

	while (true) {
		if (a >= N || b >= M) break;

		if (A[a] < B[b]) cout << A[a++] << ' ';
		else cout << B[b++] << ' ';
	}
    
    	// 나머지 출력하기
	for (int i = a; i < N; i++) cout << A[i] << ' ';
	for (int i = b; i < M; i++) cout << B[i] << ' ';

	return 0;
}

 포인터를 사용하여 문제를 풀었다. 각 포인터가 가르키는 칸의 숫자 중 더 작은 것을 출력하고, 포인터를 이동시키는 방식으로 문제의 답을 출력하도록 했다. whil문이 끝났음에도 포인터가 맨 뒤를 가르키지 않을 수 있기에 나머지를 출력하는 코드 또한 작성하였다.