2020-11-02

349. 两个数组的交集

要计算交集，就要遍历数据中的所有数，这时候想到昨天的每日一题，可以利用hash表保存数据，这样找到每一个数的时间都是O(1),同时从例子中可以看到重复的数只统计一次，可以想到利用HashSet来作为这次存储数据的结构，考虑完要使用的数组结构之后，剩下的就简单了，只需要遍历较小的那个，找到相同的数字取出来就可以了，下面附上代码：

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        Set<Integer> nums1Set = new HashSet<>();
        Set<Integer> nums2Set = new HashSet<>();
        for(int num : nums1){
        	nums1Set.add(num);
        }
        for(int num : nums2){
        	nums2Set.add(num);
        }
        return func(nums1Set, nums2Set);
    }
    int[] func(Set<Integer> nums1Set, Set<Integer> nums2Set){
    	//判断哪个较小，遍历较小的那个
    	if(nums1Set.size() > nums2Set.size()){
    		return func(nums2Set, nums1Set);
    	}
    	Set<Integer> tmp = new HashSet<>();
    	for(int num : nums1Set){
    		if(nums2Set.contains(num)){
    			tmp.add(num);
    		}
    	}
    	int[] res = new int[tmp.size()];
    	int i = 0;
    	for(int num : tmp){
    		res[i] = num;
    		i++;
    	}
    	return res;
    }
}