问题描述
LC 问题:https://leetcode.com/problems/concatenated-words/
var findAllConcatenatedWordsInADict = function(words) {
let m = new Map(),memo = new Map();
let res = [];
for (let i = 0; i < words.length; i++) {
m.set(words[i],1);
}
for (let i = 0; i < words.length; i++) {
if (isConcat(words[i],m,memo)) res.push(words[i]);
}
return res;
};
function isConcat(word,memo) {
if (memo.has(word)) return memo.get(word);
for (let i = 1; i < word.length; i++) {
let prefix = word.slice(0,i);
let suffix = word.slice(i);
if (m.has(prefix) && (m.has(suffix) || isConcat(suffix,memo))) {
memo.set(word,true);
return true;
}
}
memo.set(word,false);
return false;
};
-
此外,我尝试运行各种测试用例 ["cat","cats","dog","dogcatsdog","rat","ratcatdogcat"]。
似乎我们没有在从 word[0] ('at') 生成的第一个后缀上调用 isConcat 函数。但是,我们似乎确实将它称为“catsdog”,作为“dogcatsdog”词的一部分,所以我不确定逻辑是如何工作的,即它如何选择要调用的后缀....
解决方法
我的错误是误读了代码并假设以下条件必须为有效的连接词返回 true:
Map has both prefix and suffix; OR
isConcat(suffix) returns true
事实上,prefix 已经在 map 中(因为 map.has(prefix)
必须为真,才能首先到达递归调用!)。因此,只有当后缀也是 map 的元素或者它是另一个连接词时,连接词才有效。 map.has(suffix)
测试后缀是否在 map 中,而 isConcat(suffix,map,memo)
测试后缀是否也是连接词。