1. 程式人生 > >陣列——6、兩個陣列的交集II

陣列——6、兩個陣列的交集II

給定一個數組,編寫一個函式來計算它們的交集。
示例:

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

說明:

  • 輸出結果中每個元素出現的次數應與元素在兩個陣列中出現的次數一致;
  • 可以不考慮輸出結果的順序。

思路:判斷短列表中的元素是否在長列表中,若在,則新增到心底列表中,同時長列表刪除相應元素,避免重複。

class Solution:
    def intersect(self, nums1, nums2):
    	nums = []
    	if len(nums1)
< len(nums2): short = nums1 lon = nums2 else: short = nums2 lon = nums1 for i in short: if i in lon: lon.remove(i) nums.append(i) return nums