两个数组的交集
题目
给定两个数组，编写一个函数来计算它们的交集。
示例 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
思路
- 暴力遍历,可以将两个数组嵌套遍历,将重复数字存入set集合.最后转为int[]数组返回.但是这样时间复杂度会很高
- 将其中一个数组存入set集合,然后依次按另一个数组get,get到的放入结果set中.此处使用stream的和使用遍历差距很大.
代码
- 使用流式处理,耗时78ms
class Solution {
public int[] intersection(int[] nums1, int[] nums2) {
if(nums1.length == 0 || nums2.length == 0){
return new int[]{};
}
HashSet<Integer> set = new HashSet<>();
for(int i = 0;i < nums1.length;i++){
set.add(nums1[i]);
}
HashSet<Integer> resultSet = new HashSet<>();
for(int i = 0;i < nums2.length;i++){
if(set.contains(nums2[i])){
resultSet.add(nums2[i]);
}
}
return resultSet.stream().mapToInt(Number::intValue).toArray();
}
}
- 使用遍历处理.耗时7ms
class Solution {
public int[] intersection(int[] nums1, int[] nums2) {
//
if(nums1.length == 0 || nums2.length == 0){
return new int[]{};
}
HashSet<Integer> set = new HashSet<>();
for(int i = 0;i < nums1.length;i++){
set.add(nums1[i]);
}
HashSet<Integer> resultSet = new HashSet<>();
for(int i = 0;i < nums2.length;i++){
if(set.contains(nums2[i])){
resultSet.add(nums2[i]);
}
}
int[] result = new int[resultSet.size()];
int i = 0;
for (Integer integer : resultSet) {
result[i++] = integer;
}
return result;
}
}
