350. 两个数组的交集 II

103 阅读1分钟

好青年 | leetcode 打卡群 - 打卡记录第 二 天

给你两个整数数组 nums1 和 nums2 ,请你以数组形式返回两数组的交集。返回结果中每个元素出现的次数,应与元素在两个数组中都出现的次数一致(如果出现次数不一致,则考虑取较小值)。可以不考虑输出结果的顺序。

  示例 1:

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

示例 2:

输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[4,9]
/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number[]}
 */
var intersect = function (nums1, nums2) {
    const map = {};
    const res = [];

    if (nums1.length < nums2.length) {
        [nums1, nums2] = [nums2, nums1];
    };

    // nums1中各个元素的频次
    for (const num1 of nums1) {
        if (map[num1]) {
            map[num1]++;
        } else {
            map[num1] = 1;
        }
    };

    for (const num2 of nums2) {
        const val = map[num2];
        if (val > 0) {
            res.push(num2);
            map[num2]--;
        }
    };


    return res;
};