126. 单词接龙 II

126. 单词接龙 II

给定两个单词（beginWord 和 endWord）和一个字典 wordList，找出所有从 beginWord 到 endWord 的最短转换序列。转换需遵循如下规则：

每次转换只能改变一个字母。 转换过程中的中间单词必须是字典中的单词。
说明:
如果不存在这样的转换序列，返回一个空列表。 所有单词具有相同的长度。 所有单词只由小写字母组成。 字典中不存在重复的单词。 你可以假设beginWord 和 endWord 是非空的，且二者不相同。
示例 1:

输入: beginWord = “hit”, endWord = “cog”, wordList =[“hot”,“dot”,“dog”,“lot”,“log”,“cog”]

输出:
[
[“hit”,“hot”,“dot”,“dog”,“cog”],
[“hit”,“hot”,“lot”,“log”,“cog”]
]
示例 2:

输入: beginWord = “hit” endWord = “cog” wordList =[“hot”,“dot”,“dog”,“lot”,“log”]

输出: []

解释: endWord “cog” 不在字典中，所以不存在符合要求的转换序列。

思路

最短路径想到了BFS，可以通过构建图来描述上述单词转换关系，每个单词作为图的每个点，只要两个单词之间只差一个字母，那么这两个点就有一条双向边，建立了图映射后，就可以在起点单词中使用BFS来进行搜索最短路径了。

代码

关键数据结构：
edges是存储图边关系的邻接表
wordId存储了word到id的映射，使用了哈系表
idWord存储了id到word的映射，直接用数组即可

const int INF = 1 << 20;

class Solution {
private:
    vector<string> idWord;
    vector<vector<int>> edges;
    unordered_map<string, int> wordId;
public:
    vector<vector<string>> findLadders(string beginWord, string endWord, vector<string>& wordList) {
        int id = 0;
        //建立word和id之间的映射
        for(const string& word : wordList) {
            wordId[word] = id++;
            idWord.push_back(word);
        }
        //endWord不在单词表中直接返回空
        if(!wordId.count(endWord)) {
            return {};
        }
        //beginWord如果不在单词表中需要加入
        if(!wordId.count(beginWord)) {
            wordId[beginWord] = id++;
            idWord.push_back(beginWord);
        }
        edges.resize(idWord.size());
        //建立单词到图的映射
        for(int i = 0; i < idWord.size(); i++) {
            for(int j = i + 1; j < idWord.size(); j++) {
                //两个单词只差一个字符，建立双向边
                if(check(idWord[i], idWord[j])) {
                    edges[i].push_back(j);
                    edges[j].push_back(i);
                }
            }
        }
		
        const int dest = wordId[endWord];
        vector<vector<string>> ans;
        queue<vector<int>> q;
        vector<int> cost(id, INF);
        //初始点入队列
        q.push(vector<int>{wordId[beginWord]});
        //初始点消费为0
        cost[wordId[beginWord]] = 0;

        while(!q.empty()) {
        	//取出队列的头部
            vector<int> now = q.front();
            q.pop();
            //队列头部容器的尾部元素存储着这次路径的最终点
            int last = now.back();
            //最终点是终点
            if(last == dest) {
            	//将容器存储的单词index映射为单词
                vector<string> tmp;
                for(const int& index : now) {
                    tmp.push_back(idWord[index]);
                }
                ans.push_back(tmp);
            }
            else {
            	//对路径的最终点的所有邻接点进行访问
                for(int i = 0; i < edges[last].size(); i++) {
                    int to = edges[last][i];
                    //如果到达邻接点的消费大于等于cost[last] + 1证明该点未被访问
                    if(cost[last] + 1 <= cost[to]) {
                        cost[to] = cost[last] + 1;
                        vector<int> vv(now);
                        vv.push_back(to);
                        q.push(vv);
                    }
                }
            }
        }
        return ans;
    }

	//验证单词差异
    bool check(const string& s1,  const string& s2) {
        int different = 0;
        for(int i = 0; i < s1.size() && different < 2; i++) {
            if(s1[i] != s2[i]) {
                different++;
            }
        }
        return different == 1;
    }
};