[백준/BOJ] 11728번 배열 합치기 (C++)

www.acmicpc.net/problem/11728

11728번: 배열 합치기

병합 정렬의 근본이 되는 문제였습니다. 해당 문제를 풀고나서 2751번 문제를 풀어보면 좋습니다.

 

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

int main(void) {
	ios_base::sync_with_stdio(false);
	cin.tie(0);
	
	int n, m;
	cin >> n >> m;
	vector<int> v1(n), v2(m), answer;
	
	for(int i = 0; i < n; i++) {
		cin >> v1[i];
	} 
	
	for(int i = 0; i < m; i++) {
		cin >> v2[i];
	} 
	
	int p1 = 0, p2 = 0;
	
	while(p1 < n && p2 < m) {
		if(v1[p1] <= v2[p2]) answer.push_back(v1[p1++]);
		else answer.push_back(v2[p2++]);
	}
	
	// v1 벡터 원소가 남은 경우 
	while(p1 < n) {
		answer.push_back(v1[p1++]);
	}
	
	// v2 벡터 원소가 남은 경우 
	while(p2 < m) {
		answer.push_back(v2[p2++]);
	}
	
	for(int i = 0; i < m + n; i++) {
		cout << answer[i] << " ";
	}

}