leetcode第一刷_Word Ladder II

来源:互联网 发布:linux链接目的 编辑:程序博客网 时间:2024/06/02 21:19

这道题很难。

之前的题目我提到过一次用两个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;    }};


0 0