LeetCode 筆記 - 349. Intersection of Two Arrays

⚠️ 舊文提醒：本文發佈於約 4 年前，部分內容或指令可能已過時，請斟酌參考。

請找出兩個數列之間重複的元素

解題思維

由於題目要求答案必須是沒有重複的，所以我們可以先把數列先做成集合再開始做事

接下來再 Two Pointers 大法慢慢往下掃之前，記得需要先 Sorting

這個方法就可以在 O(N) 內把問題空間掃描一下

程式碼

class Solution:
    def intersection(self, nums1: List[int], nums2: List[int]) -> List[int]:
        if not nums1 or not nums2:
            return []
        
        nums1 = list(set(nums1))
        nums2 = list(set(nums2))
        
        size1 = len(nums1)
        size2 = len(nums2)
        
        nums1.sort()
        nums2.sort()
        
        index1 = index2 = 0
        
        result = []
        while index1 < size1 and index2 < size2:
            
            if nums1[index1] < nums2[index2]:
                index1 += 1
                continue
            elif nums1[index1] > nums2[index2]:
                index2 += 1
                continue
            
            result.append(nums1[index1])
            
            index1 += 1
            index2 += 1
            
        return result

解題思維

程式碼

也許你也會想看看