Python：编写非常复杂的代码作为列表理解

如何解决Python：编写非常复杂的代码作为列表理解

我已通过以下代码解决了Word Break问题：

def wordBreak(self,s: str,wordDict: List[str]) -> bool:
        n = len(s)
        dp = [False] * n
        for i in range(n):
            for j in range(i + 1,n + 1):
                if (s[i : j] in wordDict) and (i == 0 or dp[i - 1]):
                    dp[j - 1] = True
        return dp[n - 1]

我正在尝试将其编写为列表理解，以使其成为“或多或少的单行代码”。

我一直试图表达dp如下：

dp = [True if (s[i:j] in wordDict) and (i==0 or dp[i-1]) for j in range(i+1,n+1) for i in range(n) else False]

但是，我有点卡住了，因为在原始代码中将其设置为dp[j - 1] = True，而我无法通过列表理解来实现它。

我知道写一个大代码作为列表理解并不是一个好主意（我也很困惑），但这只是出于教育目的。

对于正确编写此列表理解的任何帮助，我们深表感谢。

解决方法

实际上，不仅要写入dp，而且还要在下一次迭代中再次读取这些新写入的值，所以它不适合用于列表理解。 但是，如果您愿意牺牲一些Python最佳实践，则可以接近。

首先，您可以将内部循环转换为列表理解：

def wordBreak(self,s: str,wordDict: List[str]) -> bool:
    n = len(s)
    dp = [False] * n
    for i in range(n):
        dp[i:] = [dp[j] or
                   (s[i:j+1] in wordDict) and (i == 0 or dp[i-1])
                   for j in range(i,n)]
    return dp[-1]

让我们稍微改变一下算法，所以列表覆盖从索引0开始，而不是从索引i开始。相反，我们逐渐缩短了名单。同时，我们将dp[0]指定为要从先前迭代中读取的值，因此我们也将其前缀为初始列表：

def wordBreak(self,wordDict: List[str]) -> bool:
    n = len(s)
    dp = [True] + [False] * n
    for i in range(n):
        dp = [dp[j-i+1] or s[i:j+1] in wordDict and dp[0]
                   for j in range(i,n)]
    return dp[0]

但是dp = 是一种原则上阻止列表理解更广泛使用的赋值。 您可以使用a-Pythonic函数解决此问题，该函数会更改并返回一些内容：

def assign(self,target,source):
    target[:] = source  # mutate the target
    return target[-1]  # for our purposes we only need it to return the last value

现在我们可以将其包含在更大的列表理解表达式中：

def wordBreak(self,wordDict: List[str]) -> bool:
    n = len(s)
    dp = [True] + [False] * n
    return [
        self.assign(dp,[dp[j-i+1] or s[i:j+1] in wordDict and dp[0]
                        for j in range(i,n)])
                        for i in range(n)
    ][-1]

但是要重复一遍：这不是pythonic。最佳做法是让函数或更改参数（或self），或返回值，但不能同时返回值。该规则只有少数例外（例如.pop()）

如果您对函数表达式感到满意，而不是列表理解，则可以使用functools.reduce：

def wordBreak(self,wordDict: List[str]) -> bool:
    n = len(s)
    return reduce(lambda dp,i: [dp[j-i+1] or s[i:j+1] in wordDict and dp[0]
                                 for j in range(i,n)],range(n),[True] + [False] * n)[0]