原题链接:https://leetcode-cn.com/problems/3sum/。
简述题目要求:在列表中找到三个数的和为0.
首先建立一个map。map中存储的键值对为<值,这个值在列中的出现的频率>
将map按照键大于等于0和小于0分成两部分。
建立两层遍历,第一层遍历大于0的map,第二层遍历小于0的map.
取出一个大于0的map中的键i和小于0的map中的一个键j。得到diff = 0 -(i+j)。
如果diff在map中。
判断diff是否在(i,j)中,而且map[diff] >= 2.出现一个成立的值。[i,j,diff]
如果diff 不在(i,j)中,根据条件:
diff + i + j = 0
i >= 0
j < 0
必然有diff < j 或者diff > i.
如果以上不等式成立,也算一个成立的值。[i,j,diff]
以下是python代码
class Solution(object):
def threeSum(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
hash_list = {}
res = list()
for num in nums:
hash_list[num] = hash_list.get(num,0) + 1
if 0 in hash_list and hash_list[0] >= 3: # 如果list中有大于等于3个0,直接添加[0,0,0]
res.append([0,0,0])
neg = list(filter(lambda x:x<0,hash_list)) # 小于0 的map
pos = list(filter(lambda x:x>=0,hash_list)) # 大于0 的map
for i in neg:
for j in pos:
delt = 0 - i - j
if delt in hash_list:
if delt in (i,j) and hash_list[delt] >= 2:
res.append([i,j,delt])
if delt < i or delt > j:
res.append([i,j,delt])
return res