难度:中等
你有一个单词列表 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 映射到同一个字母。
/**
* 构建双射
* @desc 时间复杂度 O(MN) 空间复杂度 O(N)
* @param words
* @param pattern
* @returns
*/
export function findAndReplacePattern(words: string[], pattern: string): string[] {
const ans: string[] = []
for (const word of words)
if (match(word, pattern) && match(pattern, word)) ans.push(word)
return ans
function match(word: string, pattern: string) {
const map = new Map<string, string>()
for (let i = 0; i < word.length; i++) {
const x = word[i]
const y = pattern[i]
if (!map.has(x))
map.set(x, y)
else if (map.get(x) !== y)
return false
}
return true
}
}