LeetCode-350 两个数组的交集 II

2019-02-23  本文已影响0人  FlyCharles

1. 题目

https://leetcode-cn.com/problems/intersection-of-two-arrays-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]

说明:

进阶:


2. 我的AC

class Solution(object):
    def intersect(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        result = []
        for num1 in nums1:
            if num1 in nums2:
                result.append(num1)
                nums2.remove(num1) # Important,注意以下测试案例
        return result

测试案例

输入
[1,2,2,1]
[2]
输出
[2,2]
预期结果
[2]

思路2

思想:通过set的唯一性找出两list的唯一交集,再利用min方法找到交集的共同出现次数

class Solution(object):
    def intersect(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        inter = set(nums1) & set(nums2)
        result = []
        for n in inter:
            result += [n] * min(nums1.count(n), nums2.count(n))  
        return result

3. 小结

上一篇下一篇

猜你喜欢

热点阅读