leetcode第一刷_Word Ladder II

这道题很难。

之前的题目我提到过一次用两个vector来做层序遍历的,就是因为这道题。要想最后恢复出单词变换的路径,就需要事先保存,根据dp中路径恢复的启发,保存的应该是一个单词的前一个变换节点。可能有很多个单词都能变换到当前单词,因此应该是一个set。用一个二维的vector保存当前可以变换到的单词和变换出这些单词单词。每一维的vector存放的都是一个set。设存放当前可访问单词的vector下标是current,存放变幻出这些单词的vector下标是pre,那么每一轮要开始更新current之前,都要从字典里删除pre中存放的单词。这样做比直接用一个set存放访问过的单词要快,也更加节省空间,因为字典的规模在不断变小。因为要打印出所有的路径,所以不是说遇到目标单词就直接结束,而应该等着这一层的变换全部结束之后才行。如果在这一层变换之后发现了目标单词,那么就开始打印路径。

递归打印路劲跟其他题目的写法没有太多不同,主要是注意如果用引用做的话要合适的回退,且最后是反向打印的,应该把结果reverse一下再放到结果集中。

实现的时候还发现一个问题,我之前一直以为在访问方面vector应该是最快的,所以就把所有东西都弄成了vector,然后下标访问,结果超时了。如果要顺序访问的话,迭代器要比随机访问好。

class Solution {
public:
    vector<vector<string> > res;
    void getPaths(unordered_map<string, vector<string> > &predict, vector<string> &tpres, string &word){
        if(predict[word].size() == 0){
            tpres.push_back(word);
            vector<string> pres = tpres;
            reverse(pres.begin(), pres.end());
            res.push_back(pres);
            tpres.pop_back();
            return;
        }
        tpres.push_back(word);
        for(auto it=predict[word].begin();it!=predict[word].end();++it)
            getPaths(predict, tpres, *it);
        tpres.pop_back();
    }
    vector<vector<string> > findLadders(string start, string end, unordered_set<string> &dict) {
        unordered_map<string, vector<string> > predict;
        int cu = 0, pre = 1, len = start.length();
        vector<unordered_set<string> > candi(2); 
        vector<string> tpres;
        predict[start] = tpres;
        candi[0].insert(start);
        string word;
        while(true){
            cu = !cu;
            pre = !pre;
            for(auto it=candi[pre].begin();it!=candi[pre].end();++it){
                dict.erase(*it);
            }
            candi[cu].clear();
            for(auto it=candi[pre].begin();it!=candi[pre].end();++it){
                for(int pos=0;pos<len;pos++){
                    word = *it;
                    for(char c='a';c<='z';c++){
                        if(word[pos] == c)  continue;
                        word[pos] = c;
                        if(dict.count(word)>0){
                            candi[cu].insert(word);
                            predict[word].push_back(*it);
                        }
                    }
                }
            }
            if(candi[cu].size() == 0)   return res;
            if(candi[cu].count(end))
                break;
        }
        getPaths(predict, tpres, end);
        return res;
    }
};
点赞