[JavaScript/section 5] 02 - 공통원소 구하기

📌 문제

A, B 두 개의 집합이 주어지면 두 집합의 공통 원소를 추출하여 오름차순으로 출력하는 문제이다.

 

📝 풀이

🧑🏻‍💻 나의 풀이 방법

• arr1을 sort() 메서드를 사용하여 오름차순으로 정렬한다.
• arr1을 하나씩 돌면서 includes() 메서드를 사용하여 해당 값이 arr2에 존재(true)한다면 answer에 값을 삽입한다.

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

  for (let x of arr1) {
    if (arr2.includes(x)) answer.push(x);
  }

  return answer;
}

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

 

👨🏼‍🏫 강사님 풀이 방법

• arr1, arr2 두 배열을 오름차순으로 정렬한 후 투 포인터 알고리즘을 사용한다.
• arr1[p1], arr2[p2]의 값이 같다면 answer에 해당 값을 삽입하고 포인터 p1, p2 모두 1씩 증가시킨다.
• arr1[p1]보다 arr2[p2]의 값이 크다면 포인터 p1을 1 증가시킨다.
• arr1[p1] 보다 arr2[p2]의 값이 작다면 포인터 p2를 1 증가시킨다.
• 포인터 p1, p2 둘 중에 하나라도 해당 배열의 길이보다 크거나 같은 경우 비교할 값이 더 이상 없다는 것이므로 종료한다.

function solution(arr1, arr2) {
  let answer = [];
  arr1.sort((a, b) => a - b);
  arr2.sort((a, b) => a - b);
  let p1 = (p2 = 0);

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

  return answer;
}

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