【leetcode❤python】 290. Word Pattern

时间:2022-01-27 17:07:32

#-*- coding: UTF-8 -*-
class Solution(object):
    def wordPattern(self, pattern, str):
        """
        :type pattern: str
        :type str: str
        :rtype: bool
        """
        tag=0
        tagdic={}
        tagList=[]
        i=0
        while i<len(pattern):
            if tagdic.has_key(pattern[i]):
                tagList.append(tagdic.get(pattern[i]))
            else:
                tagdic.setdefault(pattern[i],tag)
                tagList.append(tag)
                tag+=1
            i+=1
        strList=str.split(' ')
   
        tagdic={};tag=0;tagList2=[];i=0
   
        while i<len(strList):
         
            if tagdic.has_key(strList[i]):
                tagList2.append(tagdic.get(strList[i]))
            else:
                tagdic.setdefault(strList[i],tag)
                tagList2.append(tag)
                tag+=1
            i+=1
        
        return tagList==tagList2

sol=Solution()
print sol.wordPattern('abba', 'dog cat cat dog')