怎么使用Python實現(xiàn)求兩個數(shù)組交集的方法

小編給大家分享一下怎么使用Python實現(xiàn)求兩個數(shù)組交集的方法，相信大部分人都還不怎么了解，因此分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后大有收獲，下面讓我們一起去了解一下吧！

具體如下：

一、題目

給定兩個數(shù)組，編寫一個函數(shù)來計算它們的交集。

例1：

輸入: nums1 = [1,2,2,1], nums2 = [2,2]
輸出: [2,2]

例2：

輸入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
輸出: [4,9]

說明：

• 輸出結(jié)果中每個元素出現(xiàn)的次數(shù)，應(yīng)與元素在兩個數(shù)組中出現(xiàn)的次數(shù)一致

• 我們可以不考慮輸出結(jié)果的順序

二、解法

首先把兩個數(shù)組都排序，然后兩個數(shù)組進行遍歷比較，
當(dāng)值相等時，兩個數(shù)組都往后移動一位，并且相等的那個值加入到 result 數(shù)組中
當(dāng)值不相等時，值小的數(shù)組往后移動一位(因為我們已經(jīng)排序過了，如果是值大的數(shù)組往后移動一位，那么肯定還是比值小的要大)

代碼如下：

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)

運行結(jié)果：

[2, 2]

以上是“怎么使用Python實現(xiàn)求兩個數(shù)組交集的方法”這篇文章的所有內(nèi)容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內(nèi)容對大家有所幫助，如果還想學(xué)習(xí)更多知識，歡迎關(guān)注億速云行業(yè)資訊頻道！