刷刷刷Day6| 349. 两个数组的交集
349. 两个数组的交集
LeetCode题目要求
给定两个数组 nums1 和 nums2 ,返回 它们的交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。
示例:
输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2]
解题思路
本题借助于 Set 来处理元素,因为 Set 存储的元素是不能重复的。
- 首先遍历 nums1,将所有不重复的元素放入 set
- 然后遍历 nums2,通过判断 set 中是否存在 nums2 中的数值,如果存在就认为有交集,并把这个值放入到 result 中。
- 最后由于 result 是个 set 类型,将其转换为数组
上代码:
class Solution {
public int[] intersection(int[] nums1, int[] nums2) {
Set<Integer> result = new HashSet<>();
Set<Integer> set = new HashSet<>();
for (int i = 0; i < nums1.length; i++) {
set.add(nums1[i]);
}
for (int i = 0; i < nums2.length; i++) {
if (set.contains(nums2[i])) {
result.add(nums2[i]);
}
}
int[] rr = new int[result.size()];
int j = 0;
for (Iterator it = result.iterator(); it.hasNext();) {
rr[j++] = (Integer)it.next();
}
return rr;
}
}
重难点
附:学习资料链接