알고리즘/알고리즘 베이스
[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씩 배열을 옮기면 되는 심플한 문제였다.