[Leetcode] 349. Intersection of Two Arrays

🔗 문제 링크

https://leetcode.com/problems/intersection-of-two-arrays/description/

Intersection of Two Arrays - LeetCode

 

✅ 코드

import java.io.*;
import java.util.*;

class Solution {
    static Set<Integer> set;
    public int[] intersection(int[] nums1, int[] nums2) {
        set = new HashSet<>();

        int n = nums1.length;
        int m = nums2.length;

        Arrays.sort(nums1);
        Arrays.sort(nums2);

        if(n >= m){
            for(int i = 0; i < m ; i++){
                if(Arrays.binarySearch(nums1,nums2[i]) > -1){
                    set.add(nums2[i]);
                }
            }
        }else{
            for(int i = 0 ; i < n ; i++){
                if(Arrays.binarySearch(nums2,nums1[i]) > -1){
                    set.add(nums1[i]);
                }
            }
        }

        int[] answer = set.stream()
                  .mapToInt(Integer::intValue)
                  .toArray();

        return answer; 
    }
}

💡 배운 점 & 느낀 점

1. Arrays.binarySearch → 배열에서 이진탐색을 하기 위해서는 반드시 먼저 정렬(Arrays.sort())를 진행해줘야 한다