Leetcode#126 Word Ladder II

原题地址

既然是求最短路径,可以考虑动归或广搜。这道题对字典直接进行动归是不现实的,因为字典里的单词非常多。只能选择广搜了。

思路也非常直观,从start或end开始,不断加入所有可到达的单词,直到最终到达另一端。本质上广度优先遍历图。

需要注意的是,拓展下一个单词时不能对字典进行枚举,因为字典里的单词太多。幸好单词本身都不长,所以直接枚举单词所有可能的变形,看看在dict中出现没有。

当然还不止这些,上面的做法仍然会超时,需要添加剪枝策略:

1. 如果某个单词在以前遍历过了,那么以后都不用再考虑,因为之后遍历到的路径一定不是最短的

2. 在广搜法拓展下一轮单词时,注意去重

此外还需要注意的是,不能把每个单词到start或end的路径都保存下来,那样内存会爆掉。所以要压缩保存结果,通常的做法是用一个map保存当前单词下一步是什么单词。例如next[word] = {next_word1, next_word2, next_word3...}。最后从next[start]开始再次使用广度优先搜索法构造出所有解。

算法不难,但是编码非常容易出错,所以总体上还是挺难的。最后运行时间640ms,还是有挺大优化空间的。

代码写的有些啰嗦,DFS不一定要用队列(我还是用了队列),这道题用unordered_set更好,不需要用额外的数据结构去重了。

代码:

 bool adjacentp(string &a, string &b) {
for (int i = a.length() - , d = ; i >= ; i--) {
d += a[i] != b[i] ? : ;
if (d > )
return false;
}
return true;
} vector<vector<string> > findLadders(string start, string end, unordered_set<string> &dict) {
map<string, set<string> > next;
unordered_set<string> covered; // 当前已经访问过的单词
queue<string> que;
bool found = false; // 是否已经找到 next[end] = set<string>();
covered.insert(end);
que.push(end);
while (!que.empty() && !found) {
unordered_set<string> rset;
queue<string> rque; while (!que.empty()) {
string curr = que.front();
que.pop(); if (adjacentp(curr, start)) {
found = true;
next[start].insert(curr);
continue;
} for (int i = curr.length() - ; i >= ; i--) {
for (int j = ; j < ; j++) {
string prev = curr;
prev[i] = 'a' + j;
// 如果prev之前没有被访问过,且字典里有这个单词
if (covered.find(prev) == covered.end() && dict.find(prev) != dict.end()) {
next[prev].insert(curr);
// 如果在DFS的本轮拓展中还没有访问过该节点,则加入下一轮的拓展节点中
if (rset.find(prev) == rset.end()) {
rset.insert(prev);
rque.push(prev);
}
}
}
}
}
que = rque;
for (auto w : rset) {
covered.insert(w);
}
} queue<vector<string> > laddersQue;
vector<vector<string> > ladders;
laddersQue.push(vector<string>(, start));
while (!laddersQue.empty()) {
vector<string> ladder = laddersQue.front();
laddersQue.pop();
if (ladder.back() == end)
ladders.push_back(ladder);
else {
for (auto s : next[ladder.back()]) {
vector<string> newLadder = ladder;
newLadder.push_back(s);
laddersQue.push(newLadder);
}
}
} return ladders;
}
上一篇:126. Word Ladder II(hard)


下一篇:leetcode 127. Word Ladder、126. Word Ladder II