[leetcode]4Sum @ Python

时间：2014-04-30 13:28:46 收藏：0 阅读：445

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

题意：从数组中找到4个数，使它们的和为target。要求去重，可能有多组解，需要都找出来。

解题思路：一开始想要像3Sum那样去解题，时间复杂度为O(N^3)，可无论怎么写都是Time Limited Exceeded。而同样的算法使用C++是可以通过的。说明Python的执行速度比C++慢很多。还说明了一点，大概出题人的意思不是要让我们去像3Sum那样去解题，否则这道题就出的没有意义了。这里参考了kitt的解法：http://chaoren.is-programmer.com/posts/45308.html

　　　　需要用到哈希表的思路，这样可以空间换时间，以增加空间复杂度的代价来降低时间复杂度。首先建立一个字典dict，字典的key值为数组中每两个元素的和，每个key对应的value为这两个元素的下标组成的元组，元组不一定是唯一的。如对于num=[1,2,3,2]来说，dict={3:[(0,1),(0,3)], 4:[(0,2),(1,3)], 5:[(1,2),(2,3)]}。这样就可以检查target-key这个值在不在dict的key值中，如果target-key在dict中并且下标符合要求，那么就找到了这样的一组解。由于需要去重，这里选用set()类型的数据结构，即无序无重复元素集。最后将每个找出来的解(set()类型)转换成list类型输出即可。

代码：

class Solution:
    # @return a list of lists of length 4, [[val1,val2,val3,val4]]
    def fourSum(self, num, target):
        numLen, res, dict = len(num), set(), {}
        if numLen < 4: return []
        num.sort()
        for p in range(numLen):
            for q in range(p+1, numLen): 
                if num[p]+num[q] not in dict:
                    dict[num[p]+num[q]] = [(p,q)]
                else:
                    dict[num[p]+num[q]].append((p,q))
        for i in range(numLen):
            for j in range(i+1, numLen-2):
                T = target-num[i]-num[j]
                if T in dict:
                    for k in dict[T]:
                        if k[0] > j: res.add((num[i],num[j],num[k[0]],num[k[1]]))
        return [list(i) for i in res]

再贴两个过不了的O(N^3)写法。

一：

class Solution:
    # @return a list of lists of length 4, [[val1,val2,val3,val4]]
    def fourSum(self, num, target):
        num.sort(); res=[]
        for i in range(len(num)):
            if i>0 and num[i]==num[i-1]: continue
            for j in range(i+1,len(num)):
                if j>i+1 and num[j]==num[j-1]: continue
                l=j+1; r=len(num)-1
                while l<r:
                    sum=num[i]+num[j]+num[l]+num[r]
                    if sum>target:
                        r-=1
                    elif sum<target:
                        l+=1
                    elif l>j+1 and num[l]==num[l-1]:
                        l+=1
                    elif r<len(num)-1 and num[r]==num[r+1]:
                        r-=1
                    else:
                        res.append([num[i],num[j],num[l],num[r]])
                        l+=1; r-=1
        return res

二：

class Solution:
    # @return a list of lists of length 4, [[val1,val2,val3,val4]]
    def fourSum(self, num, target):
        num.sort(); res=[]
        for i in range(len(num)-3):
            if i==0 or num[i]>num[i-1]:
                for j in range(i+1,len(num)-2):
                    if j==i+1 or num[j]>num[j-1]:
                        left=j+1; right=len(num)-1
                        while left<right:
                            if num[i]+num[j]+num[left]+num[right]==target:
                                res.append([num[i],num[j],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[i]+num[j]+num[left]+num[right]>target:
                                while left<right:
                                    right-=1
                                    if num[right]<num[right+1]: break
                            else:
                                while left<right:
                                    left+=1
                                    if num[left]>num[left-1]: break
        return res

[leetcode]4Sum @ Python,布布扣,bubuko.com