[99클럽 코테 스터디 TIL 22일차] 349. Intersection of Two Arrays 해설 및 풀이 (Java)

LeetCode 349. Intersection of Two Arrays

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

---

Given two integer arrays nums1 and nums2, return an array of their intersection. Each element in the result must be unique and you may return the result in any order.

 

Example 1:

Input: nums1 = [1,2,2,1], nums2 = [2,2]
Output: [2]

 

Example 2:

Input: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
Output: [9,4]
Explanation: [4,9] is also accepted.
 

Constraints:

1 <= nums1.length, nums2.length <= 1000
0 <= nums1[i], nums2[i] <= 1000

---

문제 유형

Array Hash Table Two Pointers Binary Search Sorting

 

풀이 방법 도출

1. 먼저 nums1을 Set에 저장해서 중복을 제거합니다.
2. nums2를 순회하면서 nums1의 Set에 존재하는 값만 resultSet에 저장합니다.
3. 마지막으로 Set을 배열로 변환하여 결과를 반환합니다.

 

 

시간 복잡도

• O(n + m)

 

핵심 코드 삽입 및 설명

 

import java.util.*;

public class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        // Set을 사용하여 nums1의 중복을 제거하고 저장
        Set<Integer> set1 = new HashSet<>();
        for (int num : nums1) {
            set1.add(num);
        }

        // 결과를 저장할 Set
        Set<Integer> resultSet = new HashSet<>();
        for (int num : nums2) {
            // nums2의 요소가 nums1에도 있다면 결과 Set에 추가
            if (set1.contains(num)) {
                resultSet.add(num);
            }
        }

        // 결과 Set을 배열로 변환
        int[] result = new int[resultSet.size()];
        int i = 0;
        for (int num : resultSet) {
            result[i++] = num;
        }

        return result;
    }
}