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

1.题目描述

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

2.解题思路

使用hashSet将nums1,nums2中不同的元素分别存储在s1,s2中，然后依次遍历nums1中的每个元素，如果这个元素在s2中，则count1 += 1；如果nums2中的元素出现在s1中，则count2 += 1;遍历结束返回答案

3.代码实现

Python版本

class Solution:
    def findIntersectionValues(self, nums1: List[int], nums2: List[int]) -> List[int]:
        s1 = set(nums1)
        s2 = set(nums2)
        cnt1,cnt2 = 0,0
        for i in nums1:
            if i in s2:
                cnt1 += 1
        for j in nums2:
            if j in s1:
                cnt2 += 1
        return [cnt1,cnt2]

Java版本

class Solution {
    public int[] findIntersectionValues(int[] nums1, int[] nums2) {
        Set<Integer> s1 = new HashSet<>();
        for (int num : nums1) {
            s1.add(num);
        }
        Set<Integer> s2 = new HashSet<>();
        for (int num : nums2) {
            s2.add(num);
        }
        int[] res = new int[2];
        for (int num : nums1) {
            if (s2.contains(num)) {
                res[0] += 1;
            }
        }
        for (int num : nums2) {
            if (s1.contains(num)) {
                res[1] += 1;
            }
        }
        return res;
    }
}