本篇文章介绍Python实现求两个数组交集的方法。
一、题目
给定两个数组，编写一个函数来计算它们的交集。
例1：
时间 52ms，击败了 93%
本机测试代码如下：
运行结果：
[2, 2]
一、题目
给定两个数组，编写一个函数来计算它们的交集。
例1：
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2,2]
例2：
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]
说明：
- 输出结果中每个元素出现的次数，应与元素在两个数组中出现的次数一致
- 我们可以不考虑输出结果的顺序
二、解法
首先把两个数组都排序，然后两个数组进行遍历比较，
当值相等时，两个数组都往后移动一位，并且相等的那个值加入到 result 数组中
当值不相等时，值小的数组往后移动一位(因为我们已经排序过了，如果是值大的数组往后移动一位，那么肯定还是比值小的要大)
代码如下：
nums1.sort()
nums2.sort()
result = []
i, j = 0, 0
while i<len(nums1) and j<len(nums2):
if nums1[i]==nums2[j]:
result.append(nums1[i])
i += 1
j += 1
elif nums1[i]>nums2[j]:
j += 1
elif nums1[i]<nums2[j]:
i += 1
return result
|
时间 52ms，击败了 93%
本机测试代码如下：
# -*- coding:utf-8 -*-
nums1 = [1,2,2,1]
nums2 = [2,2]
nums1.sort()
nums2.sort()
result = []
i, j = 0, 0
while i<len(nums1) and j<len(nums2):
if nums1[i]==nums2[j]:
result.append(nums1[i])
i += 1
j += 1
elif nums1[i]>nums2[j]:
j += 1
elif nums1[i]<nums2[j]:
i += 1
print(result)
|
运行结果：
[2, 2]
原文链接：
相关文章
