[Leetcode]Word Ladder

单词梯宽搜算法

最新推荐文章于 2018-06-12 10:14:22 发布

原创最新推荐文章于 2018-06-12 10:14:22 发布 · 417 阅读

0 ·

CC 4.0 BY-SA版权

DFS&BFS 专栏收录该内容

2 篇文章

订阅专栏

本文介绍了一种使用宽度优先搜索解决单词梯问题的方法。通过遍历字符串并替换字符来寻找从开始单词到结束单词的最短路径。文章提供了两种实现方案，一种是在访问节点后标记为已访问，另一种是将已访问的节点从字典中删除。

思路：宽搜

class Solution {
public:
    int ladderLength(string start, string end, unordered_set<string> &dict) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        unordered_set<string> visited;
        queue<pair<string,int>> q;
        q.push(make_pair(start, 1));
        visited.insert(start);
        while(!q.empty())
        {
            string curStr = q.front().first;
            int curLen = q.front().second;
            q.pop();
            for(string::size_type i = 0; i < curStr.size(); ++i)
            {
                string tmp = curStr;
                for(int j = 0; j < 26; ++j)
                {
                    tmp[i] = j + 'a';
                    if(tmp == end) return curLen + 1;
                    if(dict.count(tmp) && !visited.count(tmp))
                    {
                        q.push(make_pair(tmp, curLen+1));
                        visited.insert(tmp);
                    }
                }
            }
        }
        return 0;
    }
};

优化：直接把已经访问过的元素删除

class Solution {
public:
    int ladderLength(string start, string end, unordered_set<string> &dict) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        queue<pair<string,int>> q;
        q.push(make_pair(start, 1));
        dict.erase(start);
        while(!q.empty())
        {
            string curStr = q.front().first;
            int curLen = q.front().second;
            q.pop();
            for(string::size_type i = 0; i < curStr.size(); ++i)
            {
                string tmp = curStr;
                for(int j = 0; j < 26; ++j)
                {
                    tmp[i] = j + 'a';
                    if(tmp == end) return curLen + 1;
                    if(dict.count(tmp))
                    {
                        q.push(make_pair(tmp, curLen+1));
                        dict.erase(tmp);
                    }
                }
            }
        }
        return 0;
    }
};