Leetcode290-Word Pattern

题目描述

给定一种 pattern(模式) 和一个字符串 str ，判断 str 是否遵循相同的模式。
这里的遵循指完全匹配，例如 pattern 里的每个字母和字符串 str 中的每个非空单词之间存在着双向连接的对应模式。

示例一

输入: pattern = “abba”, str = “dog cat cat dog”
输出: true

示例二

输入:pattern = “abba”, str = “dog cat cat fish”
输出: false

示例三

输入: pattern = “aaaa”, str = “dog cat cat dog”
输出: false

思路

用两个哈希表，(这里必须用两个哈希表，这样才能确保相同字母对应相同单词，且不同字母对应不同单词。)分别存储单词对应的字母，以及字母对应的单词。
然后从前往后扫描，判断相同元素对应的值，是否是相同的。

代码

class Solution {
public:
    bool wordPattern(string pattern, string str) {
    stringstream ss(str);
    vector<string>strs;
    map<char,string>ps;
    map<string,int>sp;
    string s;
    while(ss>>s)
        strs.push_back(s);
    if (pattern.size() != strs.size()) return false;//如果大小不等，直接返回
    for(int i=0;i<strs.size();i++)
    {
        if(!ps.count(pattern[i]))
            ps[pattern[i]]=strs[i];
        if(!sp.count(strs[i]))
            sp[strs[i]]=pattern[i];
        if(ps[pattern[i]]!=strs[i])//相同字母对应相同单词
            return false;
        if(sp[strs[i]]!=pattern[i])//不同字母对应不同单词
            return false;
    }
    return true;
    }
};