290. 单词规律

给定一种规律 pattern 和一个字符串 s ，判断 s 是否遵循相同的规律。

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

 

示例1:

输入: pattern = "abba", s = "dog cat cat dog"
输出: true

示例 2:

输入:pattern = "abba", s = "dog cat cat fish"
输出: false

示例 3:

输入: pattern = "aaaa", s = "dog cat cat dog"
输出: false

 

提示:

• 1 <= pattern.length <= 300
• pattern 只包含小写英文字母
• 1 <= s.length <= 3000
• s 只包含小写英文字母和 ' '
• s 不包含 任何前导或尾随对空格
• s 中每个单词都被 单个空格 分隔

 

class Solution:
    def wordPattern(self, pattern: str, s: str) -> bool:

        t = s.split()
        s = pattern

        d1 = {}
        d2 = {}
        
        l1 = len(s)
        l2 = len(t)

        if l1!=l2:
            return False

        for i in range(l1):
            a = s[i]
            b = t[i]
            if a not in d1 and b not in d2:
                d1[a]=b
                d2[b]=a
            else:
                if a in d1 and d1[a]!=b:
                    return False
                if b in d2 and d2[b]!=a:
                    return False
        return True