LeetCode—— 290 单词规律

问题描述

给定一种规律 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

执行结果

代码描述

思路：

1. 将字符串分割成单词，存储在vector中，
2. 如果vector的大小，跟pattern的字符个数，不等，则返回false。
3. 将字符当做键，分割好的字符串当做值，进行判断。
4. 同时，将分割好的字符串当做键，将字符当做值，进行判断。

class Solution {
public:
    bool wordPattern(string pattern, string str) {
        vector<string> vec;
        vec = split(str);
        if(vec.size() !=pattern.size())    return false;
        
        map<string, char> s_c;
        map<char, string> c_s;
        for(int i = 0; i < pattern.size(); ++i)
        {
            if(s_c.count(vec[i]) == 0)
                s_c[vec[i]] = pattern[i];
            else
            {
                if(s_c[vec[i]] != pattern[i])
                    return false;
            }
            if(c_s.count(pattern[i]) == 0)
                c_s[pattern[i]] = vec[i];
            else
            {
                if(c_s[pattern[i]] != vec[i])
                    return false;
            }
        }
        return true;
    }
    vector<string> split(string str)
    {
        vector<string> res;
        string s;
        for(int i = 0; i < str.size(); ++i)
        {
            if(isspace(str[i]))
            {
                if(s.empty())
                    continue;
                res.push_back(s);
                s.clear();
            }
            else
                s += str[i];
        }
        if(!s.empty())
            res.push_back(s);
        return res;
    }
};