350. 两个数组的交集 II
给定两个数组,编写一个函数来计算它们的交集。
示例 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 的元素存储在磁盘上,内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?
解法
1、先对两个数组进行排序
先对两个数组进行排序,然后使用两个指针,分别指向两个数组开始的位置。
- 如果两个指针指向的值相同,说明这个值是他们的交集,就把这个值加入到集合 list 中,然后两个指针在分别往后移一步。
- 如果两个指针指向的值不同,那么指向的值相对小的往后移一步,相对大的先不动,然后再比较
一直重复上面的操作,直到其中一个指针不能再移动为止,最后再把集合 list 转化为数组即可
/**
* @param {number[]} nums1
* @param {number[]} nums2
* @return {number[]}
*/
var intersect = function (nums1, nums2) {
nums1.sort((a, b) => a - b);
nums2.sort((a, b) => a - b);
let i = 0;
let j = 0;
let result = [];
while (i < nums1.length && j < nums2.length) {
if (nums1[i] === nums2[j]) {
result.push(nums1[i]);
i++;
j++;
} else if (nums1[i] < nums2[j]) {
i++;
} else {
j++;
}
}
return result;
};
2、使用map解决
还可以使用map来解决,具体操作如下
遍历nums1中的所有元素,把它存放到map中,其中key就是nums1中的元素,value就是这个元素在数组nums1中出现的次数。
遍历nums2中的所有元素,查看map中是否包含nums2的元素,如果包含,就把当前值加入到集合list中,然后对应的value要减1。
/**
* @param {number[]} nums1
* @param {number[]} nums2
* @return {number[]}
*/
var intersect = function (nums1, nums2) {
const map = {};
for (let i = 0; i < nums1.length; i++) {
if (map[nums1[i]] === undefined) {
map[nums1[i]] = 1;
} else {
map[nums1[i]] += 1;
}
}
let result = [];
for (let j = 0; j < nums2.length; j++) {
if (map[nums2[j]]) {
result.push(nums2[j]);
map[nums2[j]] -= 1;
}
}
return result;
};