题目:https://leetcode-cn.com/explore/interview/card/top-interview-questions-easy/1/array/26/
题目描述:
给定两个数组,编写一个函数来计算它们的交集。
示例 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来存放其中一个数组,并且利用键值对关系表示某个元素和其出现的次数。第二种方法,排序完的数组利用双指针(下标)进行比较操作,有以下情况:
1.num[i]=num[j] : 将num[i]传入临时数组中。i++,j++(两个下标都向右移动),再比较。
2.num[i]<num[j] : 较小的数组下标向右移动,i++ ,再比较。
3.num[i]>num[j] : 较小的数组下标向右移动 ,j++ ,再比较。
//解法1
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
List<Integer> tmp = new ArrayList<>();
Map<Integer, Integer> map = new HashMap<Integer, Integer>();
for (int i = 0; i < nums1.length; i++) {
Integer value = map.get(nums1[i]);
map.put(nums1[i], (value == null ? 0 : value) + 1);
}
for (int i = 0; i < nums2.length; i++) {
if (map.containsKey(nums2[i]) && map.get(nums2[i]) != 0) {
tmp.add(nums2[i]);
map.put(nums2[i], map.get(nums2[i]) - 1);
}
}
int[] result = new int[tmp.size()];
int i = 0;
for (Integer e : tmp)
result[i++] = e;
return result;
}
}
// 解法2
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
Arrays.sort(nums1);
Arrays.sort(nums2);
List<Integer> tmp = new ArrayList<>();
int i = 0;
int j = 0;
while (i < nums1.length && j < nums2.length) {
if (nums2[j] > nums1[i]) {
i++;
} else if (nums2[j] < nums1[i]) {
j++;
} else {
tmp.add(nums1[i]);
i++;
j++;
}
}
int[] result = new int[tmp.size()];
for (int k = 0; k < result.length; k++) {
result[k] = tmp.get(k);
}
return result;
}
}
反思:第一种方法比较容易想到,但是对Map的操作不是很熟练,第二种是看了大神的方法学会的。
解法参考:https://blog.csdn.net/biezhihua/article/details/79579221