leetcode Word Ladder II
和上题 Word Ladder I题目差不多,不过这里是要记录所有最段路径的可能。
不同点在于,我们不能再BFS每层的时候把相距一个字符的字符串在dict中删除,因为hot -> hit 的话其他的例如 jit -> hit 就是hit可以出现在两条路径里头。所以不能立马删除。但是我们发现我们可以删除的是我们遍历完的每层的字符串,我们用known来存遍历完的层,unknown来存没有遍历的层,那么每次求得下一层unknown的时候,就首先把known里面有的从dic中删除。
主要思路还是和上一题一样,我们还是每一层每一层的遍历考虑,知道最后的下一层出现end字符,或者是下一层为空了还没有出现end那就是返回空,没有匹配的路径。
关键点在于,因为要返回路径,所以我们需要记录路径,如何记录呢,我们用一个trace来记录每个str的上一个string,例如上例中trace[hit] = jit 和 hot,所以需要用数组,又因为索引是字符串,所以我们用map,那就用unordered_map<string, vector<string> >.
DFS127和buildTrace两个函数是利用深度优先递归出所有符合条件的答案的。当然如果再buildTrace中传入的bool flag是false的话,就说明没有符合路径,直接返回空。如果传入是1说明是有路径的,那么就利用简单的深度优先就可以了.
class Solution { public: //127 // 构造出解,利用dfs void DFS127(vector<vector<string> > &ans, unordered_map<string, vector<string> > &trace, vector<string> &tmp, string word) { tmp.push_back(word); if (trace[word].size() == 0) { vector<string> tmps = tmp; reverse(tmps.begin(), tmps.end()); ans.push_back(tmps); return ; } for (int i = 0; i < trace[word].size(); ++i) { DFS127(ans, trace, tmp, trace[word][i]); tmp.pop_back(); } } // 给定相应的路径参数,返回解,解有DFS127构造 vector<vector<string> > buildTrace(unordered_map<string, vector<string> > trace, bool flag, string end) { vector<vector<string> > ans; if (!flag) return ans; vector<string> tmp; DFS127(ans, trace, tmp, end); return ans; } // 主分支,就是隔一层判断,是否到达,如果没有到达就继续隔一层,计算下一层时要在dic中删除已知层的,因为已经走过了 vector<vector<string> > findLadders(string start, string end, unordered_set<string> &dict) { if (dict.count(start) == 0) dict.insert(start); if (dict.count(end) == 0) dict.insert(end); unordered_set<string> known, unknown; // known layer and next unknown layer unordered_map<string, vector<string> > trace; known.insert(start); while(1) { for (unordered_set<string>::const_iterator citr = known.begin(); citr != known.end(); ++citr) dict.erase(*citr); for (unordered_set<string>::const_iterator citr = known.begin(); citr != known.end(); ++citr) { string str = *citr; for (int i = 0; i < str.size(); ++i) for (char j = 'a'; j <= 'z'; ++j) { if (j == str[i]) continue; char tmpc = str[i]; str[i] = j; if (dict.count(str) > 0) { unknown.insert(str); trace[str].push_back(*citr); } str[i] = tmpc; } } if (unknown.size() == 0) return buildTrace(trace, 0, end); // 如果有为空的情况还没返回,就是无解,传入0 if (unknown.count(end) > 0) return buildTrace(trace, 1, end); // 不为空,那肯定是有解,传入1 known.clear(); known = unknown; unknown.clear(); } } };