leetcode刷题290单词规律

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

class Solution:
    def wordPattern(self, pattern: str, s: str) -> bool:
        '''
        使用字典存储，key由pattern构成，values由s构成。
        '''
        words = s.split(' ')
        if len(pattern) != len(words):
            return False
        dic = {}
        for i in range(len(pattern)):
            if pattern[i] not in dic.keys():
                if words[i] not in dic.values():
                    dic[pattern[i]] = words[i]
                else:
                    return False
            else:
                if dic[pattern[i]] != words[i]:
                    return False
        return True