leetcode-每日一题-No.2956. 找到两个数组中的公共元素

题目

给你两个下标从 0 开始的整数数组 nums1 和 nums2 ，它们分别含有 n 和 m 个元素。

请你计算以下两个数值：

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

答案代码

"""
给你两个下标从 0 开始的整数数组 nums1 和 nums2 ，它们分别含有 n 和 m 个元素。
请你计算以下两个数值：
统计 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(object):
    def findIntersectionValues(self, nums1: list[int], nums2: list[int]):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        count1 = 0
        count2 = 0
        for i in nums1:
            if i in nums2:
                count1 += 1
        for i in nums2:
            if i in nums1:
                count2 += 1
        return [count1, count2]


if __name__ == '__main__':
    print(Solution().findIntersectionValues(nums1=[4, 3, 2, 3, 1], nums2=[2, 2, 5, 2, 3, 6]))
    print(Solution().findIntersectionValues(nums1=[3, 4, 2, 3], nums2=[1, 5]))

结果分析

时间复杂度：代码中包含了两个嵌套循环，每个循环分别遍历 nums1 和 nums2 数组。如果 nums1 的长度为 n，nums2 的长度为 m，则总体的时间复杂度为 O(n * m)，在最坏情况下，两个循环都需要执行完整的遍历。因此，时间复杂度为O(mn)，其中 m 和 n 分别为 nums1​ 和 nums2​ 的长度。

空间复杂度：这段代码只需要常数级别的额外空间。具体来说，只需要两个变量 count1 和 count2 来存储统计结果，所以空间复杂度是 O(1)。