[leetcode] 126. Word Ladder II
原创
©著作权归作者所有:来自51CTO博客作者是念的原创作品,请联系作者获取转载授权,否则将追究法律责任
Description
Given two words (beginWord and endWord), and a dictionary’s word list, find all shortest transformation sequence(s) from beginWord to endWord, such that:
- Only one letter can be changed at a time
- Each transformed word must exist in the word list. Note that beginWord is not a transformed word.
Note:
- Return an empty list if there is no such transformation sequence.
- All words have the same length.
- All words contain only lowercase alphabetic characters.
- You may assume no duplicates in the word list.
- You may assume beginWord and endWord are non-empty and are not the same.
Example 1:
Input:
beginWord = "hit",
endWord = "cog",
wordList = ["hot","dot","dog","lot","log","cog"]
Output:
[
["hit","hot","dot","dog","cog"],
["hit","hot","lot","log","cog"]
]
Example 2:
Input:
beginWord = "hit"
endWord = "cog"
wordList = ["hot","dot","dog","lot","log"]
Output:
Explanation:
The endWord "cog" is not in wordList, therefore no possible
transformation.
分析
题目的意思是:给定一个开始单词,结束单词和一个单词词典,找出以开始单词为头,以结束单词为尾的所有最短路径。
- 题目的目的是找出所有的路径,建立一个路径集paths,用以保存所有路径,然后是起始路径p,在p中先把起始单词放进去。然后定义两个整型变量level,和minLevel,其中level是记录循环中当前路径的长度,minLevel是记录最短路径的长度,这样的好处是,如果某条路径的长度超过了已有的最短路径的长度,那么舍弃,这样会提高运行速度,相当于一种剪枝。
- 还要定义一个set变量words,用来记录已经循环过的路径中的词,然后就是BFS的核心了,循环路径集paths里的内容,取出队首路径,如果该路径长度大于level,说明字典中的有些词已经存入路径了,如果在路径中重复出现,则肯定不是最短路径,所以我们需要在字典中将这些词删去,然后将words清空,对循环对剪枝处理。然后我们取出当前路径的最后一个词,对每个字母进行替换并在字典中查找是否存在替换后的新词,这个过程在之前那道Word Ladder 词语阶梯里面也有。
代码
class Solution {
public:
vector<vector<string>> findLadders(string beginWord, string endWord, vector<string>& wordList) {
vector<vector<string>> res;
unordered_set<string> dict(wordList.begin(),wordList.end());
vector<string> p{beginWord};
queue<vector<string>> paths;
paths.push(p);
int level=1;
int minLevel=INT_MAX;
unordered_set<string> words;
while(!paths.empty()){
auto t=paths.front();
paths.pop();
if(t.size()>level){
for(string w:words) dict.erase(w);
words.clear();
level=t.size();
if(level>minLevel) break;
}
string last=t.back();
for(int i=0;i<last.size();i++){
string newLast=last;
for(char ch='a';ch<='z';ch++){
newLast[i]=ch;
if(!dict.count(newLast)){
continue;
}
words.insert(newLast);
vector<string> nextPath=t;
nextPath.push_back(newLast);
if(newLast==endWord){
res.push_back(nextPath);
minLevel=level;
}else{
paths.push(nextPath);
}
}
}
}
return res;
}
};
参考文献
[LeetCode] Word Ladder II 词语阶梯之二