题目描述:
给定一种规律 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 包含了由单个空格分隔的小写字母。
解题思路:
将str按照空格分割放入sstr中,如果pattern的长度和sstr的长度不同,直接返回False
将pattern作为key,sstr作为value放入字典中,如果当前key在字典中出现过,判断字典中该key的value是否等于对应的值,不相等则返回False
如果当前key在字典中没有出现过,判断相应的value有没有在字典中出现过,出现过则返回False
如果都没有出现过,则添加到字典中
class Solution(object):
def wordPattern(self, pattern, str):
"""
:type pattern: str
:type str: str
:rtype: bool
"""
sstr=str.split(' ')
if len(sstr)!=len(pattern):
return False
dic={}
for i in range(len(pattern)):
if pattern[i] in dic:
if dic[pattern[i]]!=sstr[i]:
return False
elif sstr[i] in dic.values():
return False
else:
dic[pattern[i]]=sstr[i]
return True
评论区答案:
words = str.split(' ')
if len(words) != len(pattern):
return False
return map(pattern.find, pattern) == map(words.index, words)