持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第12天,点击查看活动详情
你有一个单词列表 words 和一个模式 pattern,你想知道 words 中的哪些单词与模式匹配。
如果存在字母的排列 p ,使得将模式中的每个字母 x 替换为 p(x) 之后,我们就得到了所需的单词,那么单词与模式是匹配的。
(回想一下,字母的排列是从字母到字母的双射:每个字母映射到另一个字母,没有两个字母映射到同一个字母。)
返回 words 中与给定模式匹配的单词列表。
你可以按任何顺序返回答案。
示例:
输入:words = ["abc","deq","mee","aqq","dkd","ccc"], pattern = "abb"
输出:["mee","aqq"]
解释:
"mee" 与模式匹配,因为存在排列 {a -> m, b -> e, ...}。
"ccc" 与模式不匹配,因为 {a -> c, b -> c, ...} 不是排列。
因为 a 和 b 映射到同一个字母。
提示:
- 1 <= words.length <= 50
- 1 <= pattern.length = words[i].length <= 20
构造双射
我们可以逐个判断 中的每个单词 是否与 匹配。
根据题意,我们需要构造从字母到字母的双射,即 的每个字母需要映射到 的对应字母,并且 的每个字母也需要映射到 的对应字母。
我们可以编写一个函数 ,仅当 中相同字母映射到 中的相同字母时返回 。我们可以在遍历这两个字符串的同时,用一个哈希表记录 的每个字母 x 需要映射到 的哪个字母上,如果 x 已有映射,则需要检查对应字母是否相同。
如果 和 均为 ,则表示 与 匹配。
var findAndReplacePattern = function(words, pattern) {
const ans = [];
for (const word of words) {
if (match(word, pattern) && match(pattern, word)) {
ans.push(word);
}
}
return ans;
};
const match = (word, pattern) => {
const map = new Map();
for (let i = 0; i < word.length; ++i) {
const x = word[i], y = pattern[i];
if (!map.has(x)) {
map.set(x, y);
} else if (map.get(x) !== y) { // word 中的同一字母必须映射到 pattern 中的同一字母上
return false;
}
}
return true;
}