这个题应该也是可以像前面那样,固定住两个点,然后左右两头向中心贪心,时间复杂度O(n3)。
但是还能做一些优化:
可以先把两两之和预处理存到Map里,key值是两数之和,内容是一个list,每个list的元素是一个pair,表示两数和为key的下标组合。
然后就暴力的从左到右确保下标的递增,将所有的结果都存起来,为了保证结果的唯一性,用一个set存结果,因为下标是递增的唯一的,数据也是递增的,那么对结果去重之后也是唯一的。
讲道理这个算法的时间复杂度是超过了O(n3)的,对于一些极端数据,比如说n个0的情况,target也是0,那么复杂度就是n*n*lg(C(n,4))*C(n,4),如果再把set添加考虑进去还要再乘一个lg,肯定是会炸掉的,n越大越炸,只是LeetCode数据实在是太水了。。。。。当然对于一般的数据,Python的字典是HashMap,时间可以忽略了,而每一个key值对应的组合不太多了话,这个时间复杂度是接近O(n2)的。
最后还要说明Python内,查找key值是否在字典里,可以用 if a (not) in dict,也可以用 dict.has_key(a),但是用in的效率会高一些。。。亲测有效
class Solution(object):
def fourSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[List[int]]
"""
Len=len(nums)
if Len<4:return []
Map={}
ans=set()
nums.sort()
for i in range(Len):
for j in range(i+1,Len):
if nums[i]+nums[j] not in Map:
Map[nums[i]+nums[j]]=[(i,j)]
else:
Map[nums[i]+nums[j]].append((i,j))
for i in range(Len):
for j in range(i+1,Len-2):
t=target-nums[i]-nums[j]
if t in Map:
for k in range(len(Map[t])):
if Map[t][k][0]>j:
ans.add((nums[i],nums[j],nums[Map[t][k][0]],nums[Map[t][k][1]]))
return list(ans)