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

来源：力扣（LeetCode）

解题思路

  如题目所言，为每个单词建立映射表，然后通过映射后的单词和pattern对比来判断当前单词是否符合条件。

class Solution:
    def findAndReplacePattern(self, words: List[str], pattern: str) -> List[str]:
        def build_dict(a,b):  #建立映射表
            if len(set(a))!=len(set(b)):  #如果单词中字母的种类数目不对则不能建立映射表
                return False
            d={}
            for i,j in zip(a,b):
                d[i]=j
            return d
        ans=[]
        for word in words:
            d=build_dict(word,pattern) #为每个单词建立映射表
            if d:
                if ''.join(map(lambda x:d[x],word))==pattern:
                    ans.append(word)
        return ans
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16