1. 问题描述:
给定一种规律 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 包含了由单个空格分隔的小写字母。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/word-pattern
2. 思路分析:
① 分析题目可以知道需要判断每个字符与字符串str是否存在双射关系,所以使用两个哈希表(哈希表可以表示两个元素之间的映射关系)判断是否是一一对应的关系即可,因为使用的是python语言所以使用字典作为哈希表判断即可。
② 假设两个集合分别是s1和s2,双射:即是单射又是满射,单射:对于集合s1中的两个不同元素a1,a2,那么有f(a1) ≠ f(a2),满射的话其实很好判断对于集合s2中的元素在集合s1中都有元素与之对应,所以我们只需要判断字符串与单词长度是否相等即可判断是否是满射。此外还需要证明f是一个映射,也即需要满足集合s1中的元素能够在集合s2中找到一个唯一的元素与之对应即可。
3. 代码如下:
import collections
class Solution:
def wordPattern(self, pattern: str, s: str) -> bool:
s1 = s.split()
if len(pattern) != len(s1): return False
pw= collections.defaultdict(str)
wp= collections.defaultdict(str)
for i in range(len(pattern)):
a, b = pattern[i], s1[i]
if a in pw:
# 不是一个映射f
if pw[a] != b: return False
if b in wp:
# 不是一个单射
if wp[b] != a: return False
pw[a] = b
wp[b] = a
return True