1. 程式人生 > 其它 >每日一題-兩個陣列的交集 II

每日一題-兩個陣列的交集 II

技術標籤:每日一題

文章目錄


題目

給定兩個陣列,編寫一個函式來計算它們的交集。

來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/intersection-of-two-arrays-ii
著作權歸領釦網路所有。商業轉載請聯絡官方授權,非商業轉載請註明出處。

示例

示例 1:

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

示例 2:

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

說明:

輸出結果中每個元素出現的次數,應與元素在兩個陣列中出現次數的最小值一致。
我們可以不考慮輸出結果的順序。

進階:

如果給定的陣列已經排好序呢?你將如何優化你的演算法?
如果 nums1 的大小比 nums2 小很多,哪種方法更優?
如果 nums2 的元素儲存在磁碟上,記憶體是有限的,並且你不能一次載入所有的元素到記憶體中,你該怎麼辦?

解答

最笨的解法,遍歷短的陣列,如果元素在長的陣列中有,就將該元素追加到結果中,並在長的陣列中刪除該元素。

class Solution:
    def intersect(self, nums1, nums2):
        res = []
        if len(nums1) < len(nums2):
            for i in
nums1: if i in nums2: res.append(i) nums2.remove(i) else: for i in nums2: if i in nums1: res.append(i) nums1.remove(i) return res if __name__ == '__main__': l1 =
[4, 9, 5] l2 = [9, 4, 9, 8, 4] res = Solution().intersect(l1, l2) print(res)