COS-Pro-1차 1급 4번

2020-02-19

위 문제는 YBM COS-Pro의 1차 1급 4번 문제에 관한 설명입니다.

문제

오름차순으로 정렬되어있는 두 배열 arrA, arrB를 하나의 배열로 합치려 합니다. 단, 합친 후의 배열도 오름차순으로 정렬되어 있어야 합니다.

예를 들어 arrA = [-2, 3, 5, 9], arrB = [0, 1, 5]인 경우 두 배열을 오름차순으로 정렬된 하나의 배열로 합치면 [-2, 0, 1, 3, 5, 5, 9]가 됩니다.

오름차순으로 정렬된 두 배열 arrA와 arrB가 주어졌을 때, 두 배열을 오름차순으로 정렬된 하나의 배열로 합쳐서 return 하도록 solution 메소드를 작성하려 합니다. 빈칸을 채워 전체 코드를 완성해주세요.

입력

오름차순으로 정렬된 두 배열 arrA와 arrB가 solution 메소드의 매개변수로 주어집니다.

  • arrA의 길이는 1 이상 200,000 이하입니다.
  • arrA의 원소는 -1,000,000 이상 1,000,000 이하의 정수입니다.
  • arrB의 길이는 1 이상 200,000 이하입니다.
  • arrB의 원소는 -1,000,000 이상 1,000,000 이하의 정수입니다.

출력

두 배열 arrA, arrB를 오름차순으로 정렬된 하나의 배열로 합쳐서 return 해주세요.

예시

arrA arrB return
[-2, 3, 5, 9] [0, 1, 5] [-2, 0, 1, 3, 5, 5, 9] 

import java.util.*;

class Solution {
    public int[] solution(int[] arrA, int[] arrB) {
        int arrA_idx = 0, arrB_idx = 0;
        int arrA_len = arrA.length;
        int arrB_len = arrB.length;
        int answer[] = new int[arrA_len + arrB_len];
        int answer_idx = 0;
        while(arrA_idx < arrA_len && arrB_idx < arrB_len){
            if(arrA[arrA_idx] < arrB[arrB_idx])
                answer[answer_idx++] = arrA[arrA_idx++];
            else
                answer[answer_idx++] = arrB[arrB_idx++];
        }
        while(arrA_idx < arrA_len)
            answer[answer_idx++] = arrA[arrA_idx++];
        while(arrB_idx < arrB_len)
            answer[answer_idx++] = arrB[arrB_idx++];
        
        return answer;
    }
    
    public static void main(String[] args) {
        Solution sol = new Solution();
        int[] arrA = {-2, 3, 5, 9};
        int[] arrB = {0, 1, 5};
        int[] ret = sol.solution(arrA, arrB);
 
        System.out.println(Arrays.toString(ret));
    }
}

코드 설명

  1. arrA와 arrB의 값을 넣을 answer배열을 만들어줍니다.(이때 길이는 이 둘의 합만큼)

  2. arrA_idx < arrA_len && arrB_idx < arrB_len. 즉, 두 배열의 길이보다 index가 작다면 넣는 것을 반복합니다.

  3. 두 배열의 값들을 비교하면서 answer에 하나씩 집어넣습니다.

  4. 덜 들어간 값들에 대해서 16~19번 라인을 통해 넣습니다.