Leetcode 890.查找和替换模式
1 题目描述(Leetcode题目链接)
你有一个单词列表 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
2 题解
这里使用一一映射,将每个字符串的字符与字符在字符串中第一次出现的下标映射起来。例如 a b b abb abb模式的单词都会映射为 011 011 011。
class Solution:
def findAndReplacePattern(self, words: List[str], pattern: str) -> List[str]:
res = []
pat = list(map(pattern.index, pattern))
for word in words:
if list(map(word.index, word)) == pat:
res.append(word)
return res