BOJ 11728 - 배열 합치기 / C++

문제 링크 : https://www.acmicpc.net/problem/11728

11728번: 배열 합치기

 

 

풀이

주어지는 두 배열은 이미 정렬되어 있는 상태이므로 각 배열의 0번째 인덱스부터 비교하여 순서대로 출력하면 된다.

한 포인터라도 크기를 벗어나면 while문이 종료되므로 나머지 포인터도 증가시키며 값들을 출력하면 된다.

 

코드

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

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

    int n, m;
    vector<int> v1, v2;

    cin >> n >> m;
    for(int i=0; i<n; i++){
        int tmp;
        cin >> tmp;
        v1.push_back(tmp);
    }
    for(int i=0; i<m; i++){
        int tmp;
        cin >> tmp;
        v2.push_back(tmp);
    }

    int p1 = 0, p2 = 0;
    
    while(p1 < v1.size() && p2 < v2.size()){
        if(v1[p1] <= v2[p2]) {
            cout << v1[p1++] << ' ';
        }
        else {
            cout << v2[p2++] << ' ';
        }
    }

    while(p1 < v1.size()){
        cout << v1[p1++] << ' ';
    }
    while(p2 < v2.size()){
        cout << v2[p2++] << ' ';
    }
}