두 배열 합치기

문제 정보

• 사이트 : Infrean 자바 알고리즘 문제 풀이 강의
• 문제 번호 : 3강 1번
• 문제 분류 : Two Pointer

문제 풀이

내가 짠 코드

• 각각의 인덱스를 두고 두 배열의 인덱스를 비교하여 작은 것의 값을 ArrayList에 추가하고 index를 증가시켰다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;

public class Main {

    public static void sortTwoArray(int sSize, int bSize, int[] sArr, int[] bArr, ArrayList<Integer> result){
        int sIndex = 0;
        int bIndex = 0;
        while(sIndex < sSize){
            if(bArr[bIndex] < sArr[sIndex]){
                result.add(bArr[bIndex++]);
            } else {
                result.add(sArr[sIndex++]);
            }
        }
        for (int i = bIndex; i < bSize; i++) {
            result.add(bArr[i]);
        }
    }

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        int[] nArr = new int[n];
        ArrayList<Integer> result = new ArrayList<>();
        String[] input = br.readLine().split(" ");
        for (int i = 0; i < n; i++) {
            nArr[i] = Integer.parseInt(input[i]);
        }
        int m = Integer.parseInt(br.readLine());
        int[] mArr = new int[m];
        input = br.readLine().split(" ");
        for (int i = 0; i < m; i++) {
            mArr[i] = Integer.parseInt(input[i]);
        }

        if(n >= m){
            sortTwoArray(m, n, mArr, nArr, result);
        } else{
            sortTwoArray(n, m, nArr, mArr, result);
        }

        for (Integer integer : result) {
            System.out.print(integer + " ");
        }

        br.close();
    }
}

문제 출처

• Infrean 자바 알고리즘 문제 풀이

썸네일 출처

Image by storyset on Freepik