890.查找和替换模式

890.查找和替换模式

你有一个单词列表 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

思路

先把pattern转成统一形式，这里用函数transfor设定了一个同一形式：当出现非重复项时，其值为当前索引值，若为重复项的时候，其值为第一次出现该项的索引值；后面只要将words中的字符串一个个转为统一形式再对比就可以了。

class Solution:
    def findAndReplacePattern(self, words, pattern):
        """
        :type words: List[str]
        :type pattern: str
        :rtype: List[str]
        """
        res = []
        p0 = self.transfor(pattern)
        for item in words:
            if self.transfor(item) == p0:
                res.append(item)
        return res
    
    
    def transfor(self,word):
        res = ""
        for index,item in enumerate(word):
            if res == "":
                res += str(index)
            elif word[0:index].find(item) != -1:
                res += str(word[0:index].find(item))
            else:
                res += str(index)
        return res