LeetCode(#15) 三数之和(python)&去掉列表中相同数字&去掉二维数组中相同行

leetcode #15

给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。

注意:答案中不可以包含重复的三元组。

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],

满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]

class Solution:
    def threeSum(self, nums):
        # 先排序
        nums.sort()
        res=[]
        length=len(nums)
        for k in range(length-2):
            if nums[k]>0:break
            if k>0 and nums[k]==nums[k-1]:
                continue
            i,j=k+1,length-1
            while i<j:
                s=nums[k]+nums[i]+nums[j]
                if s<0:
                    i+=1
                    while i<j and nums[i]==nums[i-1]:
                        i+=1
                elif s>0:
                    j-=1
                    while i<j and nums[j]==nums[i+1]:
                        i-=1
                else:
                    res.append([nums[k],nums[i],nums[j]])
                    i+=1
                    j-=1
                    while i<j and nums[j]==nums[i-1]:
                        i+=1
                    while i<j and nums[j] == nums[j+1]:
                        j -= 1
        return res

if __name__=="__main__":
    array=list(int(x) for x in input().split())
    lcy=Solution()
    print(lcy.threeSum(array))

去掉列表中相同数字

方法一:

#利用集合,直接将列表转化为集合,自动去重后转回列表。有一个问题,转换为集合的同时,数据无序了。
li = [11,22,22,33,44,44]
set = set(li)
li = list(set)
print(li)
#
#
第二种运用新建字典的方式,去除重复的键
list = [11,22,33,22,44,33]
dic = {}
list = dic.fromkeys(list).keys()#字典在创建新的字典时,有重复key则覆盖
print(list)
#
#
#
#第三种是用列表的推导
list = [11,22,33,22,44,33]
lis = []                          #创建一个新列表
for i in list:                    #循环list里的每一个元素
	if  i  not in lis:            #判断元素是否存在新列表中,不存在则添加,存在则跳过,以此去重
        lis.append(i)
print(lis)
#
#
#
#第四种仅仅只是将for循环变为while循环
list = [11,22,33,22,44,33]
result_list=[]
temp_list=list
i=0
while i<len(temp_list):
    if temp_list[i] not in result_list:
       result_list.append(temp_list[i])
    else:
        i+=1
print(result_list)
import numpy as np
A = [1, 2, 2, 5,3, 4, 3]
a = np.unique(A)

去掉二维数组中相同行

import numpy as np
arr = np.array([[1, 2],[3, 4],[5, 6],[7, 8],[3, 4],[1, 2]])
print(np.array(list(set([tuple(t) for t in arr]))))

只有元组可以哈希(set),列表不可以哈希。
所以要把列表先转为元组tuple()

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值