python数圈算法_算法题：三数之和

题目来源于力扣

理论基础哈希表

三数之和

题目描述三数之和

示例给定 [-1,0,1,2,-1,-4], target=0

解题思路暴力解法 O(N^3)

a+b+c=0 O(N^2)

双指针

Python 解法# a+b+c=0

def threeSum(self, nums):

if len(nums) < 3:

return []

nums.sort()

res = set()

for i, v in enumerate(nums[:-2]):

if i >= 1 and v == nums[i-1]:

continue

d = {}

for x in nums[i+1:]:

if x not in d:

d[-v-x] = 1

else:

res.add((v, -v-x, x))

return map(list, res)

# 双指针

def threeSum(self, nums):

res = []

nums.sort()

for i in xrange(len(nums)-2):

if i > 0 and nums[i] == nums[i-1]:

cintinue

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

while l < r:

s = nums[i] + nums[l] + nums[r]

if s < 0: l += 1

elif s > 0: r -= 1

else:

res.append((nums[i], nums[l], nums[r]))

# 判重处理

while l < r and nums[l] == nums[l+1]:

l += 1

while l < r and nums[r] == nums[r-1]:

r -= 1

l += 1; r -= 1

return res