공통원소 구하기(효율성-투 포인터 알고리즘)

문제(출처 : 인프런 자바스크립트 알고리즘 문제풀이 강의, 정보올림피아드)

 

A, B 두 개의 집합이 주어지면 두 집합의 공통 원소를 추출하여 오름차순으로 출력하는 프로 그램을 작성하세요.

▣ 입력설명
첫 번째 줄에 집합 A의 크기 N(1<=N<=30,000)이 주어집니다.
두 번째 줄에 N개의 원소가 주어집니다. 원소가 중복되어 주어지지 않습니다. 세 번째 줄에 집합 B의 크기 M(1<=M<=30,000)이 주어집니다. 네 번째 줄에 M개의 원소가 주어집니다. 원소가 중복되어 주어지지 않습니다. 각 집합의 원소는 1,000,000,000이하의 자연수입니다.

▣ 출력설명
두 집합의 공통원소를 오름차순 정렬하여 출력합니다.

 

▣ 입력예제 1

5
13952
5

32578

 

▣ 출력예제 1

235

 

Tip

 

문제풀이

//내가 작성한 답(투 포인터를 활용)
function solution(arr1, arr2){
    let result = [];
    let p1 = p2 = 0;
    arr1.sort((a,b)=>a-b);
    arr2.sort((a,b)=>a-b);

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

let arr1 = [1,3,9,5,2];
let arr2 = [3,2,5,7,8];
console.log(solution(arr1,arr2));