前言:剑指offer刷题系列
问题:
给定一种规律 pattern
和一个字符串 s
,判断 s
是否遵循相同的规律。
这里的 遵循 指完全匹配,例如, pattern
里的每个字母和字符串 s
中的每个非空单词之间存在着双向连接的对应规律。
示例:
输入: pattern = "abba", s = "dog cat cat dog"
输出: true
思路:
这道题是上一道题的升级版,由字符之间的一一映射升级成了字符与字符串之间的一一映射。 首先本质是一样的,要实现一一映射,就要用到两个哈希表分别记录字符到字符串的映射和字符串到字符的映射。 其次,我们要对s中的单词进行提取,比较单词数量和pattern中的数量是否一致,如果数量上不一致,二者一定不匹配。
在下面的方法中,首先创建了两个空字典p2s
和s2p
,分别用于存储模式字符到单词和单词到模式字符的映射关系。
接下来,将字符串s
按空格分割成单词列表words
。如果模式字符串的长度与单词列表的长度不相等,则直接返回False
,表示不满足条件。
然后,使用zip
函数将模式字符串和单词列表中的每个字符和对应的单词一一对应起来,并使用循环遍历这些对应的字符和单词对。
在每次循环中,通过检查当前字符和单词是否满足以下条件来判断它们是否符合单词模式:
- 如果模式字符
ch
已经在字典p2s
中存在,并且其对应的值不等于当前单词word
,或者当前单词word
已经在字典s2p
中存在,并且其对应的值不等于模式字符ch
,则说明不满足条件,返回False
。 - 否则,将模式字符
ch
和当前单词word
添加到相应的字典中,建立映射关系。
最后,如果循环结束后没有提前返回False
,则说明所有的字符和单词都满足单词模式的条件,返回True
。
基于上述思考,代码如下:
class Solution:
def wordPattern(self, pattern: str, s: str) -> bool:
p2s = {}
s2p = {}
words = s.split(" ")
if len(pattern) != len(words):
return False
for ch, word in zip(pattern, words):
if (ch in p2s and p2s[ch] != word) or (word in s2p and s2p[word] != ch):
return False
p2s[ch] = word
s2p[word] = ch
return True