# -*- 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) #將所有可能的“有序對”加到“空集合”中return 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()#計算幾個數之和，下面就填幾 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]]

總結

以上是生活随笔為你收集整理的python 四数之和的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。