leetcode15--三數之和
阿新 • • 發佈:2019-01-09
題目
給定一個包含 n 個整數的陣列 nums,判斷 nums 中是否存在三個元素 a,b,c ,使得 a + b + c = 0 ?找出所有滿足條件且不重複的三元組。
注意:答案中不可以包含重複的三元組。
例如, 給定陣列 nums = [-1, 0, 1, 2, -1, -4],
滿足要求的三元組集合為:
[
[-1, 0, 1],
[-1, -1, 2]
]
思路
給定一個數列a+b+c=0,求的問題可以轉換為求a+b=-c:, 我們將輸入的list排序後,給定nums[i], 我們需要尋找 nums[j]+nums[k] = - nums[i], 這裡的 。
題目要求不可包含重複的解,對於排序後的陣列,如果num[i] = num[i+1],那麼他們獲得的nums[j]和nums[k]肯定是相同的(如果解存在的話),所以應該跳過這種情況。同理,如果num[j] = num[j+1], 那麼也會找到相同的nums[k]。這種情況也應該跳過。
class Solution:
def threeSum(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
#儲存結果列表
res=[]
#對nums列表進行排序,無返回值,排序直接改變nums順序
nums.sort()
# 對排序後的nums進行遍歷
for i in range(len(nums)):
#如果排序後第一個數都大於0,則跳出迴圈,不可能有為0 的三數之和
if nums[0]>1:
break
# 排序後相鄰兩數如果相等,則跳出當前迴圈繼續下一次迴圈,相同的數只需要計算一次
if i>0 and nums[i]==nums[i-1]:
continue
#記錄i的下一個位置
j=i+1
#最後一個元素的位置
k=len(nums)-1
while j< k:
#判斷三數之和是否為零
if nums[j]+nums[k] ==-nums[i]:
#把結果加入陣列之中
res.append([nums[i],nums[j],nums[k]])
#判斷j相鄰元素是否相等,有的話直接跳過這個
while j<k and nums[j] ==nums[j+1]:
j+=1
while j<k and nums[k]==nums[k-1]:
k-=1
#沒有相等則j+1,k-1,縮小範圍
j+=1
k-=1
#小於-nums[i]的話還能繼續往後取
elif nums[j]+ nums[k] < -nums[i] :
j +=1
else:
k-=1
return res