[인프런 C++] 39. 두 배열 합치기

🍧 문제

오름차순으로 정렬이 된 두 배열이 주어지면 두 배열을 오름차순으로 합쳐 출력하는 프로그램을 작성하세요.

 

🍧 입력

첫 번째 줄에는 첫 번째 배열의 크기 N이 주어진다.

두 번째 줄에 N개의 배열 원소가 오름차순으로 주어진다.

세 번째 줄에는 두 번째 배열의 크기 M이  주어진다.

네 번째 줄에 M개의 배열 원소가 오름차순으로 주어진다.

각 배열의 원소는 int형 변수의 크기를 넘지 않는다.

 

🍧 출력

오름차순으로 정렬된 배열을 출력한다.

 

🍧 1트 (강의 풀이)

a 배열과 b 배열을 두고, 각 배열의 값들을 비교하여 c 배열에 삽입한다.

한쪽이 끝나면 중지하고, 나머지를 c 배열에 넣어서 구현한다.

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int a[101], b[101], c[300];
int main(void){
	freopen("input.txt", "rt", stdin);
	
	int n, m, p1 = 1, p2 = 1, p3 = 1;
	cin >> n;
	for(int i = 1; i <= n; ++i){
		cin >> a[i];
	}
	
	cin >> m;
	for(int i = 1; i <= m; ++i){
		cin >> b[i];
	}
	
	while(p1 <= n && p2 <= m){
		if(a[p1] < b[p2]){
			c[p3++] = a[p1++];
		}
		else{
			c[p3++] = b[p2++];
		}
	}
	
	while(p1 <= n) c[p3++] = a[p1++];
	while(p2 <= m) c[p3++] = b[p2++];
	
	for(int i = 1; i < p3; ++i){
		cout << c[i] << ' ';
	}
	
	return 0;
}