python实现给定列表和数值,找到列表中的两个或者多个元素和为该数值的元素下标

背景:

给定一个列表,给定一个数值,取列表中和为给定数值的两个元素或者3个元素。

举个例子:

给定列表为:numbers = [2, 7, 11, 15]

给定数值为:target = 9

取列表中那两个元素之和为9

献上代码:

numbers = [2, 7, 11, 15]

target = 9

def MakeSum(numbers,sum):
    for i in range(len(numbers)-1):
        for j in range(i+1,len(numbers)):
            if numbers[i] + numbers[j] == sum:
                print([i,j])

MakeSum(numbers,target)

执行结果:

如果想取列表中的三个元素之和为20的呢?

numbers = [2, 7, 11, 15]

target = 20
def MakeSum(numbers,sum):
    for i in range(len(numbers)-1):
        for j in range(i+1,len(numbers)):
            if numbers[i] + numbers[j] == sum:
                print([i,j])
            for k in range(j+1,len(numbers)):
                if numbers[i] + numbers[j] + numbers[k] == sum:
                    print([i,j,k])


MakeSum(numbers,target)

执行结果:

如果列表中2个数值的合为18,3个数值的合也为18呢?

上述代码可以满足该情况的。

numbers = [9,2, 7, 11, 15]

target = 18
def MakeSum(numbers,sum):
    for i in range(len(numbers)-1):
        for j in range(i+1,len(numbers)):
            if numbers[i] + numbers[j] == sum:
                print([i,j])
            for k in range(j+1,len(numbers)):
                if numbers[i] + numbers[j] + numbers[k] == sum:
                    print([i,j,k])


MakeSum(numbers,target)

执行结果为:

猜你喜欢

转载自blog.csdn.net/Mojitoice/article/details/107027057