[COS PRO 1급] 1-7 병합 and 정렬 자바(Java)

문제링크

구름HOME

 

문제유형

빈칸 채우기

 

문제

import java.util.*;
 
class Main{
    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(________){
            if(arrA[arrA_idx] < arrB[arrB_idx])
                answer[answer_idx++] = arrA[arrA_idx++];
            else
                answer[answer_idx++] = arrB[arrB_idx++];
        }
        while(________)
            answer[answer_idx++] = arrA[arrA_idx++];
        while(________)
            answer[answer_idx++] = arrB[arrB_idx++];
        return answer;
    }
 
public static void main(String[] args) {
        Main sol = new Main();
        int[] arrA = {-2, 3, 5, 9};
        int[] arrB = {0, 1, 5};
        int[] ret = sol.solution(arrA, arrB);
 
        System.out.println("solution 함수의 반환 값은 " + Arrays.toString(ret) + " 입니다.");
    }
}

 

문제 풀이

import java.util.*;
 
class Main {
    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) {
        Main sol = new Main();
        int[] arrA = {-2, 3, 5, 9};
        int[] arrB = {0, 1, 5};
        int[] ret = sol.solution(arrA, arrB);
 
        System.out.println("solution 함수의 반환 값은 " + Arrays.toString(ret) + " 입니다.");
    }
}