给定一种规律 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 包含了由单个空格分隔的小写字母。
思路:hash
建立两张hash表,存储映射(字符–>单词)和(单词–>字符)的映射关系。
枚举每一段字符与字符串,不断更新hash表,如果发生了冲突,则说明不满足映射关系,返回false。
注意判断字符个数与单词个数是否匹配,不匹配返回false。
bool wordPattern(string pattern, string s) {
unordered_map<string, char> str_ch;
unordered_map<char, string> ch_str;
int m = s.size(), i = 0;
for (char c : pattern) {
if (i >= m) {
//单词个数少于字符个数
return false;
}
int j = i;
while (j < m && s[j] != ' ') {
j++;
}
string tmp = s.substr(i, j - i);
if (str_ch.find(tmp) != str_ch.end() && str_ch[tmp] != c) {
return false;
}
if (ch_str.find(c) != ch_str.end() && ch_str[c] != tmp) {
return false;
}
str_ch[tmp] = c;
ch_str[c] = tmp;
i = j + 1;
}
if (i < m) {
//字符个数少于单词个数
return false;
}
return true;
}