【Leetcode】python – [15] 3Sum 個人解法筆記 (last update: 2022/4/6)

➣ Reading Time: 19 minutes

題目出處

15. 3Sum

難度

Medium

題目分類

Array, TwoPointers

個人範例程式碼 – two pointers (第二次解:2022/4/6)

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        nums.sort()
        ans = []
        for idx, target in enumerate(nums):
            if idx > 0 and nums[idx] == nums[idx-1]: # prevent duplicate
                continue
            ans = self.two_sum(nums, ans, idx+1, len(nums)-1, -target)
        return ans


    def two_sum(self, nums, ans, start, end, target):
        last_ans = []
        while(start < end):
            if(nums[start] + nums[end] == target):
                if [-target, nums[start], nums[end]] != last_ans: # prevent duplicate
                    last_ans = [-target, nums[start], nums[end]]
                    ans.append(last_ans)
                start += 1
                end -= 1
            elif(nums[start] + nums[end] < target):
                start += 1
            else: # if(nums[start] + nums[end] > target):
                end -= 1
        else:
            return ans

算法說明

3Sum 可以當作是 2Sum 的進階版,
而變化的部分在於 target 的變化變成可以動態的,

可以視為:3Sum = 2Sum + 「動態 target

因此我們可以沿用 2Sum 的作法,
並把 2Sum 中 target 改為動態的尋找。

最近在練習程式碼本身就可以自解釋的 Coding style,可以嘗試直接閱讀程式碼理解

input handling

處理 input 為 [] 的情況,應該要輸出 []。
處理 input 為 [0, 0, 0] 的情況,應該也要輸出 [[0, 0, 0]]。

Boundary conditions

有幾個重點:

防止重複 1 : 「a <= b」 <= c

防止重複的第一個部分,就是避免重複數值時,
a, b 都被重算,我們都統一取第一個去運算。

  • 此外需要特別注意:因為判斷式為 「nums[idx] nums[idx-1]」,需要再多判斷一個「idx > 0」。
if idx > 0 and nums[idx] == nums[idx-1]: # prevent duplicate
    continue

防止重複 2 : a <=「b <= c」

這邊基本上我們統一拿來結果來判斷,只要 b, c 組合不相同,就視為不同的解。

3. two pointers: start < end,交錯或等於就結束

two pointers 的基本結束條件。

個人解法筆記 – (第一次解:2021/4/2)

這題是第 1 題的延伸題目,這次我們要處理 3數之和 的問題。

大概念

我們將這個題目看成,「任兩數和」 等於 「負的第三數」
「負的第三數」我們稱之為我們要的 target
為了更乾淨我們處理的順序,我們會先進行「排序」

示意圖:

TwoPointers

在排序後,我們選定 target,我們使用 TwoPointers,
從左 ( l ) 開始小往大掃、從右 ( r ) 開始大往小掃,
當左( l ) 大於等於右( r ) ,也就是交錯甚至超過。
就是我們的終止條件。

示意圖:

重複處理 case1 : target 的重複

重複時的選擇:我們選擇第一個 target。

除了 target 重複會造成重複解答之外,
選擇第一個也能夠保證當解答中有與target重複的數值也能被我們考慮到。

  • 示意圖:(處理 target 的重複)

  • 可以觀察 i 與 target 的變化,仔細思考,可以得到一些心得:

重複處理 case2 : r, l 選到值的重複

  • 重複時的選擇:我們選擇「l」的第一個值,r不同時處理。

因為,當我們得到答案並更新「l」之後,對應的 nums[r] 直接不成立。
思考時,只需要思考單邊重複如何處理即可。 (多思考會錯,可以見下面錯誤區)

選擇第一個值的原因是,我們預期「剩下的值」有可能存在答案。
當找到答案時,我們再將「l」移至下一個值 (也就是說,反覆 l+1 直到值不同)

這樣才能處理極端狀況:[0, 0, 0]

  • 示意圖:(處理 l, r 的重複)

