Word Ladder I & II (Leetcode 127

2016-12-06  本文已影响0人  stepsma

Word Ladder I:
https://leetcode.com/problems/word-ladder/

对于I,BFS层序遍历就行。为避免重复遍历,wordDict将找到的neighbor单词删掉。同时,找到下一个合适的单词(one character away) 的办法,是遍历该单词所有字符,将其替换成另外25个,然后查找WordDict,看新单词是否存在。

int ladderLength(string beginWord, string endWord, unordered_set<string>& wordList) {
        wordList.insert(endWord);
        queue<pair<string, int>> q;
        q.push({beginWord, 1});
        while(!q.empty()){
            string cur = q.front().first; 
            int lvl = q.front().second; q.pop();
            if(cur == endWord) return lvl;
            vector<string> neighbors = findNeighbors(cur, wordList);
            for(string it : neighbors){
                q.push({it, lvl+1});
            }
        }
        return 0;
    }
    
    vector<string> findNeighbors(string cur, unordered_set<string>& wordList){
        vector<string> ret;
        for(int i=0; i<cur.length(); i++){
            char c = cur[i];
            for(int j=0; j<26; j++){
                if('a'+j == c) continue;
                cur[i] = 'a'+j;
                if(wordList.count(cur)){
                    ret.push_back(cur);
                    wordList.erase(cur);
                }
            }
            cur[i] = c;
        }
        return ret;
    }

Word Ladder II:
https://leetcode.com/problems/word-ladder-ii/

II 则非常复杂,是leetcode里面acceptance非常低的题目之一。难点在于要返回所有的最短路径。不仅DFS搜索,还要选择最短的。
九章的思路是建立在 I 的基础上,从endWord开始往回BFS (同 I 一样)。这样的目的是建立每个单词与endWord的最短距离。然后再从起点DFS,每次判断neighbor 单词的距离是否小1,只有小1,才加入result set继续往下搜索。

class Solution {
public:
    vector<vector<string>> findLadders(string beginWord, string endWord, unordered_set<string> &wordList) {
        vector<vector<string>> allcomb;
        wordList.insert(beginWord);
        wordList.insert(endWord);
        unordered_map<string, int> mp;
        unordered_map<string, vector<string>> nexts;
        bfs(endWord, wordList, mp, nexts);
        /*for(auto it : mp){
            cout << it.first << " " << it.second << endl;
        }*/
        vector<string> comb;
        comb.push_back(beginWord);
        dfs(allcomb, comb, mp, nexts, beginWord, endWord);
        return allcomb;
    }
    
    void dfs(vector<vector<string>> &allcomb, vector<string> &comb, unordered_map<string, int> &mp, unordered_map<string, vector<string>> &nexts, string cur, string end){
        
        if(cur == end){
            allcomb.push_back(comb);
            return;
        }
        vector<string> neighbors = nexts[cur];
        for(int i=0; i<neighbors.size(); i++){
            if(mp[neighbors[i]] == mp[cur]-1){
                comb.push_back(neighbors[i]);
                dfs(allcomb, comb, mp, nexts, neighbors[i], end);
                comb.pop_back();
            }
        }
    }
    
    void bfs(string end, unordered_set<string> &dict, unordered_map<string, int> &mp, unordered_map<string, vector<string>> &nexts){
        queue<pair<string, int>> q;
        q.push({end, 0});
        mp[end] = 0;
        while(!q.empty()){
            string cur = q.front().first;
            int lvl = q.front().second; q.pop();
            vector<string> neighbors = findNeighbors(cur, dict);
            for(int i=0; i<neighbors.size(); i++){
                nexts[neighbors[i]].push_back(cur);
                if(!mp.count(neighbors[i])){
                    mp[neighbors[i]] = lvl+1;
                    q.push({neighbors[i], lvl+1});
                }
            }
        }
    }
    
    vector<string> findNeighbors(string cur, unordered_set<string> &dict){
        vector<string> ret;
        for(int i=0; i<cur.length(); i++){
            char c = cur[i];
            for(int j=0; j<26; j++){
                if('a' + j == c) continue;
                cur[i] = 'a' + j;
                if(dict.count(cur)){
                    ret.push_back(cur);
                }
            }
            cur[i] = c;
        }
        return ret;
    }
};

注意细节:要设 mp[end] = 0,并insert beginWord to set;

上一篇下一篇

猜你喜欢

热点阅读