這篇文章主要介紹了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程序設計有所幫助。