Intersection of Two Arrays（交集）

来源：https://leetcode.com/problems/intersection-of-two-arrays

Given two arrays, write a function to compute their intersection.

Example:
Given nums1 = [1, 2, 2, 1], nums2 = [2, 2], return [2].

Note:

• Each element in the result must be unique.
• The result can be in any order.

1. 直接使用HashSet

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        Set<Integer> set = new HashSet<>();
        Set<Integer> intersect = new HashSet<>();
        for(int i=0; i<nums1.length; i++) {
            set.add(nums1[i]);
        }
        for(int i=0; i<nums2.length; i++) {
            if(set.contains(nums2[i])) {
                intersect.add(nums2[i]);
            }
        }
        int[] result = new int[intersect.size()];
        int i = 0;
        for(Integer num: intersect) {
            result[i++] = num;
        }
        return result;
    }
}// 6 ms

2. 类似BitMap的思想，LeetCode 1ms sample，缺点是数组中的元素不能为负数……

（1）求出两个数组的最大值

（2）建立 长度=最大值+1 的数组A，假设数组的索引为i，那么A[i]的值表示i是否在两个数组中存在，若为0表示不存在于任何一个数组中，为1表示存在于第一个数组中，为2表示两个数组中都存在

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        int max = 0;
        for(int num: nums1) {
            if(num > max) {
                max = num;
            }
        }
        for(int num: nums2) {
            if(num > max) {
                max = num;
            }
        }
        int[] indexMap = new int[max+1];
        for(int num: nums1) {
            indexMap[num] = 1;
        }
        int cnt = 0;
        for(int num: nums2) {
            if(indexMap[num] == 1) {
                indexMap[num] = 2;
                cnt++;
            }
        }
        int[] result = new int[cnt];
        for(int i=0; i<max+1; i++) {
            if(indexMap[i] == 2) {
                result[--cnt] = i;
            }
        }
        return result;
    }
}