【LeetCode 126】Word Ladder II

最新推荐文章于 2022-02-07 22:27:20 发布

涛涛酱

最新推荐文章于 2022-02-07 22:27:20 发布

阅读量81

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 搜索

本文链接：https://blog.youkuaiyun.com/iCode_girl/article/details/92838466

LeetCode 同时被 2 个专栏收录

295 篇文章

订阅专栏

搜索

88 篇文章

订阅专栏

博客围绕一道题展开，题目要求每次改变一个字符，中间单词需在字典出现，输出最短路径所有方案。解题思路是先用bfs建图保存最短路径结点的父亲结点，再用dfs从最后单词开始保存路径，还提到bfs的step在每层结束后加1，并给出代码情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述：

和125一样的题意，每次改变一个字符，中间单词只能是字典里出现过的，要求输出最短路径的所有方案。

思路：

先用bfs建图，保存所有最短路径里的每个结点的父亲结点。
然后用dfs，从最后的单词开始，保存路径。
这里的bfs和之前的不太一样，step在每一层结束以后+1。

代码：

class Solution {
public:
    vector<vector<string>> findLadders(string beginWord, string endWord, vector<string>& wordList) {
        set<string> st(wordList.begin(), wordList.end());
        map<string, int> step;
        if (!st.count(endWord)) return {};
        st.erase(beginWord);
        st.erase(endWord);
        step[beginWord] = 1;
        queue<string> que;
        que.push(beginWord);
        bool found = false;
        int stepcnt = 0;
        int len = beginWord.length();
        map<string, vector<string>> parents;
            
        while(!que.empty() && !found) {
            stepcnt++;
            for (int size=que.size(); size>0; size--) {
                string cur = que.front();
                que.pop();
                for (int j=0; j<len; ++j) {
                    string nxt = cur;
                    char ch = nxt[j];
                    for (int k=0; k<26; ++k) {
                        nxt[j] = char('a' + k);
                        if (ch == nxt[j]) continue;
                        if (nxt == endWord) {
                            found = true;
                            parents[endWord].push_back(cur);
                        }else if (step.count(nxt) && step[nxt] == stepcnt + 1) {
                            parents[nxt].push_back(cur);
                        }
                        if (st.count(nxt) == 0) continue;
                        st.erase(nxt);
                        que.push(nxt);
                        step[nxt] = step[cur] + 1;
                        parents[nxt].push_back(cur);
                    }
                }
            }
        }
        vector<vector<string> > ans;
        vector<string> path;
        path.push_back(endWord);
        dfs(endWord, beginWord, parents, path, ans);
        return ans;
    }
    
    void dfs(const string& endWord, const string& beginWord, map<string, vector<string>>& parents, vector<string>& path, vector<vector<string>>& ans) {
        if (endWord == beginWord) {
            ans.push_back(vector<string>(path.rbegin(), path.rend()));
            return;
        }
        for (auto nxtWord : parents[endWord]) {
            path.push_back(nxtWord);
            dfs(nxtWord, beginWord, parents, path, ans);
            path.pop_back();
        }
    }
};