349. 两个数组的交集:
题目:
分析:
因为输出结果中的每个元素一定是唯一的,且不要求输出结果的顺序,所以我们使用 HashSet(不允许重复数据,且无序)。
我们先创建一个编译类型为 Set 的 set1,泛型设定为,用来将 nums1 中的数据取出,并存放于 set1 中(使用HashSet的add方法)。
我们有必要创建一个 set2 来存取 nums2 中的数据吗?并不需要,我们可以直接创建一个 resSet 来存取数组的交集,怎么判断是否要添加呢?HashSet中提供了contains(key)来判断是否存在该元素,若 nums2 中的元素存在于 set1 中,那么就将该元素添加到 resSet 中。
这样我们得到了编译类型为 Set 的结果集合——resSet。但是我们要返回的是一个int类型的数组。所以我们需要进行类型转换,或者将 resSet 中存储的对象存放到一个 int[] 对象中。
最后将该对象返回,就完成了这道题。
代码:
class Solution {
public int[] intersection(int[] nums1, int[] nums2) {
Set<Integer> set1 = new HashSet<>();
Set<Integer> resSet = new HashSet<>();
int[] res;
for(int i : nums1){
set1.add(i);
}
for(int i : nums2){
if(set1.contains(i))
resSet.add(i);
}
int[] resList = new int[resSet.size()];
int j = 0; // 这样子写是没有想到的
for(Integer i : resSet){
resList[j++] = i;
}
return resList;
}
}