Method findAndReplacePattern

String/FindAndReplacePattern.py:44–72

:type words: List[str] :type pattern: str :rtype: List[str]

(self, words, pattern)

class Solution(object):
    def findAndReplacePattern(self, words, pattern):
        """
        :type words: List[str]
        :type pattern: str
        :rtype: List[str]
        """
        
        def translate_pattern(word):
            result = ['0']
            current = '0'
            patterns = {word[0]: '0'} 
            for i in word[1:]:
                if patterns.get(i) is not None:
                    result.append(patterns.get(i))
                else:
                    current = str(int(current)+1)
                    patterns[i] = current
                    result.append(current)

            return ''.join(result)
        
        x = translate_pattern(pattern)

        result = []
        for i in words:
            if x == translate_pattern(i):
                result.append(i)
        
        return result