python改进代码效率

ABCD_

1. class Solution:
   
2.     def twosum(self,nums,target):
   
3.         self.nums=nums
   
4.         self.target=target
   
5.         self.result=[]
   
6.         for each_1 in range(len(nums)-2):
   
7.             for each_2 in range(each_1+1,(len(nums)-1)):
   
8.                 if self.nums[each_1] + self.nums[each_2]==self.target:
   
9.                     self.result.append(each_1)
   
10.                     self.result.append(each_2)
    
11.         return self.result

复制代码

题目是这样的写一个class他下面有two sum， 然后你输入list[2,3,4,5],target=7的话会返回[0,3]，上面的程序已经实现了全部功能，有没有什么方法提高代码的效率！让他更快一点的找到

塔利班

1. from itertools import combinations
   
2. def fun(a,s):
   
3.     return [(a.index(each[0]),a.index(each[1])) for each in combinations(a,2) if sum(each)==s]
   
4. print(fun([2,3,4,5],7))

复制代码

如果没有重复元素的话

ABCD_

塔利班 发表于 2018-10-26 09:04
如果没有重复元素的话

啊不好意思忘记说了可以有重复元素（[0,3,4,0],0）会返回[0,3]