LeetCode : Word Pattern

最新推荐文章于 2021-06-09 11:27:34 发布

原创最新推荐文章于 2021-06-09 11:27:34 发布 · 255 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

25 篇文章

订阅专栏

本文介绍了一种用于字符串模式匹配的算法实现，通过双向映射确保pattern与字符串子串一一对应，实现了pattern到字符串及字符串到pattern的有效转换。

pattern和substr的对应应该是双向的，因此设置双向的比较。

Solution

bool wordPattern(string pattern, string str)
{
    map<char, string> m;
    map<string, char> p;

    int begin = 0, end = 0;
    int index = 0;
    string token;

    while(index < pattern.length() && end <= str.length())
    {
        if(end == str.length() || str[end] == ' ')
        {           
            token.assign(str, begin, end - begin);

            // token -> pattern
            if(m.find(pattern[index]) == m.end())
                m[pattern[index]] = token;
            else if(m[pattern[index]] != token)
                return false;

            // pattern -> token
            if(p.find(token) == p.end())
                p[token] = pattern[index];
            else if(p[token] != pattern[index])
                return false;

            begin = end + 1;
            index++;
        }
        end++;
    }

    if(index != pattern.length() || end != str.length() + 1)
        return false;

    return true;
}