Python实现"两个数组的交集||"的两种方法

给定两个数组,写一个方法输出它们的交集

Example 1:

Input: nums1 = [1,2,2,1], nums2 = [2,2]
Output: [2,2]

Example 2:

Input: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
Output: [4,9]

注意:

输出结果中每一个元素出现的次数和两个数组中 该元素同时出现的次数的一致

输出结果元素顺序不做要求

进阶:

如果给定的数组已经按序排好,如何调整优化你的算法?

如果nums1的长度小于nums2,哪一种算法更好?

如果nums2的元素存储在磁盘上,磁盘内存有限以致于无法一次性加载所有元素,此时该怎么办?

1:不考虑进阶。排序nums1和nums2,然后进行相应操作

def intersect(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        nums1.sort()
        nums2.sort()
        numsList = []
        count1 = 0
        count2 = 0
        while count1 < len(nums1) and count2 < len(nums2):
            if nums1[count1] == nums2[count2]:
                numsList.append(nums1[count1])
                count1 += 1
                count2 += 1
            elif nums1[count1] < nums2[count2]:
                count1 += 1
            else:
                count2 += 1
        return numsList

2:(不考虑元素排序)借助字典,存放nums1中出现的元素和对应的次数,然后访问nums2,进行对应操作(参考他人)

def intersect(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        nums1Dic = {}
        numsList = []
        for i in nums1:
            if nums1Dic.get(i):
                nums1Dic[i] += 1
            else:
                nums1Dic[i] = 1
        for i in nums2:
            if nums1Dic.get(i) > 0:
                numsList.append(i)
                nums1Dic[i] -= 1
        return numsList

算法题来自:https://leetcode-cn.com/problems/intersection-of-two-arrays-ii/description/

你可能感兴趣的:(Algorithms)