leetcode-单词模式（python）

题目：
给定一种 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 包含了由单个空格分隔的小写字母。

代码：

class Solution(object):
    def wordPattern(self, pattern, str):
        """
        :type pattern: str
        :type str: str
        :rtype: bool
        """
        str=str.split()
        dict={}  #pattern与str一一对应,pattern为key,str为value
        lists=[] #存储当前字典中的pattern
        if len(pattern)!=len(str): #长度不同
            return False
        for i in range(len(pattern)): #构造字典
            if pattern[i] in dict:    #key已经在字典中
                if str[i]!=dict[pattern[i]]: #判断value是否相同
                    return False
            else:                    #key不在字典中
                if str[i] in lists:  #判断value是否在字典中
                    return False    
                lists.append(str[i])   #存储字典中的value值
                dict[pattern[i]]=str[i]  #更新字典
        return True