描述
给定一种规律 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 包含了由单个空格分隔的小写字母。
链接:https://leetcode-cn.com/problems/word-pattern
思路
利用hash存储pattern的每个字母c和s每个单词v的对应关系,相同的c应该有相同的v,注意细节即可。
代码
class Solution {
public:
bool wordPattern(string pattern, string s) {
unordered_map<char, string> hash;
unordered_set<string> values;
int start = 0;
for (const char& c : pattern) {
int i = start;
while (i < s.size() && s[i] != ' ') i++;
if (i > s.size()) return false;
string v = s.substr(start, i - start);
start = i + 1;
if (hash.find(c) == hash.end()) {
if (values.find(v) != values.end()) return false;
hash[c] = v;
values.insert(v);
} else {
if (hash[c] != v) return false;
}
}
return start > s.size();
}
};
复杂度
时间复杂度:O(N)
空间复杂度:O(N)