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()