三个嵌套的 for 循环会降低性能

Question

我想知道是否有另一种方法来解决这个问题，而无需像我在这里所做的那样使用 3 个嵌套的 for 循环？我知道，如果要在足够大的列表上测试该方法，那么以这种方式嵌套循环很可能会导致很多问题。

问题来了：

from typing import List

def can_pay_with_three_coins(denoms: List[int], amount: int) -> bool:
    """Return True if and only if it is possible to form amount, which is a
    number of cents, using exactly three coins, which can be of any of the
    denominations in denoms.

    >>> can_pay_with_three_coins([1, 5, 10, 25], 36)
    True
    >>> can_pay_with_three_coins([1, 5, 10, 25], 37)
    False

    """

这是我的解决方案：

for i in range(len(denoms)):
    one_coin = denoms[i]
    for j in range(len(denoms)):
        another_coin = denoms[j]
        for k in range(len(denoms)):
            last_coin = denoms[k]
            if one_coin + another_coin + last_coin == amount:
                return True
return False

我确信还有另一种方法可以解决这个问题，只是我真的想不出来。
感谢您的帮助！

Answer 1

这是一个著名的问题，名为 3 sum。

这个解决方案的时间复杂度是O(n^3)，你可以实现一个O(n^2)的算法，在下面的链接中用多种语言解释和实现：

Find a triplet that sum to a given value

Answer 2

好吧，让我们用 itertools 作弊吧:)

import itertools
from typing import List


def can_pay_with_three_coins(denoms: List[int], amount: int) -> bool:
    """Return True if and only if it is possible to form amount, which is a
    number of cents, using exactly three coins, which can be of any of the
    denominations in denoms.

    >>> can_pay_with_three_coins([1, 5, 10, 25], 36)
    True
    >>> can_pay_with_three_coins([1, 5, 10, 25], 37)
    False

    """

    for variant in itertools.permutations(denoms, len(denoms)):
        if sum(variant[:3]) == amount:
            return True

    return False


print(can_pay_with_three_coins([1, 5, 10, 25], 36))
print(can_pay_with_three_coins([1, 5, 10, 25], 37))
print(can_pay_with_three_coins([1, 1, 5, 10, 25], 37))
print(can_pay_with_three_coins([1, 3, 5, 10, 25], 37))
print(can_pay_with_three_coins([20, 20, 20, 50], 60))

输出

True
False
False
False
True