本文實例講述了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, 0while 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 += 1return result
時間 52ms,擊敗了 93%
本機測試代碼如下:
# -*- coding:utf-8 -*-nums1 = [1,2,2,1]nums2 = [2,2]nums1.sort()nums2.sort()result = []i, j = 0, 0while 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 += 1print(result)
運行結(jié)果:
[2, 2]
更多關(guān)于Python相關(guān)內(nèi)容感興趣的讀者可查看本站專題:《Python數(shù)組操作技巧總結(jié)》、《Python數(shù)據(jù)結(jié)構(gòu)與算法教程》、《Python列表(list)操作技巧總結(jié)》、《Python函數(shù)使用技巧總結(jié)》、《Python字符串操作技巧匯總》及《Python入門與進階經(jīng)典教程》
希望本文所述對大家Python程序設(shè)計有所幫助。
新聞熱點
疑難解答