python 四数之和
生活随笔
收集整理的這篇文章主要介紹了
python 四数之和
小編覺得挺不錯的,現(xiàn)在分享給大家,幫大家做個參考.
# -*- coding: utf-8 -*-
"""
Created on Sat Jul 17 17:23:50 2021@author: luogantt
"""
class Solution(object):def cartesian_product(self,a,b):l = [] #空的集合for x in a: #x為a中所有的成員for y in b: #y為b中所有的成員kk=str(x)+str(y)# print(kk)if len(set(kk))==len(kk):l.append(kk) #將所有可能的“有序?qū)Α奔拥健翱占稀敝衦eturn ldef threeSum(self, nums,n):""":type nums: List[int]:rtype: List[List[int]]"""a=list(range(len(nums)))c=afor p in range(n-1):c=self.cartesian_product(c,a)temp=[]for k in c:t1=list(k)t1.sort()t2=''.join(t1)
# print(t2)if t2 not in temp:temp.append(t2)all_combin=[] for p in temp:p1=[int(m) for m in p]count=0for q in p1 :count+=nums[q]if count==0:
# print(p1)all_combin.append([nums[k] for k in p1])return all_combinnums = [1,0,-1,0,-2,2]
c=Solution()
import time
t1=time.time()#計算幾個數(shù)之和,下面就填幾
c1=c.threeSum( nums,4)
print(time.time()-t1)print(c1)
0.003011465072631836
[[1, 0, -1, 0], [1, -1, -2, 2], [0, 0, -2, 2]]
總結(jié)
以上是生活随笔為你收集整理的python 四数之和的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: leetcode -python 三数之
- 下一篇: leetcode 在排序数组中查找元素的