leetcode 15 三數之和

給你乙個包含 n 個整數的陣列 nums，判斷 nums 中是否存在三個元素 a，b，c ，使得 a + b + c = 0 ？請你找出所有滿足條件且不重複的三元組。

注意：答案中不可以包含重複的三元組。

示例：給定陣列 nums = [-1, 0, 1, 2, -1, -4]，

滿足要求的三元組集合為：

[[-1, 0, 1],

[-1, -1, 2]

]

class solution(object):
def threesum(self, nums):
""":type nums: list[int]
:rtype: list[list[int]]
"""n = len(nums)
nums.sort()
ans = 
if not nums or n < 3:
return ans
for i in range(n-2):
if nums[i] > 0:
return ans
if nums[i] == nums[i-1] and i > 0:
continue
left = i + 1
right = n - 1
while left < right:
if nums[i] + nums[left] + nums[right] == 0:
while left < right and nums[left] == nums[left+1]: 
left += 1
while left < right and nums[right] == nums[right-1]: 
right -= 1
right -= 1
left += 1
elif nums[i] + nums[left] + nums[right] > 0:
right -= 1
else:
left += 1
return ans

對於三數相加，最直觀的是使用暴力解法，但是需要窮舉n^3個例子，時間效率為o(n^3)，效能非常差。對此我們可以先對其進行排序，然後在陣列排序的基礎上再對其進行遍歷。在遍歷取到第i個值時，我們設定兩個指標分別指向i之後一位，以及陣列的最後一位，然後讓兩個指標向內收斂。每次判斷nums[i] + nums[left] + nums[right] 是否為0，由於陣列以及排好序，那麼如果和大於0，就需要更小的數字，即將右指標向左移，如果和小於0，就需要更大的數字，即將左指標向右移。

最後只需要處理重複的情況即可，當指標在移動時如果相鄰兩個值相等就會出現重複情況，那麼只要將指標再進一步移動就可以防止重複。

時間複雜度分析：排序時間複雜度o(nlogn)，遍歷一遍數字o(n)，內部指標掃瞄陣列o(n)，所以整個搜尋時間效率為o(n^2)，比暴力法小乙個量級。

leetcode 15 三數之和

LeetCode 15 三數之和

leetcode 15 三數之和

leetcode15 三數之和

leetcode 15 三數之和

LeetCode 15 三數之和

leetcode 15 三數之和

leetcode15 三數之和

相關推薦