5. Leetcode 15. 三数之和 (数组-双向双指针)
生活随笔
收集整理的這篇文章主要介紹了
5. Leetcode 15. 三数之和 (数组-双向双指针)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
給你一個包含 n 個整數的數組?nums,判斷?nums?中是否存在三個元素 a,b,c ,使得?a + b + c = 0 ?請你找出所有和為 0 且不重復的三元組。注意:答案中不可以包含重復的三元組。示例 1:輸入:nums = [-1,0,1,2,-1,-4]
輸出:[[-1,-1,2],[-1,0,1]]
示例 2:輸入:nums = []
輸出:[]主要思想: 采取固定一個數,同時用雙指針來查找另外兩個數的方式具體步驟如下:
1、我們可以先對數組進行排序,然后我們選擇一個數字做 C 位,然后我們在這 個 C 位數字的右邊進行雙指針搜索:
2、從最左邊 i+1(最小值)和最右邊 len(nums)-1(最大值)兩個數字開始,加 上 C 位,計算總和是否等于 0。
如果大于 0,說明實力太強了,就把右側的數字左移一位。
如果小于 0,說明實力太弱了,就把左邊的數字右移一位。 3、當雙指針碰到的時候,這輪循環結束,以該數字為 C 位的所有可能都已經嘗 試完畢了。class Solution:def threeSum(self, nums: List[int]) -> List[List[int]]:nums.sort()res = []for i in range(len(nums)):if i > 0 and nums[i] == nums[i-1]:continueleft, right = i + 1, len(nums) - 1while left < right:total = nums[left] + nums[right] + nums[i]if total < 0:left += 1elif total > 0:right -= 1else:res.append([nums[i],nums[left],nums[right]])while left < right and nums[left] == nums[left + 1]:left += 1while left < right and nums[right] == nums[right - 1]:right -= 1left += 1right -= 1return res
總結
以上是生活随笔為你收集整理的5. Leetcode 15. 三数之和 (数组-双向双指针)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 2. Leetcode 167. 两数之
- 下一篇: 3. Leetcode 16. 最接近的