码迷,mamicode.com
首页 > 其他好文 > 详细

leetcode Word Ladder II

时间:2014-12-06 00:00:39      阅读:416      评论:0      收藏:0      [点我收藏+]

标签:style   blog   http   io   ar   color   sp   for   on   

和上题 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();
        }
    }

};

 

leetcode Word Ladder II

标签:style   blog   http   io   ar   color   sp   for   on   

原文地址:http://www.cnblogs.com/higerzhang/p/4147667.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!