Python LeetCode(290.单词模式)

Python LeetCode(290.单词模式)

给定一种 pattern(模式) 和一个字符串 str ，判断 str 是否遵循相同的模式。

这里的遵循指完全匹配，例如， pattern 里的每个字母和字符串 str 中的每个非空单词之间存在着双向连接的对应模式。

示例1:

输入: pattern = “abba”, str = “dog cat cat dog”
输出: true

示例 2:

输入:pattern = “abba”, str = “dog cat cat fish”
输出: false

示例 3:

输入: pattern = “aaaa”, str = “dog cat cat dog”
输出: false

示例 4:

输入: pattern = “abba”, str = “dog dog dog dog”
输出: false

说明:
你可以假设 pattern 只包含小写字母， str 包含了由单个空格分隔的小写字母。

Solution:(建立pattern和string的字典来判断ab形式，但要注意’abba’对应的’dog dog dog dog’情况，用字典判断会出错，使用set去重单独判断)

class Solution(object):
    def wordPattern(self, pattern, str):
        """
        :type pattern: str
        :type str: str
        :rtype: bool
        """
        str_list = str.split()
        if len(str_list) != len(pattern) or len(set(str_list)) != len(set(pattern)):
            return False
        
        d = {}
        for i in range(len(pattern)):
            if pattern[i] not in d:
                d[pattern[i]] = str_list[i]
            else:
                if d[pattern[i]] != str_list[i]:
                    return False
        return True

solution = Solution()
print(solution.wordPattern('abba', 'dog fish fish dog'))

True