《LeetCode之每日一题》:260.单词规律

题目链接： 单词规律

有关题目

给定一种规律 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 包含了由单个空格分隔的小写字母。

题解

法一：哈希表
参考官方题解

class Solution {
public:
    bool wordPattern(string pattern, string str) {
        unordered_map<char, string> ch2str;
        unordered_map<string, char> str2ch;

        int m = str.length();
        int i = 0;
        for (auto &ch : pattern)
        {
            //s 中单词数量不够pattern中的字符 数量
            if (i >= m)
                return false;
            
            int j = i;
            while(j < m && str[j] != ' ') j++;

            const string &temp = str.substr(i, j - i);
            //不满足双向连接的关系直接返回错误
            if (ch2str.count(ch) && ch2str[ch] != temp)
                return false;
            if (str2ch.count(temp) && str2ch[temp] != ch) 
                return false;
            
            ch2str[ch] = temp;
            str2ch[temp] = ch;

            i = j + 1;//指向str中的下一个单词为位置的起始位置
        }

        return i >= m;//pattern中的字符数量应该恰好等于str中的单词个数
    }
};