個人範例程式碼

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        nums.sort()
        ans = []

        # decide target
        for i in range(len(nums)):
            if(i > 0 and nums[i-1] == nums[i]): # when same target, only calculate first 
                continue # go to next loop
            target = -nums[i]
            # print(f"i = {i}, target = {target}") # debug

            l = i+1
            r = len(nums)-1

            while(l < r and l < len(nums) and r >= 0): # normal condition
                # print(f"l = {l}, r = {r}") # debug
                if nums[l] + nums[r] == target:
                    ans.append([nums[i], nums[l], nums[r]])
                    l += 1
                    while l < r and nums[l] == nums[l-1]: # when same value, only calculate first
                        l += 1 
                elif nums[l] + nums[r] > target: # need smaller, r - 1
                    r -= 1                  
                else: # nums[l] + nums[r] < target: need bigger, l + 1
                    l += 1        

        return ans

「錯誤紀錄」 手繪筆記

重點:錯誤錯在兩邊同時尋找,無法解[0,0,0] 這個 case!

這題是第 1 題的延伸題目,這次我們要處理 3數之和 的問題。

大概念

我們將這個題目看成,「任兩數和」 等於 「負的第三數」
「負的第三數」我們稱之為我們要的 target
為了更乾淨我們處理的順序,我們會先進行「排序」

示意圖:

TwoPointers

在排序後,我們選定 target,我們使用 TwoPointers,
從左 ( l ) 開始小往大掃、從右 ( r ) 開始大往小掃,
當左( l ) 大於等於右( r ) ,也就是交錯甚至超過。
就是我們的終止條件。

示意圖:

重複處理 case1 : target 的重複

重複時的選擇:我們選擇第一個 target。

除了 target 重複會造成重複解答之外,
選擇第一個也能夠保證當解答中有與target重複的數值也能被我們考慮到。

示意圖:

重複處理 case2 : r, l 選到值的重複

重複時的選擇:我們選擇最後一個值。

選擇最後一個值的原因是,我們預期「剩下兩個值」的組合應該會不同。
所以我們只看最後一個。

而且,就算真的重複了。(例如極端狀況:[0, 0, 0]) (此方法並沒有辦法處理)
我們也會先進行 -target = nums[l] + nums[r] 的檢查後,
抓出這樣的極端情況。

示意圖:

Reference

