1 题目描述
给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2]
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [9,4]
说明:
输出结果中的每个元素一定是唯一的。
我们可以不考虑输出结果的顺序。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/intersection-of-two-arrays
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2 解题思路
看到这道题的想法就是 设置两个去重之后的set,然后短的那个去长的那个里面查找,
看到题解之后 其实是不太明白的 题解的思路我也没有看太懂
然后翻了一下评论区, 很喜欢这个做法 觉得比较是个小白上手
假设长度小的数组为 less, 长度为 M;长度大的数组为 more, 长度为 N
使用长度小的数组建立哈希表 lessSet,用于匹配查找,时间复杂度为 O(M), 最大额外空间复杂度为 O(M).
循环长度大的数组,通过哈希表 lessSet 匹配与长度小的数组的交集,查找时间复杂度为 O(1),总的时间复杂度为 O(N).
总体的时间复杂度为 O(M+N), 额外空间复杂度为 O(M+N).
3 解决代码
class Solution {
public int[] intersection(int[] nums1, int[] nums2) {
int len1 = nums1.length;
int len2 = nums2.length;
int[] less = len1 < len2 ? nums1:nums2;
int[] more = len1 < len2 ? nums2:nums1;
//长度小的建立哈希表,用于匹配查找
HashSet<Integer> lessSet = new HashSet<Integer>();
for(int i = 0,len = less.length;i<len;i++){
lessSet.add(less[i]);
}
//长度大的数组进行查找,与哈希表进行匹配
HashSet<Integer> moreSet = new HashSet<Integer>();
for(int i = 0,len = more.length;i<len;i++){
if(!moreSet.contains(more[i]) && lessSet.contains(more[i])){
moreSet.add(more[i]);
}
}
//Set 转为 数组
int size = moreSet.size(), k = 0;
int[] ans = new int[size];
for(int n:moreSet){
ans[k++] = n;
}
return ans;
}
}