【LeetCode 290】关关的刷题日记28 Word Pattern- 专知

会员服务 ·

【LeetCode 290】关关的刷题日记28 Word Pattern

关关

点击上方“专知”关注获取更多AI知识!

关关的刷题日记28 – Leetcode 290. Word Pattern

题目

Given a pattern and a string str, find if str follows the same pattern.

Here follow means a full match, such that there is a bijection between a letter in pattern and a non-empty word in str.

Examples: pattern = "abba", str = "dog cat cat dog" should return true. pattern = "abba", str = "dog cat cat fish" should return false. pattern = "aaaa", str = "dog cat cat dog" should return false. pattern = "abba", str = "dog dog dog dog" should return false. Notes: You may assume pattern contains only lowercase letters, and str contains lowercase letters separated by a single space.

题目的意思是给定两个字符串pattern和str，判断str是否和pattern的模式一样。pattern中的每个字符对应着str中的每个单词，单词之间用空格来分隔。

思路

思路：建立一个map，key存pattern中的每个字符，value存str中的每个单词，看看是否能一一对应上。不过这题目坑比较多，一方面可能pattern中的字符数和str中的word数个数不一样，要格外判断一下。另一方面有可能出现题目中给出的第四个例子那种多对一的情况，为了把这种情况挑出来，我额外设置了一个set，但是感觉做法不够精简。

class Solution {public:
    bool wordPattern(string pattern, string str) {
        map<char,string>m;
        vector<string>store;
        set<string>t;
        stringstream ss(str);
        string temp;
        while(getline(ss,temp,' '))
        {
            store.push_back(temp);
            t.insert(temp);
        }
        if(store.size()!=pattern.size())
            return false;
        for(int i=0; i<pattern.size(); i++)
        {
            if(m.find(pattern[i])==m.end())
                m[pattern[i]]=store[i];
            else
            {
                if(m[pattern[i]]!=store[i])
                    return false;
            }                
        }
        if(m.size()!=t.size())
           return false;
        return true;
    }};