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

题目内容

给你两个下标从 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]。

初始思路

分别遍历nums1，nums2。遍历nums1时，对每个nums[i]，若nums2中存在相等数字，对应ans++，然后break即可（break防止重复计数），nums2同理。

class Solution {
public:
    vector<int> findIntersectionValues(vector<int>& nums1, vector<int>& nums2) {
        int ans1 = 0, ans2 = 0;
        int n1 = nums1.size(), n2 = nums2.size();
        for(int i = 0; i < n1; i++)
        {
            for(int j = 0; j < n2; j++)
            {
                if(nums1[i] == nums2[j])
                {
                    ans1++;
                    break;
                }
            }
        }
        for(int i = 0; i < n2; i++)
        {
            for(int j = 0; j < n1; j++)
            {
                if(nums2[i] == nums1[j])
                {
                    ans2++;
                    break;
                }
            }
        }
        vector<int> ans;
        ans.push_back(ans1);
        ans.push_back(ans2);
        return ans;
    }
};