假设我们有一个名为 w 的单词列表,带有小写字符串。我们必须找到 w 的最长序列的长度,其中每个前一个词是下一个词的前缀,下一个词只附加一个新字符。
所以,如果输入像 w = ["pqr", "pq", "m", "mn", "pqrs"],那么输出将是 3,因为我们可以得到序列:["pq", " pqr", "pqrs"],其长度为 3。
让我们看看以下实现以获得更好的理解 -
from collections import defaultdict def solve(w): w.sort() dp = defaultdict(int) res = 0 for word in w: dp[word] = dp[word[:-1]] + 1 res = max(res, dp[word]) return res w = ["pqr", "pq", "m", "mn", "pqrs"] print(solve(w))
["pqr", "pq", "m", "mn", "pqrs"]输出结果
3