美文网首页
贪心1 两个数组的交集

贪心1 两个数组的交集

作者: 是黄小胖呀 | 来源:发表于2020-07-13 23:05 被阅读0次

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

    示例 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 的大小比 nums2 小很多,哪种方法更优?

    如果 nums2 的元素存储在磁盘上,磁盘内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?

    解题思路:

    (1)利用哈希存储两个数组数据

    (2)再遍历其中一个数组,增加两个数组中长度较小的元素

    (3)利用一个中间哈希表存储已经判断过的元素

    优点:时间复杂度低,哈希检索得快;空间复杂度高,需要三个新的哈希表存储数据

    代码如下:

    class Solution:

        def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:

            dic1={}

            dic2={}

            tmp={}

            result=[]

            for i in range(len(nums1)):

                if nums1[i] in dic1:

                     dic1[nums1[i]]=dic1[nums1[i]]+1

                else:

                    dic1[nums1[i]]=1

            for i in range(len(nums2)):

                if nums2[i] in dic2:

                     dic2[nums2[i]]=dic2[nums2[i]]+1

                else:

                    dic2[nums2[i]]=1

            for i in range(len(nums1)):

                if (nums1[i] in dic2) and (nums1[i] not in tmp):

                    k=min(dic1[nums1[i]],dic2[nums1[i]])

                    for j in range(k):

                             result.append(nums1[i])

                    if nums1[i] not in tmp:

                        tmp[nums1[i]]=1

                    else:

                        tmp[nums1[i]]=tmp[nums1[i]]+1

                else:

                    continue

            return result

    相关文章

      网友评论

          本文标题:贪心1 两个数组的交集

          本文链接:https://www.haomeiwen.com/subject/iddthktx.html