LeetCode刷题记录——第350题（两个数组的交集 II）

题目描述

给定两个数组，编写一个函数来计算它们的交集。

示例 1:

输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2,2]

示例 2:

输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]

说明：

输出结果中每个元素出现的次数，应与元素在两个数组中出现的次数一致。
我们可以不考虑输出结果的顺序。

思路分析

• 从两者长度小的那一方着手，如果nums1长度较小，则遍历nums1，将nums1和nums2中的共同元素append进result，然后将该元素从nums2中remove掉，这一步是和349题不同的地方。

代码示例

class Solution(object):
    def intersect(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        result = []
        if len(nums1) <= len(nums2):
            for num in nums1:
                if num in nums2:
                    result.append(num)
                    nums2.remove(num)
            return result
        else:
            for num in nums2:
                if num in nums1:
                    result.append(num)
                    nums1.remove(num)
            return result