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

发布于 2024-07-16  5 次阅读


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

class Solution {
    public int[] findIntersectionValues(int[] nums1, int[] nums2) {
        HashMap<Integer,Integer> map1 = new HashMap<Integer,Integer>();
        HashMap<Integer,Integer> map2 = new HashMap<Integer,Integer>();
        for(int num : nums1){
            map1.put(num,map1.getOrDefault(num,0)+1);
        }
        for(int num : nums2){
            map2.put(num,map2.getOrDefault(num,0)+1);
        }
        int ans1 = 0;
        int ans2 = 0;
        for(int num : nums2){
            if(map1.containsKey(num)){
                ans1++;
            }
        }

        for(int num : nums1){
            if(map2.containsKey(num)){
                ans2++;
            }
        }

        return new int[]{ans2,ans1};
    }
}