LintCode python 小白-简单题-548两数组的交Ⅱ

最新推荐文章于 2019-03-21 20:57:44 发布

xiongxu3381

最新推荐文章于 2019-03-21 20:57:44 发布

阅读量371

点赞数 1

CC 4.0 BY-SA版权

分类专栏： python_lintcode

本文链接：https://blog.youkuaiyun.com/xiongxu3381/article/details/76383199

python_lintcode 专栏收录该内容

55 篇文章

订阅专栏

本文介绍了一种计算两个数组交集的有效方法。通过先对数组排序，然后比较元素，找到相同的项并添加到结果列表中。文章提供了一个Python类实现，并讨论了如何避免超时问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：计算两个数组的交

注意事项
每个元素出现次数得和在数组里一样
答案可以以任意顺序给出

样例：nums1 = [1, 2, 2, 1], nums2 = [2, 2], 返回 [2, 2].

该知识可以直接解决lintCode的547 的问题：
获取两个list的交集：list(set(a).intersection(set(b)))
获取两个list的并集：list(set(a).union(set(b)))
获取两个list的差集：list(set(a).difference(set(b)))

思路：这道题我换了很多种思路，一直在80多%出现TLE。
下面的思路就不会，主要是先将两个列表进行排序，因为根据答案可以以任意顺序给出，所以排完序，进行比较，是否相等情况，这里主要要点在：什么时候列表的下表+1，得到下一个元素。

class Solution:  
    # @param {int[]} nums1 an integer array  
    # @param {int[]} nums2 an integer array  
    # @return {int[]} an integer array  
    def intersection(self, nums1, nums2):  
        # Write your code here  
        if nums1 == None or nums2 == None:  
            return []  
        nums1.sort()
        nums2.sort()  
        nums = []  
        i = 0  
        j = 0  
        while i < len(nums1) and j < len(nums2):  
            if  nums1[i] ==nums2[j]:  #相等，添加到nums列表
                nums.append(nums1[i])  
                i += 1  
                j += 1  
            else:
                if nums1[i] > nums2[j]: #若1大于2，证明2要到下一个，否则1下一个
                    j+=1
                else:
                    i+=1
        return nums