简体中文

两个数组的交集

给定两个数组，编写一个函数来计算它们的交集。

示例

输入：nums1 = [1,2,2,1], nums2 = [2,2] 输出：[2]
输入：nums1 = [4,9,5], nums2 = [9,4,9,8,4] 输出：[9,4]

题解

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number[]}
 */
var intersection = function(nums1, nums2) {
    nums1.sort((a, b) => a - b);
    nums2.sort((a, b) => a - b);
    let target = [];
    let i = 0, k = 0;
    let n1 = nums1.length, n2 = nums2.length;
    while(i<n1 && k<n2){
        if(nums1[i] < nums2[k]){
            ++i;
        }else if(nums1[i] > nums2[k]){
            ++k;
        }else{
            let last = target.length-1;
            if(target[last] !== nums1[i]) target.push(nums1[i]);
            ++i;
            ++k;
        }
    }
    return target;
};

思路

比较常规的题目，计算两个数组的交集最简单的方式就是遍历数组nums1，对于其中的每个元素，遍历数组nums2判断该元素是否在数组nums2中，如果存在，则将该元素添加到返回值，这样的方式时间复杂度是O(mn)，在这里使用排序加双指针的方式，首先对于两个数组分别进行排序，之后分别对于两个数组设立指针进行遍历，对比两个指针所指向的元素，较小的值的指针后移，如果相等则判断是否已经在目标数组中，不在则将其推入数组，之后同时将两个指针后移，最终返回目标数组即可。首先将两个数组分别从小到大进行排序，之后定义目标数组target，以及两个指针i、k与两个数组的长度n1、n2，定义循环，在两个指针分别小于其指向的目标数组的长度下执行循环，如果i指针指向的值小于k指针指向的值，将i指针后移，如果大于则将k指针后移，如果相等则首先得到目标数组的最后一个值的索引，当然在数组为空的情况下会得到-1，在Js中会取得undefined值，在下方比较时不会相等，之后比较最后一个值是否与此时指针指向的值相等，不相等则将值推入数组，这样用来进行去重操作，之后将两个指针分别后移，循环结束后返回目标数组即可。

每日一题

https://github.com/WindrunnerMax/EveryDay

参考

https://leetcode-cn.com/problems/intersection-of-two-arrays/