Python LeetCode(290.单词模式)
给定一种 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 包含了由单个空格分隔的小写字母。
Solution:(建立pattern和string的字典来判断ab形式,但要注意’abba’对应的’dog dog dog dog’情况,用字典判断会出错,使用set去重单独判断)
class Solution(object):
def wordPattern(self, pattern, str):
"""
:type pattern: str
:type str: str
:rtype: bool
"""
str_list = str.split()
if len(str_list) != len(pattern) or len(set(str_list)) != len(set(pattern)):
return False
d = {}
for i in range(len(pattern)):
if pattern[i] not in d:
d[pattern[i]] = str_list[i]
else:
if d[pattern[i]] != str_list[i]:
return False
return True
solution = Solution()
print(solution.wordPattern('abba', 'dog fish fish dog'))
True