LeetCode.15.三数之和

原题链接: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        
        
                
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值