far

[JS 알고리즘] 두 배열 합치기 (투포인터) 본문

알고리즘/알고리즘 베이스

[JS 알고리즘] 두 배열 합치기 (투포인터)

Eater 2023. 3. 27. 00:29

두 배열 합치기

오름차순으로 정렬이 된 두 배열이 주어지면 두 배열을 오름차순으로 합쳐 출력하는 프로그램을 작성하세요.

 

입력설명
첫 번째 줄에 첫 번째 배열의 크기 N(1<=N<=100)이 주어집니다.
두 번째 줄에 N개의 배열 원소가 오름차순으로 주어집니다. 
세 번째 줄에 두 번째 배열의 크기 M(1<=M<=100)이 주어집니다.
네 번째 줄에 M개의 배열 원소가 오름차순으로 주어집니다. 
각 리스트의 원소는 int형 변수의 크기를 넘지 않습니다.

출력설명
오름차순으로 정렬된 배열을 출력합니다.

입력예제
1 3 5
2 3 6 7 9

출력예제
1 2 3 3 5 6 7 9

 

// 나의 풀이

function solution(arr1, arr2){
    // 1번 방법
    const answer = [...arr1, ...arr2].sort((a, b) => a - b) ;

    // 2번 방법
    const answer = arr1.concat(arr2).sort((a, b) => a - b)

    // 3번 방법 (Two Pointers 기초)
    let answer = []
    let p1 = 0
    let p2 = 0

    while(p1 < arr1.length && p2 < arr2.length) {
        if (arr1[p1] <= arr2[p2]) {
            answer.push(arr1[p1])
            p1++
        } else {
            answer.push(arr2[p2])
            p2++
        }
    }
    while (p2 < arr2.length) {
        answer.push(arr2[p2])
        p2++
    }

    while (p1 < arr1.length) {
        answer.push(arr1[p1])
        p1++
    }

    return answer;
}
            
let a=[1, 3, 5];
let b=[2, 3, 6, 7, 9];
console.log(solution(a, b));

처음에 투 포인터 알고리즘이 뭔지 몰라서 1번과 2번 방식으로 풀었다가 설명을 듣고 3번 방식으로 풀었다. 

 

투 포인터 알고리즘이란 리스트에 순차적으로 접근해야 할 때 두 점의 위치를 기록하면서 처리하는 알고리즘이다.

두개의 점(p1, p2)을 준비해서 조건에 맞으면 +1씩 배열을 옮기면 되는 심플한 문제였다.

Comments