[leetcode]-3sums @python

转载：http://www.cnblogs.com/zuoyuan/p/3699159.html

原题地址：http://oj.leetcode.com/problems/3sum/

题意：从一个数组中找到三个数，使这三个数的和为0。有可能存在多组解，也有可能存在重复的解，所以需要去重。比如：num=[-1,0,1,2,-1,-4];那么存在两组解：[[-1,0,1],[-1,-1,2]]，解中的数需要是从小到大排序状态。

解题思路：1，先将数组排序。

　　　　　2，排序后，可以按照TwoSum的思路来解题。怎么解呢？可以将num[i]的相反数即-num[i]作为target，然后从i+1到len(num)-1的数组元素中寻找两个数使它们的和为-num[i]就可以了。下标i的范围是从0到len(num)-3。

　　　　　3，这个过程要注意去重。

　　　　   4，时间复杂度为O(N^2)。

代码：

class Solution:
    # @return a list of lists of length 3, [[val1,val2,val3]]
    def threeSum(self, num):
        num.sort()
        res = []
        for i in range(len(num)-2):
            if i == 0 or num[i] > num[i-1]:
                left = i + 1; right = len(num) - 1
                while left < right:
                    if num[left] + num[right] == -num[i]:
                        res.append([num[i], num[left], num[right]])
                        left += 1; right -= 1
                        while left < right and num[left] == num[left-1]: left +=1
                        while left < right and num[right] == num[right+1]: right -= 1
                    elif num[left] + num[right] < -num[i]:
                        while left < right:
                            left += 1
                            if num[left] > num[left-1]: break
                    else:
                        while left < right:
                            right -= 1
                            if num[right] < num[right+1]: break
        return res

补充个人代码：

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        #lens=len(nums)
        result={}
        nums.sort() #排序的意义在于便于排除已经处理的数字，若是与上次处理的数字相同，则不处理
        k=-1
        for i in range(0,len(nums)-2):
            other1=0-nums[i]
            if i!=0 and nums[i]==nums[i-1]:#排序后若有重复数字，一定是相邻的，且结果一定重复，要排除
                continue
            for j in range(i+1,len(nums)-1):
                if j!=i+1 and nums[j]==nums[j-1]:
                    continue
                for m in range(len(nums)-1,j,-1):
                        if nums[j]+nums[m]==other1:
                            k+=1
                            result[k]=[nums[i],nums[j],nums[m]]
                            break
        return result.values()

根据参考重新更改代码：

class Solution:
    # @return a list of lists of length 3, [[val1,val2,val3]]
    def threeSum(self,nums):
        nums.sort()
        r=[]
        for i in range(0,len(nums)-2):
            if i == 0 or nums[i] > nums[i-1]:
                left = i + 1; right = len(nums) - 1
            while left<right:
                if nums[left]+nums[right]==-nums[i]:
                    r.append([nums[i],nums[left],nums[right]])
                    left+=1;right-=1 #左右两边都要改变，只改变左边的值，则和不为0或是和为0但重复结果
                    while left<right and nums[left]==nums[left-1]: left+=1 #去除重复数字的搜索
                    while left<right and nums[right]==nums[right+1]: right-=1
                elif nums[left]+nums[right]<-nums[i]:    #小于和，说明不够大，左边向右移
                   while left<right:
                          left+=1
                          if nums[left]!=nums[left-1]:break#循环去除重复数字的处理
                else:
                    while left<right:
                        right-=1
                        if nums[right]!=nums[right+1]:break
        return r

总结：

1、通过对循环条件的细致分析，去除部分循环过程。例如题目从两边逼近中间的过程中，原和比预期值小，则说明左边的值小，需要向右移。而不是简单的对所有的数字组合简单罗列判断。

2、python ：

1）列表的嵌套

res=[]

res.append（[n[i],n[j],n[k]])

2)不存在++，--

3）循环条件

顺序循环：for i in range(0,len(n)) 从0到len(n)-1

倒序循环：for i in range(len(n)-1,-1,-1) 从len(n)-1到0