two sum python_python 实现一个TwoSum的例子

今天无意中看到一个题目，也不是很难，就想着用python实现以下：

题目是数组中的两个数相加等于输入的一个target，然后输出数组的下标。

比如：

[1,2,3,4,5,6] target=7  返回[1,4]

主要是坑就是避免有重复的数字

代码如下

# -*- coding:utf -8 -*-

def TwoSum(nums,target):

if len(nums) <0:

return False

for m in nums:

n = target-m

if n in nums:

if m != n:

index_n = [i for i,a in enumerate(nums) if a==n][0]

index_m = [i for i,a in enumerate(nums) if a==m][0]

return [index_m,index_n]

if(m == n and len([i for i,a in enumerate(nums) if a==n]) >1):

index_m = [i for i,a in enumerate(nums) if a==n][0]

index_n = [i for i,a in enumerate(nums) if a==n][1]

return [index_m,index_n]