leetcode 15 3sum问题

刚刚完成了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])