Python：防止双for循环返回相同的列表索引

Question

我目前正在Leetcode中处理问题1，名为“两个和”

给定一个整数数组，返回两个数字的索引，使它们相加到一个特定的目标。

您可以假设每个输入都有一个精确的解决方案，并且您可以不使用相同的元素两次。

示例：给定nums=[2,7,11,15]，target=9，

因为nums[0]nums[1]=27=9，所以返回[0,1]。

我目前的代码是：

def twosum_indices(nums, target):
    for i in nums:
        for v in nums[i + 1:]:
            if i + v == target:
                return nums.index(i), nums.index(v)

在这种情况下，Nums是一个整数列表，程序必须在列表中返回两个不同的索引，这样它们的真实值加起来就是给定的目标。尽管这在大多数测试用例上运行良好，但它在像[3,3]这样两个值相同的列表中失败，并且返回两次相同的索引，比如[0,0]，而不是返回[0,1]的实际答案。为什么会发生这种情况？

Answer 1

def twosum_indices(nums, target):
    for i in range(len(nums)):
        for v in range(len(nums)):
            if nums[i] + nums[v] == target and i != v:
                return i, v

Answer 2

#! /usr/bin/env python3


def two_sum_indices(nums, target):

    def dup(i, j):
        return i == j

    d = {num: i
         for i, num in enumerate(nums)}

    for i, num in enumerate(nums):
        if target - num in d:
            if not dup(i, d[target - num]):
                return i, d[target - num]
    return -1, -1


if __name__ == '__main__':

    print(two_sum_indices([2, 7, 11, 15], target=9))
    print(two_sum_indices([3], target=6))
    print(two_sum_indices([3, 3], target=6))

Answer 3

代码中存在多个错误，其中最重要的一个错误是未能使用enumerate而不是list.index。例如，[3,3]。索引（3）当然总是0。

这个答案的重点不是找到最有效的解决方案，而是改进您的具体方法。你也可以看到O（n）解。

先决条件是，首先要了解列表中如何存在多个for循环。

def sums(nums):
    return [x + y for x in nums for y in nums[:x]]

以上相当于：

def sums(nums):
    output = []
    for x in nums:
       for y in nums[:x]:
           output.append(x + y)
    return output

def twosum_indices(nums, target):
    return next((i, j) for i in range(len(nums)) for j in range(len(nums[:i])) if (nums[i] + nums[j] == target))

例子：

print(sorted(twosum_indices([2, 7, 11, 15], 9)))
[0, 1]

print(sorted(twosum_indices([3, 3], 6)))
[0, 1]

这是一个稍微简单一点的迭代工具：

import itertools

def twosum_indices_it(nums, target):
    return next((i, j) for (i, x), (j, y) in itertools.combinations(enumerate(nums), 2) if (x + y == target))

例子：

print(sorted(twosum_indices_it([2, 7, 11, 15], 9)))
[0, 1]

print(sorted(twosum_indices_it([3, 3], 6)))
[0, 1]