JAVA解法
class Solution {
public int[] intersection(int[] nums1, int[] nums2) {
// 用于保存 nums1 的 HashSet
Set<Integer> set1 = new HashSet<>();
// 用于存储相交元素的 HashSet
Set<Integer> set2 = new HashSet<>();
// 遍历 nums1 将其存到 set1 中
for(int i:nums1){
set1.add(i);
}
// 若 nums2 中存在 set1 中的元素,则加入 set2
for(int i:nums2){
if(set1.contains(i)){
set2.add(i);
}
}
// 开辟一个长度为 set2.size() 的存放相交元素的数组
int[] arr = new int[set2.size()];
int j = 0;
// 把 set2 里的值存到 arr 数组中
for(int i:set2){
arr[j++] = i;
}
return arr;
}
}
题解分析
首先想到的是用 Set 来去重存储传进来的数组 nums1,所以遍历 nums1 将所有数组存进去。接下来,遍历 nums2 并加个判断,将与 set1 相同的元素储存到 set2,最后把 set2 存到数组里并返回。
leetcode原题: 349. 两个数组的交集
评论区