LintCode-829. Word Pattern II (DFS难题)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/roufoo/article/details/81050718

这题挺难的,我也是参考的网上的解法,这个解法可以算是一种DFS暴力遍历所有可能解把。
注意 :
1)在helper()里面,如果调用的helper()返回false,不能直接返回false,而是要继续处理其他的solution。注意这个跟其他返回vector< vector<> >的情况不一样,因为这里只是返回true/false。
2) map和set没有pop_back()操作,只能用erase。

class Solution {
public:
    /**
     * @param pattern: a string,denote pattern string
     * @param str: a string, denote matching string
     * @return: a boolean
     */
    bool wordPatternMatch(string &pattern, string &str) {
        map<char, string> m;
        set<string> s;
        return helper(pattern, str, m, s);
    }


private:
    bool beginWith(string const &str, string const &subStr) {
        if (subStr.size() > str.size()) 
            return false;

        if (subStr == str.substr(0, subStr.size())) 
            return true;

        return false;
    }

    bool helper(string pat, string str, map<char, string> &m, set<string> &s) {
        if (pat.size() == 0)
            return str.size() == 0;

        char c = pat[0];

        if (m.find(c) != m.end()) { //如果c已经存在,则看str是不是以c对应字符串开头
            if (beginWith(str, m[c])) {
                return helper(pat.substr(1), str.substr(m[c].size()), m, s);
            }
            return false;
        }

        //如果pat[0]是一个新的char, 则要试探性的把str[0], str[0..1], ..., str[0..n]跟pat[0]匹配
        //这里要用到set,表示这个str[]已经被用
        for (int i = 0; i < str.size(); ++i) {
            string subWord = str.substr(0, i + 1);
            if (s.find(subWord) == s.end()) { //如果subWord是新的
                s.insert(subWord);
                m[c] = subWord; //也可以m.insert(make_pair(c, subWord));

                //注意:这里不能简单return,因为如果helper()是false, 其他solution仍然可能对
                //return helper(pat.substr(1), str.substr(i + 1), m, s);
                if (helper(pat.substr(1), str.substr(i + 1), m, s))
                    return true;

                s.erase(subWord);     //set和map没有pop_back()操作!!!
                m.erase(c);
            }    
        }

        return false;
      }
};

猜你喜欢

转载自blog.csdn.net/roufoo/article/details/81050718
今日推荐