⭐ Leetcode 解題紀錄 ⭐題型資料結構Python SolutionC++ SolutionNote
⭐BFS 相關題型 ⭐
104Maximum Depth of Binary TreeBFS (分層)Python
94Binary Tree Inorder TraversalBFS (分層)TreePython 內含 處理 Tree 樹問題的重點
102Binary Tree Level Order TraversalBFS (分層)TreePython
103Binary Tree Zigzag Level Order TraversalBFS (分層)TreePython
107Binary Tree Level Order Traversal IIBFS (分層)TreePython
133Clone GraphBFS (分層)GraphPython Graph 的基本操作 #重要題型
127Word LadderBFS (分層), DFSGraphPython
[Lint] 127Topological SortingBFS (拓撲)Python
內有 indegree, outdegree 介紹 #重要題型
207Course ScheduleBFS (拓樸)GraphPython
210Course Schedule IIBFS (拓樸)GraphPython
[Lint] 892Alien DictionaryBFS (拓樸)GraphPython
[Lint] 431Connected Component in Undirected GraphBFS (連通塊)GraphPython 內含 BFS 模板 #重要題型
1091Shortest Path in Binary MatrixBFS (最短路徑)MatrixPython
⭐ Binary Serach 相關題型 ⭐
33Search in Rotated Sorted ArrayBinary SerachArrayPython #重要題型
34Find First and Last Position of Element in Sorted ArrayBinary SerachPython
50Pow(x, n) Binary SerachPython
153Find Minimum in Rotated Sorted Array Binary SerachArrayPython
162Find Peak ElementBinary SerachPython
278First Bad VersionBinary SerachPython
658Find K Closest ElementsBinary SerachPython
704Binary SearchBinary SerachPython
852Peak Index in a Mountain ArrayBinary SerachPython
[Lint] 14First Position of TargetBinary SerachPython
[Lint] 140Fast PowerBinary SerachPython
[Lint] 447Search in a Big Sorted ArrayBinary SerachArrayPython
[Lint] 458Last Position of TargetBinary SerachPython
191Number of 1 BitsBit ManipulationPython
⭐ Data Stream 相關題型 ⭐
[Lint] 642Moving Average from Data StreamData StreamQueuePython
⭐ Design 相關題型 ⭐
155Min StackDesignPython
[Lint] 659Encode and Decode StringsDesignPython
232Implement Queue using StacksDesignQueue, StackPython
⭐ DFS 相關題型 ⭐
98Validate Binary Search TreeDFSPython
2265Count Nodes Equal to Average of SubtreeDFSPython
292nd Weekly Contest
2261K Divisible Elements SubarraysDFSPython
內含 python substring 常見作法 / 291st LeetCode Weekly Contest
22Generate ParenthesesDFSPython
79Word SearchDFSMatrixPython
126Word Ladder IIDFSPython
212Word Search IIDFSTreePython
290Word PatternDFSPython
[Lint] 829Word Pattern IIDFSPython
31Next PermutationDFS (排列)Python
46PermutationsDFS (排列)Python
#重要題型
47Permutations IIDFS (排列)Python
#重要題型
51N-QueensDFS (排列)Python
52N-Queens IIDFS (排列)Python
[Lint] 862Next Closest TimeDFS (排列)Python
內有 set 判斷是否 subset 的用法
10Regular Expression MatchingDFS (組合)Python
77CombinationsDFS (組合)Python
#重要題型
39Combination SumDFS (組合)Python
40Combination Sum IIDFS (組合)Python
216Combination Sum IIIDFS (組合)Python
377Combination Sum IVDFS (組合)Python
44Wildcard MatchingDFS (組合)Python
78SubsetsDFS (組合)Python
#重要題型
90Subsets IIDFS (組合)Python
#重要題型
131Palindrome PartitioningDFS (組合)Python
139Word BreakDFS (組合)Python
140Word Break IIDFS (組合)Python
[Lint] 90k Sum IIDFS (組合)Python
[Lint] 680Split StringDFS (組合)Python
173Binary Search Tree IteratorDFS (BST)BSTPython
#重要題型
230Kth Smallest Element in a BSTDFS (BST)BSTPython
[Lint] 448Inorder Successor in BSTDFS (BST)BSTPython
[Lint] 900Closest Binary Search Tree ValueDFS (BST)BSTPython
[Lint] 901Closest Binary Search Tree Value IIDFS (BST)BSTPython
#綜合難題
208Implement Trie (Prefix Tree)DFS (Graph)TreePython
17Letter Combinations of a Phone NumberDFS (Graph)GraphPython
332Reconstruct ItineraryDFS (Graph)GraphPython
#重要題型
110Balanced Binary TreeDFS (Tree)TreePython
226Invert Binary TreeDFS (Tree)TreePython
572Subtree of Another TreeDFS (Tree)TreePython
105Construct Binary Tree from Preorder and Inorder TraversalDFS (Tree)TreePython
112Path SumDFS (Tree)Python
113Path Sum IIDFS (Tree)TreePython
235Lowest Common Ancestor of a Binary Search TreeDFS (Tree)TreePython
236Lowest Common Ancestor of a Binary TreeDFS (Tree)TreePython
#重要題型
257Binary Tree PathsDFS (Tree)TreePython
[Lint] 474Lowest Common Ancestor IIDFS (Tree)TreePython
[Lint] 578Lowest Common Ancestor IIIDFS (Tree)TreePython
[Lint] 596Minimum SubtreeDFS (Tree)TreePython
543Diameter of Binary TreeDFS (Tree)Python
144Binary Tree Preorder TraversalDFS (Tree)TreePython
內含 處理 Tree 樹問題的重點
145Binary Tree Postorder TraversalDFS (Tree)TreePython
114Flatten Binary Tree to Linked ListDFS (Tree)TreePython
⭐ Dynamic Programming 相關題型 ⭐
338Counting BitsDPPython
309Best Time to Buy and Sell Stock with CooldownDPPython
2266Count Number of TextsDPPython
292nd Weekly Contest
2262Total Appeal of A StringDPPython
291st Weekly Contest
91Decode Ways DPPython C++
70Climbing Stairs DPPython C++
221Maximal SquareDPPython
53Maximum SubarrayDP (一維)ArrayPython C++ 內含 C++ vector max 用法整理
91Decode WaysDP (一維)Python
55Jump GameDP (一維)Python
#重要題型
45Jump Game IIDP (一維)Python
#重要題型
198House RobberDP (一維)Python
213House Robber IIDP (一維)Python
509Fibonacci NumberDP (一維)Python
122Best Time to Buy and Sell Stock IIDP (一維)Python
300Longest Increasing SubsequenceDP (一維接龍, LIS)Python
#重要題型
62Unique PathsDP (二維)Python C++
63Unique Paths IIDP (二維)Python C++
152Maximum Product SubarrayDP (二維)Python
#重要題型
118Pascal’s TriangleDP (二維)Python
內含 python sum of two list (list add 相加方法整理)
322Coin ChangeDP (背包問題)Python
內含 DP 背包問題模板 #重要題型
518Coin Change 2DP (背包問題)Python
#重要題型
[Lint] 92BackpackDP (背包問題)Python
[Lint] 125Backpack IIDP (背包問題)Python
[Lint] 440Backpack IIIDP (背包問題)Python
[Lint] 562Backpack IVDP (背包問題)Python
[Lint] 563Backpack VDP (背包問題)Python
[Lint] 798Backpack VIIDP (背包問題)Python
[Lint] 799Backpack VIIIDP (背包問題)Python
[Lint] 800Backpack IXDP (背包問題)Python
[Lint] 801Backpack XDP (背包問題)Python
1143Longest Common SubsequenceDP (LCS)Python
494Target SumDP (Memoization)Python
2267Check if There Is a Valid Parentheses String PathDP (Memoization)Python
內含:Memoization 記憶化搜索筆記 / 292nd Weekly Contest
⭐ Hash 相關題型 ⭐
13Roman to IntegerHashPython
73Set Matrix Zeroes HashPython C++ 內含 python while-else 用法說明
692Top K Frequent WordsHashPython
內含 python 自定義排序 key function 的使用方法
846Hand of StraightsHashPython
347Top K Frequent ElementsHashPython
205Isomorphic StringsHashPython
268Missing NumberHashPython
242Valid AnagramHashPython
763Partition LabelsHashIntervalPython
383Ransom NoteHashPython
387 First Unique Character in a String HashPython
2186Minimum Number of Steps to Make Two Strings Anagram IIHashPython
282nd Weekly Contest
[Lint] 793Intersection of ArraysHashPython
146LRU CacheHashPython
387First Unique Character in a StringHashPython
[Lint] 920Meeting RoomsHashIntervalPython
#重要題型
[Lint] 920Meeting RoomsHashIntervalPython
#重要題型
[Lint] 919Meeting Rooms IIHashIntervalPython
#重要題型
[Lint] 919Meeting Rooms IIHashIntervalPython
#重要題型
349 Intersection of Two ArraysHashPython
217Contains DuplicateHashArrayPython
C++ 內含 C++ set, unordered_set 用法整理
137Single Number IIHashPython
350Intersection of Two Arrays IIHashPython
2248Intersection of Multiple ArraysHashPython
290th Weekly Contest
49Group AnagramsHash (Anagrams)Python
217Contains DuplicateHash (Duplicate)Python
128Longest Consecutive SequenceHash (LCS)Python
136Single NumberHash, Bit ManipulationPython
⭐ Heap 相關題型 ⭐
264Ugly Number IIHeapPython
⭐ Stack 相關題型 ⭐
739Daily TemperaturesStackPython
496Next Greater Element IStackPython
503Next Greater Element IIStackPython
20Valid ParenthesesStack (Parentheses)Python
內含用 python List 組出 Stack, Queue 的方法整理
⭐ Two pointers 相關題型 ⭐
11Container With Most WaterTwo pointersPython
42Trapping Rain WaterTwo pointersPython
74Search a 2D Matrix Two pointersMatrixPython
141Linked List CycleTwo pointersLinked ListPython
142Linked List Cycle IITwo pointersLinked ListPython
內含 python while-else 用法介紹
283Move ZeroesTwo pointersPython
876Middle of the Linked ListTwo pointersPython
2260Minimum Consecutive Cards to Pick UpTwo pointers (快慢)Python
291st Weekly Contest
21Merge Two Sorted ListsTwo pointers (Merge)Linked ListPython
內含 python Linked List 基本操作 (for 新手教學)
88Merge Sorted Array Two pointers (Merge)Python C++ 內含 python while-else 用法說明
1Two SumTwo pointers (NSum)ArrayPython C++ 內有 Python list comprehesion / dict comprehesion 整理 / C++ map find 方法補充 (dict find)
153Sum Two pointers (NSum)ArrayPython
163Sum ClosestTwo pointers (NSum)Python
167Two Sum II – Input Array Is SortedTwo pointers (NSum)Python
[Lint] 382Triangle CountTwo pointers (NSum)Python
[Lint] 533Two Sum – Closest to TargetTwo pointers (NSum)Python
[Lint] 587Two Sum – Unique PairsTwo pointers (NSum)Python
5Longest Palindromic Substring Two pointers (Palindrome)Python C++ 內含 C++ string.substr() 用法筆記
9Palindrome NumberTwo pointers (Palindrome)Python
125Valid PalindromeTwo pointers (Palindrome)StringPython內含 python isalpha(), isalnum() 的整理
680Valid Palindrome IITwo pointers (Palindrome)Python
409Longest Palindrome
Two pointers (Palindrome)Python
647Palindromic SubstringsTwo pointers (Palindrome)Python
234Palindrome Linked ListTwo pointers (Palindrome)Linked ListPython
內含 reverse LinkedList 方法
75Sort ColorsTwo pointers (partition)Python
#重要題型
[Lint] 5Kth Largest ElementTwo pointers (partition)Python
#重要題型
[Lint] 31Partition ArrayTwo pointers (partition)Python
#重要題型
[Lint] 143Sort Colors IITwo pointers (partition)Python
#重要題型
[Lint] 461Kth Smallest Numbers in Unsorted ArrayTwo pointers (partition)Python
#重要題型
3Longest Substring Without Repeating CharactersTwo pointers (Sliding Window)Python
76 Minimum Window SubstringTwo pointers (Sliding Window)Python
239Sliding Window MaximumTwo pointers (Sliding Window)Python
⭐ 其他題型 / 待分類 ⭐
2Add Two NumbersLinked ListPython
7Reverse IntegerPython
19Remove Nth Node From End of ListLinked ListPython
36Valid SudokuPython
48Rotate ImageMatrixPython
621Task SchedulerPython
202Happy NumberPython
238Product of Array Except SelfPython
222Count Complete Tree NodesTreePython
674Longest Continuous Increasing SubsequencePython
435Non-overlapping IntervalsIntervalPython
88Merge Sorted ArrayPython
內含 python while-else 用法說明
2264Largest 3-Same-Digit Number in StringPython
292nd Weekly Contest
56Merge IntervalsIntervalPython
內含:python sorted key 搭配 lambda 的用法範例
57Insert IntervalIntervalPython
61Rotate ListLinked ListPython
2259Remove Digit From Number to Maximize ResultPython
291st Weekly Contest
53Maximum SubarrayPython
54Spiral MatrixPython
228Summary RangesPython
263Ugly NumberPython
203Remove Linked List ElementsLinked ListPython
內含 Linked List remove 操作 part 2 (for 新手教學)
206Reverse Linked ListLinked ListPython
內含 Linked List reverse 反轉操作 part 3 (for 新手教學)
189Rotate ArrayArrayPython
2185Counting Words With a Given PrefixStringPython
282nd Weekly Contest
134Gas StationPython
121Best Time to Buy and Sell StockPython
83Remove Duplicates from Sorted ListLinked ListPython
566Reshape the MatrixMatrixPython
內含 python array 初始化, index 操作與控制範例
2243Calculate Digit Sum of a StringPython
289th Weekly Contest
2244Minimum Rounds to Complete All TasksPython
289th Weekly Contest
2249Count Lattice Points Inside a CirclePython
290th Weekly Contest
⭐【喜歡我的文章嗎? 歡迎幫我按讚~ 讓基金會請創作者喝一杯咖啡!
如果喜歡我的文章,請幫我在下方【按五下Like】 (Google, Facebook 免註冊),會由 「LikeCoin」 贊助作者鼓勵繼續創作,讀者們「只需幫忙按讚,完全不用出錢」哦!

likecoin-steps
預設圖片
Howard Weng

我是 Howard Weng,很多人叫我嗡嗡。這個網站放了我的各種筆記。希望這些筆記也能順便幫助到有需要的人們!如果文章有幫助到你的話,歡迎幫我點讚哦!

文章: 673

★留個言吧!內容有誤或想要補充也歡迎與我討論!