不为别的,只为做一个连自己都羡慕的人

给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 n个 整数,并返回他们的数组下标。。

eg:

给定 nums =[3, 2, 3, 1, 5], target = 13

因为 nums[0] + nums[1] +nums[2] + nums[4]= 3+2+3+5=13 所以返回 [0, 1, 2, 4]

from itertools import combinations

a = [3, 2, 3, 1, 5]
target = 13


# 为数组标记下标
def markingIndex(a):
    arr = []
    for key, value in enumerate(a):
        obj = {"index": key, "value": value}
        arr.append(obj)
    return arr


bb = markingIndex(a)
# 为数组做了排列组合
arr1 = []
for i in range(1, len(bb)):
    if i + 1 > len(a):
        break
    else:
        arr1 += list(combinations(bb, i + 1))
# 对排列组合的数据求和,与目标值去做对比,当相等的时候,结束对比,最终返回下标
ee = []
for cc in arr1:
    print(cc)
    sumData = 0
    for dd in cc:
        sumData += dd["value"]
    if sumData == target:
        for e in cc:
            ee.append(e["index"])
        break
print(ee)

  

 

posted @ 2022-05-20 14:25  升级打怪  阅读(271)  评论(0编辑  收藏  举报