力扣刷题--2956. 找到两个数组中的公共元素【简单】

码农小苏242024-05-28 23:28

题目描述

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

请你计算以下两个数值：

统计 0 <= i < n 中的下标 i ，满足 nums1[i] 在 nums2 中至少出现了一次。

统计 0 <= i < m 中的下标 i ，满足 nums2[i] 在 nums1 中至少出现了一次。

请你返回一个长度为 2 的整数数组 answer ，按顺序分别为以上两个数值。

示例 1：

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

输出：[3,4]

解释：分别计算两个数值：

nums1 中下标为 1 ，2 和 3 的元素在 nums2 中至少出现了一次，所以第一个值为 3 。
nums2 中下标为 0 ，1 ，3 和 4 的元素在 nums1 中至少出现了一次，所以第二个值为 4 。
示例 2：

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

输出：[0,0]

解释：两个数组中没有公共元素，所以两个值都为 0 。

算法分析

算法1：暴力，两个for循环，时间复杂度O(n*n)
算法2：哈希表，把两个数组放入哈希表中，时间复杂度O(n)

完整代码

算法1：

cpp 复制代码

在这里插入代码class Solution {
public:
    vector<int> findIntersectionValues(vector<int>& nums1, vector<int>& nums2) {
       //暴力 //优化：哈希表 unorder_set<int>s;
        vector<int>ans;
        int sum1=0;
        int sum2=0;
        for(auto i:nums1)
        {
            if(find(nums2.begin(),nums2.end(),i)!=nums2.end())
                sum1++;
        }
        for(auto i:nums2)
        {
            if(find(nums1.begin(),nums1.end(),i)!=nums1.end())
                sum2++;
        }
        ans.push_back(sum1);
        ans.push_back(sum2);
        return ans;
    }
};

算法2：

cpp 复制代码

class Solution {
public:
    vector<int> findIntersectionValues(vector<int> &nums1, vector<int> &nums2) {
        unordered_set<int> set1(nums1.begin(), nums1.end());
        unordered_set<int> set2(nums2.begin(), nums2.end());
        vector<int> ans(2);
        for (int x: nums1) ans[0] += set2.count(x);
        for (int x: nums2) ans[1] += set1.count(x);
        return ans;
    }
};