【Lintcode】547. Intersection of Two Arrays

题目地址：

https://www.lintcode.com/problem/intersection-of-two-arrays/description

给定两个数组，求其交集。要求交集去重。

可以用一个哈希表存储较短的数组中的所有数，然后遍历较长的那个数组，看每个数是否在哈希表里，如果在，就将其加入另一个哈希表里，这个哈希表就存储所有的交集的数。代码如下：

import java.util.HashSet;
import java.util.Set;

public class Solution {
    /**
     * @param nums1: an integer array
     * @param nums2: an integer array
     * @return: an integer array
     */
    public int[] intersection(int[] nums1, int[] nums2) {
        // write your code here
        Set<Integer> set = new HashSet<>();
        int[] shorter = nums1.length <= nums2.length ? nums1 : nums2, longer = nums1 == shorter ? nums2 : nums1;
        for (int num : shorter) {
            set.add(num);
        }
        Set<Integer> intersection = new HashSet<>();
        for (int num : longer) {
            if (set.contains(num)) {
                intersection.add(num);
            }
        }
        
        int[] res = new int[intersection.size()];
        int ind = 0;
        for (int num : intersection) {
            res[ind++] = num;
        }
        
        return res;
    }
}

时间复杂度$O(n+m)$，空间$O(\min (m,n))$。