刚刚完成了leetcode的15题, 在一个数组里找到所有的三个数字和为0的序列.
看完问题, 第一个想法, 三次循环暴力啊, 但只是想想
我想的第二种方法, 和他们评论最高的相似, 只是没有想去排序, 结果还是以三次方终结.......
所以总结来说,当一个问题的复杂度很高的时候, 尤其是与数字有关, 我们是可以考虑通过排序, 来降低整体的复杂度. 在数字有了顺序后, 遍历和判断结合在一起, 就方便了.每一次总结学到的方法, 增加自己的解决问题的能力.
#!/usr/bin/env python
# coding=utf-8
class Solution(object):
def threeSum(self, nums):
res = []
nums.sort()
for index in range(len(nums) - 2):
if index > 0 and nums[index] == nums[index-1]:
continue
start, end = index + 1, len(nums) - 1
while start < end:
tmp_sum = nums[index] + nums[start] + nums[end]
if tmp_sum > 0:
end -= 1
elif tmp_sum < 0:
start += 1
else:
res.append([nums[index], nums[start], nums[end]])
while start < end and nums[start] == nums[start + 1]:
start += 1
while start < end and nums[end] == nums[end - 1]:
end -= 1
start += 1
end -= 1
return res
s = Solution()
print s.threeSum([-1, 0, 1, 2, -1, -4])