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

2024-07-16

题目:

给你两个下标从 0 开始的整数数组 nums1nums2 ,它们分别含有 nm 个元素。

请你计算以下两个数值:

  • 统计 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 。

提示:

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

思路:

直接通过哈希表查找就行

代码:

class Solution {
public:
    vector<int> findIntersectionValues(vector<int>& nums1, vector<int>& nums2) {
        unordered_set<int> nums1_Cnt;
        unordered_set<int> nums2_Cnt;
        for (int i = 0; i < nums1.size(); i++) {
            nums1_Cnt.insert(nums1[i]);
        }
        for (int i = 0; i < nums2.size(); i++) {
            nums2_Cnt.insert(nums2[i]);
        }
        int a = 0, b = 0;
        for (int i = 0; i < nums1.size(); i++) {
            if (nums2_Cnt.find(nums1[i]) != nums2_Cnt.end()) {
                a++;
            }
        }
        for (int i = 0; i < nums2.size(); i++) {
            if (nums1_Cnt.find(nums2[i]) != nums1_Cnt.end()) {
                b++;
            }
        }
        return {a, b};
    }
};