[LeetCode 127] - 单词梯(Word Ladder)

问题

给出两个单词（start和end）与一个字典，找出从start到end的最短转换序列的长度。规则如下：

• 一次只能改变一个字母
• 中间单词必须在字典里存在

例如：

给出

start = "hit"
end = "cog"
dict = ["hot","dot","dog","lot","log"]

因为其中一条最短的转换序列为"hit" -> "hot" -> "dot" -> "dog" -> "cog"，返回它的长度5。

注意

• 如果不存在转换序列，返回0
• 所有单词的长度一样
• 所有单词中只有小写字母

 

初始思路

有了单词梯II的分析，本题就很容易实现了。这里我们选择双vector的方案来进行改造。需要注意的几点：

• 只需要返回最短序列的长度，找到一个就可以返回了
• 同样由于只需返回长度，不需要像单词梯II那样每过一层才能把用过的单词去掉，使用一次就可以去除了。
• 每处理完一个vector表示一层处理结束，对当前序列长度加1

改造后的代码如下：

class Solution {
public:
    int ladderLength(std::string start, std::string end, std::unordered_set<std::string> &dict)
    {
        std::vector<std::unordered_set<std::string>> candidates(2);
        
        int current = 0;
        int previous = 1;
        int ladderLength = 1;
        
        candidates[current].insert(start);
        dict.erase(start);
        
        while(true)
        {
            current = !current;
            previous = !previous;
            
            candidates[current].clear();
            
            for(auto iter = candidates[previous].begin(); iter != candidates[previous].end(); ++iter)
            {
                for(size_t pos = 0; pos < iter->size(); ++pos)
                {
                    std::string word = *iter;
                    for(int i = 'a'; i <= 'z'; ++i)
                    {
                        if(word[pos] == i)
                        {
                            continue;
                        }
                        
                        word[pos] = i;
                        
                        if(dict.count(word) > 0)
                        {
                            if(word == end)
                            {
                                return ladderLength + 1;
                            }
                            
                            candidates[current].insert(word);
                            dict.erase(word);
                        }
                    }
                }
            }
            ++ladderLength;
            
            if (candidates[current].size() == 0)
            {
                return 0;
            }
        }
        
        return 0;
    }
};

提交后一次通过Judge Small和Judge Large。