[백준] 11728번 - 배열 합치기 [Java]

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

 

[백준] 11728번 - 배열 합치기 [Java]

[백준] 11728번 - 배열 합치기 [Java]

---

1.  아이디어

 

정렬된 배열 두 개를 합치는 경우 각 배열의 앞 원소들을 비교하며 추가하면 합쳐진 배열을 따로 정렬하지 않아도 정렬을 유지함을 이용했다.

---

2. 문제풀이

 

각 배열의 앞 원소를 비교해서 작은 원소를 새로운 배열에 넣고 작았던 원소의 다음 위치을 비교할 원소로 바꿔주는 과정을 두 배열의 모든 원소를 확인할 때까지 반복하면 된다. 구체적 구현으로는 Queue와 투 포인터 알고리즘을 활용했고 Queue는 peek 값이 배열의 앞 원소가 되면서 Queue에서 원소를 제거하면 자동으로 다음 원소를 바라보게 되어 적합하다고 생각했고, 투 포인터 알고리즘은 배열을 그대로 둔 채 포인터의 이동으로 확인을 한다는 점에서 역시 적합하다고 판단했다.

---

3. 코드

 

import java.io.*;
import java.util.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringBuilder sb = new StringBuilder();
        StringTokenizer st = new StringTokenizer(br.readLine());

        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());

        Queue<Integer> A = new ArrayDeque<>();
        st = new StringTokenizer(br.readLine());
        while (st.hasMoreTokens()) {
            A.offer(Integer.parseInt(st.nextToken()));
        }

        Queue<Integer> B = new ArrayDeque<>();
        st = new StringTokenizer(br.readLine());
        while (st.hasMoreTokens()) {
            B.offer(Integer.parseInt(st.nextToken()));
        }

        while (!(A.isEmpty() && B.isEmpty())) {
            if (A.isEmpty()) sb.append(B.poll()).append(" ");
            else if (B.isEmpty()) sb.append(A.poll()).append(" ");
            else sb.append(A.peek() < B.peek() ? A.poll() : B.poll()).append(" ");
        }

        bw.write(sb.toString());
        bw.flush();
    }
}

import java.io.*;
import java.util.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringBuilder sb = new StringBuilder();
        StringTokenizer st = new StringTokenizer(br.readLine());

        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());

        int[] A = new int[N];
        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < N; i++) {
            A[i] = Integer.parseInt(st.nextToken());
        }

        int[] B = new int[M];
        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < M; i++) {
            B[i] = Integer.parseInt(st.nextToken());
        }

        int pivotA = 0;
        int pivotB = 0;

        while (!(pivotA == N && pivotB == M)) {
            if (pivotA == N) sb.append(B[pivotB++]).append(" ");
            else if (pivotB == M) sb.append(A[pivotA++]).append(" ");
            else sb.append(A[pivotA] < B[pivotB] ? A[pivotA++] : B[pivotB++]).append(" ");
        }

        bw.write(sb.toString());
        bw.flush();
    }
}

---

4. 후기

 

- Queue

- 투 포인터 알고리즘