Leetcode每日一题 20240716 2956.找到两个数组中的公共元素

题目描述

给你两个下标从 0 开始的整数数组 nums1 和 nums2 ，它们分别含有 n 和 m 个元素。
请你计算以下两个数值：
统计 0 <= i < n 中的下标 i ，满足 nums1[i] 在 nums2 中 至少 出现了一次。
统计 0 <= i < m 中的下标 i ，满足 nums2[i] 在 nums1 中 至少 出现了一次。
请你返回一个长度为 2 的整数数组 answer ，按顺序 分别为以上两个数值。
2956. 找到两个数组中的公共元素

测试案例及提示

示例 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

解题思路

首先遍历nums1，如果元素在nums2中，则将ans[0]加一。然后遍历nums2，如果元素在nums1中，则将ans[-1]加一。
（判断时使用set(nums1/2),可以节省一些时间。）
python

class Solution:
    def findIntersectionValues(self, nums1: List[int], nums2: List[int]) -> List[int]:
        ans = [0,0]
        set_nums1 = set(nums1)
        set_nums2 = set(nums2)
        for a in nums1:
            ans[0] += 1 if a in set_nums2 else 0
        for b in nums2:
            ans[-1] += 1 if b in set_nums1 else 0
        return ans

Java

class Solution {
    public int[] findIntersectionValues(int[] nums1, int[] nums2) {
        int[] ans = new int[2];
        for (int a: nums1) {
            for (int b: nums2) {
                if (a == b) {
                    ans[0]++;
                    break;
                }
            }
        }
        for (int b: nums2) {
            for (int a: nums1) {
                if (b == a) {
                    ans[1]++;
                    break;
                }
            }
        }
        return ans;
    }
}

C

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* findIntersectionValues(int* nums1, int nums1Size, int* nums2, int nums2Size, int* returnSize) {
    int *ans = malloc(2 * sizeof(int));
    ans[0] = 0;
    ans[1] = 0;
    for (int i = 0; i < nums1Size; i++) {
        for (int j = 0; j < nums2Size; j++) {
            if (nums1[i] == nums2[j]) {
                ans[0]++;
                break;
            }
        }
    }
    for (int j = 0; j < nums2Size; j++) {
        for (int i = 0; i < nums1Size; i++) {
            if (nums2[j] == nums1[i]) {
                ans[1]++;
                break;
            }
        }
    }
    *returnSize = 2;
    return ans;
}