题目
给定两个数组 nums1和nums2 ,返回它们的交集 。输出结果中的每个元素一定是唯一的。我们可以不考虑输出结果的顺序 。
思路
由于是返回交集,且题目强调输出结果中的元素要唯一,所以使用set集合是很好的一个方法。如果直接遍历数组进行查找,时间复杂度会更高,如果使用哈希表存储元素,可以在O(1)的时间内判断元素是否在集合中,可以降低时间复杂度。
Set集合特点:
Set集合是一个存储元素不能重复的集合方式,它无序且唯一,没有下标。
首先可以先遍历数组nums1,将其加入进第一个set集合,而后再遍历nums2,进行筛选,如果有相同的元素,再加入另一个set集合中。
class Solution {
public int[] intersection(int[] nums1, int[] nums2) {
if(nums1.length==0||nums2.length==0){
return new int[0];
}
Set<Integer> set1=new HashSet<>();
Set<Integer> set2=new HashSet<>();
for(int ele:nums1){
set1.add(ele);
}
for(int ele:nums2){
if(set1.contains(ele)){
set2.add(ele);
}
}
int[] arr=new int[set2.size()];
int num=0;
for(int i:set2){
arr[num++]=i;
}
return arr;
}
}
有一点需要注意,在new set集合时,需要加泛型,如果不加泛型,set是Object类型,在set转为数组加入元素时,会出现错误,因为对象类型无法转换为int,所以要给set加上Integer泛型,这样能自动拆箱为int类型。