[알고리즘 문제] 39. 두 배열 합치기

코드

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

using namespace std;

int main(void)
{
	int n, m, i, j;
	
	cin >> n;
	vector<int> vN(n);
	
	for (i = 0; i < n; ++i)
		cin >> vN[i];

	cin >> m;
	vector<int> vM(m);

	vector<int> vAnswer;
	vAnswer.resize(n + m);
	
	for (i = 0; i < m; ++i)
		cin >> vM[i];

	i = j = 0;

	while (true)
	{
		if (vN[i] > vM[j])
		{
			vAnswer[i + j] = vM[j];
			++j;
		}
		else if (vN[i] < vM[j])
		{
			vAnswer[i + j] = vN[i];
			++i;
		}
		else
		{
			// 같은 경우
			vAnswer[i + j] = vN[i];
			vAnswer[i + j + 1] = vM[j];
			++i; ++j;
		}

		if (i == n)
		{
			for (; j < m; ++j) vAnswer[i + j] = vM[j];
			break;
		}
		else if (j == m)
		{
			for (; i < n; ++i) vAnswer[i + j] = vN[i];
			break;
		}
	}

	for (i = 0; i < vAnswer.size(); ++i)
		cout << vAnswer[i] << " ";

	return 0;
}

 

출력 결과

 

채점 결과

 

 

해당 포스트는 'it 취업을 위한 알고리즘 문제풀이 (with C/C++) : 코딩테스트 대비' 강의를 수강하며 개인 백업용으로 메모하였습니다.

인프런: https://www.inflearn.com/course/%EC%95%8C%EA%B3%A0%EB%A6%AC%EC%A6%98#

it 취업을 위한 알고리즘 문제풀이 (with C/C++) : 코딩테스트 대비 - 인프런