LeetCode中使用Python实现两个数组的交集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]  

说明：

• 输出结果中每个元素出现的次数，应与元素在两个数组中出现的次数一致。
• 我们可以不考虑输出结果的顺序。

进阶：

• 如果给定的数组已经排好序呢？你将如何优化你的算法？
• 如果 nums1 的大小比 nums2 小很多，哪种方法更优？
• 如果 nums2 的元素存储在磁盘上，磁盘内存是有限的，并且你不能一次加载所有的元素到内存中，你该怎么办？

废话少说直接上代码：

class Solution:
	def intersect(self, nums1, nums2):
		"""
		:type nums1: List[int]
		:type nums2: List[int]
		:rtype: List[int]
		"""
		nums1.sort()  
		nums2.sort()  
		
		new_li = list()
		i, j = 0, 0
		while i < len(nums1) and j < len(nums):  
			if nums1[i] == nums2[j]:  
				new_li.append(nums1[i])
				i += 1
				j += 1
			elif nums1[i] < nums2[j]:  
				i += 1
			else:
				j += 1
		
		return new_li

总结：

先创建用于存储相同元素的空列表，然后初始化游标，在游标不超出数组长度的情况下，判断两个数组里面的元素在当前索引位置是否相等，相等就添加进入新列表，同时移动两个游标（因为已判为相等，元素相当于被消耗掉了），如果不相等，则根据元素的大小移动较小一方的游标（因为数组已排序），知道任意索引超过数组长度跳出循环，返回新数组，即两数组的交集元素。