题意:给定一个包含n个数的数组S,找出所有S中符合a+b+c=0的不同的三元组(a,b,c),注意解中不包含相同的三元组。
For example, given array S = [-1, 0, 1, 2, -1, -4],
A solution set is:
[
[-1, 0, 1],
[-1, -1, 2]
]
思路:既然是数组的题,很显然想到用三个指针分别指向这三个数,首先当然先要sort,当然这题有一个问题是如何处理那些相同的三元组,其实方法有两个,一个是直接用循环跳过那些重复的数字,还有一种是处理时不管,最后return的时候求set。
class Solution(object):
def threeSum(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
nums = sorted(nums)
answer = []
for i in xrange(len(nums)-2):
if i > 0 and nums[i] == nums[i-1]:
continue
j, k = i+1, len(nums)-1
while j < k:
if nums[i]+nums[j]+nums[k] > 0:
k -= 1
elif nums[i]+nums[j]+nums[k] < 0:
j += 1
else:
answer.append([nums[i], nums[j], nums[k]])
j, k = j+1, k-1
while j < k and nums[j] == nums[j - 1]:
j += 1
while j < k and nums[k] == nums[k + 1]:
k -= 1
return answer