[LeetCode] 349. Intersection of Two Arrays

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

两个数组的交集。

版本二350题。题目即是题意。

这道题有三种写法，1. 两个hashset；2. 双指针；3. 二分法。

首先是两个hashset的思路。用第一个hashset记录num1中的所有元素，然后遍历num2，看num2中每个元素是否在第一个hashset里面出现过，若出现过，则加入第二个hashset，最后用数组返回hashset所有的key。

时间O(n)

空间O(n)

Java实现

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        // corner case
        if (nums1 == null || nums1.length == 0 || nums2 == null || nums2.length == 0) {
            return new int[] {};
        }

        // normal case
        HashSet<Integer> set1 = new HashSet<>();
        HashSet<Integer> set2 = new HashSet<>();
        for (Integer num : nums1) {
            set1.add(num);
        }
        for (Integer num : nums2) {
            if (set1.contains(num)) {
                set2.add(num);
            }
        }
        int i = 0;
        int[] res = new int[set2.size()];
        for (Integer element : set2) {
            res[i++] = element;
        }
        return res;
    }
}

JavaScript实现

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number[]}
 */
var intersection = function(nums1, nums2) {
    let set = new Set(nums1);
    let res = new Set();
    for (let i = 0; i < nums2.length; i++) {
        if (set.has(nums2[i])) {
            res.add(nums2[i]);
        }
    }
    return Array.from(res);
};

再来是双指针的思路。也是需要用到一个hashset，因为要求输出unique的结果集。首先sort两个数组，接着用双指针分别遍历两个数组，看是否能找到一样的元素，加入hashset，最后返回的也是hashset的所有的key。

时间O(nlogn) - 因为有sort

空间O(n)

Java实现

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        Set<Integer> set = new HashSet<>();
        Arrays.sort(nums1);
        Arrays.sort(nums2);
        int i = 0;
        int j = 0;
        while (i < nums1.length && j < nums2.length) {
            if (nums1[i] < nums2[j]) {
                i++;
            } else if (nums1[i] > nums2[j]) {
                j++;
            } else {
                set.add(nums1[i]);
                i++;
                j++;
            }
        }
        int[] result = new int[set.size()];
        int k = 0;
        for (Integer num : set) {
            result[k++] = num;
        }
        return result;
    }
}

最后是二分法。先sort num2，对于num1中的每个元素，用二分法去查找是否存在于num2。

时间O(nlogn) - sort

空间O(n)

Java实现

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        Set<Integer> set = new HashSet<>();
        Arrays.sort(nums2);
        for (int num : nums1) {
            if (helper(nums2, num)) {
                set.add(num);
            }
        }
        int i = 0;
        int[] res = new int[set.size()];
        for (int num : set) {
            res[i++] = num;
        }
        return res;
    }

    private boolean helper(int[] nums, int target) {
        int low = 0;
        int high = nums.length - 1;
        while (low <= high) {
            int mid = low + (high - low) / 2;
            if (nums[mid] == target) {
                return true;
            }
            if (nums[mid] > target) {
                high = mid - 1;
            } else {
                low = mid + 1;
            }
        }
        return false;
    }
}

LeetCode 题目总结