题目
给定两个数组,编写一个函数来计算它们的交集。
示例 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 的元素存储在磁盘上,内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?
解法
方法一: 把两个数组放进map中,利用哈希表性质。
class Solution {
/**
方法一:把两个数组放进map中,利用哈希表性质。
*/
public int[] intersect(int[] nums1, int[] nums2) {
if(nums1 == null || nums2 == null){
return null;
}
Map<Integer,Integer> map1 = new HashMap<>();
for(int n : nums1){
map1.put(n,map1.get(n) == null ? 1 : map1.get(n)+1);
}
Map<Integer,Integer> map2 = new HashMap<>();
for(int n : nums2){
map2.put(n,map2.get(n) == null ? 1 : map2.get(n)+1);
}
List<Integer> list = new ArrayList<>();
for(Map.Entry<Integer,Integer> entry : map1.entrySet()){
Integer key = entry.getKey();
Integer value = entry.getValue();
Integer value2 = map2.get(key);
if(value2 != null){
int n = Math.min(value,value2);
for(int i = 0 ; i < n ; i++){
list.add(key);
}
}
}
int[] res = new int[list.size()];
int index = 0;
for(int n : list){
res[index++] = n;
}
return res;
}
}
方法二: 排序之后使用双指针
class Solution {
// 方法二:排序之后使用双指针。
public int[] intersect(int[] nums1, int[] nums2) {
if(nums1 == null || nums2 == null){
return null;
}
Arrays.sort(nums1);
Arrays.sort(nums2);
int i = 0 , j = 0;
List<Integer> list = new ArrayList<>();
while(i < nums1.length && j < nums2.length){
if(nums1[i] == nums2[j]){
list.add(nums1[i]);
i++;
j++;
} else if (nums1[i] < nums2[j]){
i++;
} else {
j++;
}
}
int[] res = new int[list.size()];
int index = 0 ;
for(int n : list){
res[index++] = n;
}
return res;
}
}
总结
本篇文章讲解了算法题目的思路和解法,代码和笔记由于纯手打,难免会有纰漏,如果发现错误的地方,请第一时间告诉我,这将是我进步的一个很重要的环节。以后会定期更新算法题目以及各种开发知识点,如果您觉得写得不错,不妨点个关注,谢谢。