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

11728번 : 배열 합치기

---

문제)

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

 

 

 

입력 :

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

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

 

 

 

출력 :

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

 

 

 

 

풀이)

 

// 11728. 배열 합치기
#include <iostream>
#include <vector>
 
using namespace std;
 
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(0);
 
    int n, m;
    cin >> n >> m;
 
    // 입력 받기
    vector<long long> A(n, 0);
    vector<long long> B(m, 0);
    for (long long& ele : A) cin >> ele;
    for (long long& ele : B) cin >> ele;
 
    vector<long long> answer;
 
    // 배열A를 탐색할 포인터
    int s1 = 0;
    // 배열 B를 탐색할 포인터
    int s2 = 0;
 
    // 두 포인터가 배열의 끝에 도달할 때까지
    while (s1 < n && s2 < m)
    {
        long long numA = A[s1];
        long long numB = B[s2];
 
        // numA와 numB의 크기를 비교해가며 포인터를 이동시킨다.
        if (numA < numB)
        {
            answer.push_back(numA);
            s1++;
        }
        else if (numA > numB)
        {
            answer.push_back(numB);
            s2++;
        }
        else if (numA == numB)
        {
            answer.push_back(numA);
            s1++;
        }
    }
 
    // s1가 A배열의 끝에 도달하지 않았다면 
    // 끝에 도달할때까지 값을 넣어가며 포인터를 이동시킨다.
    if (s1 != n)
    {
        while (s1 < n)
        {
            answer.push_back(A[s1]);
            s1++;
        }
    }
    // s2가 B배열의 끝에 도달하지 않았다면 
    // 끝에 도달할때까지 값을 넣어가며 포인터를 이동시킨다.
    else if (s2 != m)
    {
        while (s2 < m)
        {
            answer.push_back(B[s2]);
            s2++;
        }
    }
 
    for (long long ele: answer) cout << ele << " ";
    
    return 0;
}
 
 

---

출처 : https://www.acmicpc.net/problem/11728

11728번: 배열 합치기