LeetCode 350[Python]. 两个数组的交集 II.

1.题目描述

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

示例 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 的元素存储在磁盘上，内存是有限的，并且你不能一次加载所有的元素到内存中，你该怎么办？

2.解题思路&代码

• 方法1.

# 定义(非递归)二分查找函数,提高查找效率
def binary_chop(list, item):
    n = len(list)
    start = 0
    end = n - 1
    while start <= end:
        mid = (start + end) // 2
        if list[mid] == item:
            return item
        elif list[mid] < item:
            start = mid + 1
        elif list[mid] > item:
            end = mid - 1
    return None


class Solution(object):
    def intersect(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        # 定义空列表用来存储交集在中的数据
        list1 = []

        # 使用二分查找前需要对列表进行排序(二分查找使用的前提是数据必须通过顺序表存储且数据必须有序)
        nums2.sort()

        for item in nums1:
            result = binary_chop(nums2, item)
            if result is not None:
                # 如果nums2中找到了item,就把item从nums2中删除.此时item不用从nums1中删除是因为for循环会一直向后进行,既当前循环中的数据不会在后续循环中出现
                nums2.remove(result)
                list1.append(result)
        return list1

• 方法2.

# 定义(非递归)二分查找函数,提高查找效率
def binary_chop(list, item):
    n = len(list)
    start = 0
    end = n - 1
    while start <= end:
        mid = (start + end) // 2
        if list[mid] == item:
            return item
        elif list[mid] < item:
            start = mid + 1
        elif list[mid] > item:
            end = mid - 1
    return None


class Solution(object):
    def intersect(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        # 定义空列表用来存储交集在中的数据
        list1 = []

        # 使用二分查找前需要对列表进行排序(二分查找使用的前提是数据必须通过顺序表存储且数据必须有序)
        nums2.sort()

        for i in range(len(nums1) - 1, -1, -1):
            result = binary_chop(nums2, nums1[i])
            if result is not None:
                # 如果nums2中找到了item,就把item从nums2中删除.
                nums2.remove(result)
                # 把item从nums1中删除,只能在循环倒序遍历时从nums1删除数据,若正序遍历时从nums1中删数据,结果是不正确的
                del nums1[i]
                list1.append(result)
        return list1