leetcode 之 349. 两个数组的交集

1,403 阅读1分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第4天,点击查看活动详情

一 描述

349. 两个数组的交集

给定两个数组 nums1 和 nums2 ,返回 它们的交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。

 

示例 1:

输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2]

示例 2:

输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[9,4]
解释:[4,9] 也是可通过的

提示:

  • 1 <= nums1.length, nums2.length <= 1000
  • 0 <= nums1[i], nums2[i] <= 1000

二 分析

首先将两个数组分别从小到大进行排序

之后定义目标数组 target ,以及两个指针i、k 与两个数组的长度 n1、n2

定义循环,在两个指针分别小于其指向的目标数组的长度下执行循环

如果 i 指针指向的值小于 k 指针指向的值,将 i 指针后移

如果大于则将 k 指针后移,如果相等则首先得到目标数组的最后一个值的索引

当然在数组为空的情况下会得到 -1

在 JS 中会得到为 undefined 值,在下方比较时不会相等

之后比较最后一个值是否与此时指针指向的值相等,不相等则将值推入数组

这样用来进行去重操作,之后将两个指针分别后移操作,循环结束后返回目标数组即可

三 答案

/**
 * @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;
};

参考

349. 两个数组的交集(C++「BF法+Set法」AC) - 两个数组的交集 - 力扣(LeetCode)

【两个数组的交集】【C语言详解】【2种思路】 - 两个数组的交集 - 力扣(LeetCode)