Python实现求两个数组交集的方法示例

 更新时间:2019年02月23日 10:50:47   作者:linfeng886  
这篇文章主要介绍了Python实现求两个数组交集的方法,涉及Python数组遍历、排序、判断、追加等相关操作技巧,需要的朋友可以参考下

本文实例讲述了Python实现求两个数组交集的方法。分享给大家供大家参考,具体如下:

一、题目

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

例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]

更多关于Python相关内容感兴趣的读者可查看本站专题:《Python数组操作技巧总结》、《Python数据结构与算法教程》、《Python列表(list)操作技巧总结》、《Python函数使用技巧总结》、《Python字符串操作技巧汇总》及《Python入门与进阶经典教程

希望本文所述对大家Python程序设计有所帮助。

相关文章

  • python爬虫爬取某站上海租房图片

    python爬虫爬取某站上海租房图片

    这篇文章主要介绍了学习python爬虫并爬取某站上海租房图片的原理介绍以及相关代码分享,对此有兴趣的朋友学习下。
    2018-02-02
  • 使用Nibabel库对nii格式图像的读写操作

    使用Nibabel库对nii格式图像的读写操作

    这篇文章主要介绍了使用Nibabel库对nii格式图像的读写操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-07-07
  • 跟老齐学Python之通过Python连接数据库

    跟老齐学Python之通过Python连接数据库

    现在在做python的时候需要用到数据库,于是自己重新整理了一下数据库的知识,并且熟悉了python中MysqlDB模块的功能和函数等接口,现在系统地来总结一下吧
    2014-10-10
  • Python matplotlib图例放在外侧保存时显示不完整问题解决

    Python matplotlib图例放在外侧保存时显示不完整问题解决

    这篇文章主要介绍了Python matplotlib图例放在外侧保存时显示不完整问题解决,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2020-07-07
  • PyTorch: 梯度下降及反向传播的实例详解

    PyTorch: 梯度下降及反向传播的实例详解

    今天小编就为大家分享一篇PyTorch: 梯度下降及反向传播的实例详解,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2019-08-08
  • Python求字符串的长度示例代码

    Python求字符串的长度示例代码

    这篇文章主要介绍了Python求字符串的长度,求一个字符串的长度,在main函数中输入字符串,并输出其长度,文中给大家介绍了Python计算字符串长度的函数,需要的朋友可以参考下
    2023-02-02
  • YOLOv5目标检测之anchor设定

    YOLOv5目标检测之anchor设定

    在训练yolo网络检测目标时,需要根据待检测目标的位置大小分布情况对anchor进行调整,使其检测效果尽可能提高,下面这篇文章主要给大家介绍了关于YOLOv5目标检测之anchor设定的相关资料,需要的朋友可以参考下
    2022-05-05
  • 我在七夕佳节用Python制作的表白神器,程序员也应该拥有爱情!建议收藏

    我在七夕佳节用Python制作的表白神器,程序员也应该拥有爱情!建议收藏

    这篇文章主要介绍了我在七夕佳节用Python制作的表白神器,建议收藏,程序员也该拥有爱情,感兴趣的小伙伴快来看看吧
    2021-08-08
  • 10行Python代码就能实现的八种有趣功能详解

    10行Python代码就能实现的八种有趣功能详解

    Python凭借其简洁的代码,赢得了许多开发者的喜爱,因此也就促使了更多开发者用Python开发新的模块。面我们来看看,我们用不超过10行代码能实现些什么有趣的功能吧
    2022-03-03
  • 利用python代码写的12306订票代码

    利用python代码写的12306订票代码

    这篇文章主要介绍了利用python代码写的12306订票代码,自己写的python代码,是非常实用的技巧,具有一定参考借鉴价值,需要的朋友可以参考下
    2015-12-12

最新评论