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

最新推荐文章于 2024-10-14 18:52:56 发布

肆·意

最新推荐文章于 2024-10-14 18:52:56 发布

阅读量160

点赞数 8

文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/2301_79659699/article/details/140479333

版权

一、题目描述

给你两个下标从 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) {
        int res[] = new int[2];
        for(int i=0;i<nums1.length;i++){
            for(int j=0;j<nums2.length;j++){
                if(nums1[i]==nums2[j]){
                    res[0]++;
                    break;
                }
            }
        }
        for(int i=0;i<nums2.length;i++){
            for(int j=0;j<nums1.length;j++){
                 if(nums2[i]==nums1[j]){
                    res[1]++;
                    break;
                }
            }
        }
        return res;
    }
}

方法二：只需遍历一次

class Solution {
    public int[] findIntersectionValues(int[] nums1, int[] nums2) {
        int[] nums = new int[101];
        for (int i : nums1) {
            nums[i]++;
        }
        int n1 = 0, n2 = 0;
        for (int i : nums2) {
            if (nums[i] != 0) {
                n2++;
                n1 += (nums[i] == -1 ? 0 : nums[i]);
                nums[i] = -1;
            }
        }
        return new int[]{n1, n2};
    }
}

方法三：哈希表

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