题目描述
给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2,2]
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]
说明:
输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
我们可以不考虑输出结果的顺序。
进阶:
如果给定的数组已经排好序呢?你将如何优化你的算法?
如果 nums1 的大小比 nums2 小很多,哪种方法更优?
如果 nums2 的元素存储在磁盘上,磁盘内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?
解题思路
既然是求两数组的重复元素,我们可以借助hashmap来充当一个计数器,记录元素较多的那个数组的各个元素出现频次,然后遍历另外一个数组,如果该数组的元素在hashmap中出现频次大于0,证明有重复元素,将其插入到第二个新的数组中去,并将hashmap中元素value减1!!!
代码实现
去博客设置页面,选择一款你喜欢的代码片高亮样式,下面展示同样高亮的 代码片
.
import java.util.Arrays;
import java.util.HashMap;
public class arrayIntersect {//acc:5ms 48%
//两个数组的交集
public int[] intersect(int[] nums1,int[] nums2){
if (nums1.length>nums2.length){
return intersect(nums2,nums1);
}
HashMap<Integer,Integer> map = new HashMap<>();
for (int n:nums1){
map.put(n,map.getOrDefault(n,0)+1);
}
int k=0;//计数器,记录重复的个数
for (int n:nums2){
int cnt=map.getOrDefault(n,0);
if (cnt>0){
nums1[k++]=n;
map.put(n,cnt-1);
}
}
return Arrays.copyOfRange(nums1,0,k);
}
public static void main(String[] args) {
arrayIntersect test=new arrayIntersect();
int[] nums1={1,2,2,1};
int[] nums2={2,2};
int[] res=test.intersect(nums1,nums2);
System.out.println(res[0]+","+res[1]);
}
}
总结
本题来源于Leetcode中 归属于排序类型题目。
同许多在算法道路上不断前行的人一样,不断练习,修炼自己!
如有博客中存在的疑问或者建议,可以在下方留言一起交流,感谢各位!
觉得本博客有用的客官,可以给个赞鼓励下! 嘿嘿