leetcode-2956.找到两个数组中的公共元素

2956. 找到两个数组中的公共元素

思路

• 新建两个哈希数组，分别存放所给两个数组中数字出现的次数

• 以原数组元素为下标得到哈希数组对应下标的元素和且一个下标仅加一次

Code

int fast_io=[](){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    return 0;
}();
class Solution {
public:
    vector<int> findIntersectionValues(vector<int>& nums1, vector<int>& nums2) {
        int new1[101]={0},new2[101]={0},sum1=0,sum2=0;
        for(auto e:nums1) new1[e]++;
        for(auto e:nums2) new2[e]++;
        for(auto e:nums1) if(new2[e]!=0) sum2+=new2[e],new2[e]=0;
        for(auto e:nums2) if(new1[e]!=0) sum1+=new1[e],new1[e]=0;
        return {sum1,sum2};
    }
};

复杂度

时间复杂度 $O(n)$	空间复杂度 $O(n)$

题面

给你两个下标从 0 开始的整数数组 nums1 和 nums2 ，它们分别含有 n 和 m 个元素。请你计算以下两个数值：

• answer1：使得 nums1[i] 在 nums2 中出现的下标 i 的数量。
• answer2：使得 nums2[i] 在 nums1 中出现的下标 i 的数量。

返回 [answer1, answer2]。

示例 1：

输入： nums1 = [2,3,2], nums2 = [1,2]

输出： [2,1]

解释：

示例 2：

输入： nums1 = [4,3,2,3,1], nums2 = [2,2,5,2,3,6]

输出： [3,4]

解释：

nums1 中下标在 1，2，3 的元素在 nums2 中也存在。所以 answer1 为 3。

nums2 中下标在 0，1，3，4 的元素在 nums1 中也存在。所以 answer2 为 4。

示例 3：

输入： nums1 = [3,4,2,3], nums2 = [1,5]

输出： [0,0]

解释：

nums1 和 nums2 中没有相同的数字，所以答案是 [0,0]。

提示：

• n == nums1.length
• m == nums2.length
• 1 <= n, m <= 100
• 1 <= nums1[i], nums2[i] <= 100