LeetCode 力扣 136. 单词接龙 II

题目描述(困难难度)

给定两个单词,一个作为开始,一个作为结束,还有一个单词列表。然后依次选择单词,只有当前单词到下一个单词只有一个字母不同才能被选择,然后新的单词再作为当前单词,直到选到结束的单词。输出这个的最短路径,如果有多组,则都输出。

思路分析

结合了开始自己的想法,又看了 Discuss,这道题有些难讲清楚,一个原因就是解法的代码会很长,这里理一下整个的思路。

如果我们从开始的单词,把与之能够转换的单词连起来,它就会长成下边的样子。

橙色表示结束单词,上图橙色的路线就是我们要找的最短路径。所以我们要做的其实就是遍历上边的树,然后判断当前节点是不是结束单词,找到结束单词后,还要判断当前是不是最短的路径。说到遍历当然就是两种思路了,DFS 或者 BFS

解法一 DFS

利用回溯的思想,做一个 DFS。

首先要解决的问题是怎么找到节点的所有孩子节点。这里有两种方案。

第一种,遍历 wordList 来判断每个单词和当前单词是否只有一个字母不同。

for (int i = 0; i < wordList.size(); i++) {
    String curWord = wordList.get(i);
    //符合只有一个单词不同,就进入递归
    if (oneChanged(beginWord, curWord)) {
        //此时代表可以从 beginWord -> curWord
    }
}

private boolean oneChanged(String beginWord, String curWord) {
    int count = 0;
    for (int i = 0; i < beginWord.length(); i++) {
        if (beginWord.charAt(i) != curWord.charAt(i)) {
            count++;
        }
        if (count == 2) {
            return false;
        }
    }
    return count == 1;
}

这种的时间复杂度的话,如果 wordList 长度为 m,每个单词的长度为 n。那么就是 O(mn)
第二种,将要找的节点单词的每个位置换一个字符,然后看更改后的单词在不在 wordList 中。

//dict 就是 wordList,为了提高速度,从 List 转为 HashSet
//cur 是我们要考虑的单词
private List<String> getNext(String cur, Set<String> dict) {
    List<String> res = new ArrayList<>();
    char[] chars = cur.toCharArray();
    //考虑每一位
    for (int i = 0; i < chars.length; i++) {
        char old = chars[i];
        //考虑变成其他所有的字母
        for (char c = 'a'; c <= 'z'; c++) {
            if (c == old) {
                continue;
            }
            chars[i] = c;
            String next = new String(chars);
            //判断 wordList 是否包含修改后的单词
            if (dict.contains(next)) {
                res.add(next);
            }
        }
        chars[i] = old;
    }
    return res;
}

这种的话,由于用到了 HashSet ,所以 contains 函数就是 O(1)。所以整个计算量就是 26n,所以是 O(n)

还要解决的一个问题是,因为我们要找的是最短的路径。但是事先我们并不知道最短的路径是多少,我们需要一个全局变量来保存当前找到的路径的长度。如果找到的新的路径的长度比之前的路径短,就把之前的结果清空,重新找,如果是最小的长度,就加入到结果中。

看下一递归出口。

//到了结尾单词
if (beginWord.equals(endWord)) {
    //当前长度更小,清空之前的,加新的路径加入到结果中
    if (min > temp.size()) {
        ans.clear();
        min = temp.size();
        ans.add(new ArrayList<String>(temp));
    //相等的话就直接加路径加入到结果中
    } else if (min == temp.size()) {
        ans.add(new ArrayList<String>(temp));
    }
    return;
}
//当前的长度到达了 min,还是没有到达结束单词就提前结束
if (temp.size() >= min) {
    return;
}

得到下一个节点刚才讲了两种思路,我们先采用第一种解法,看一下效果。

public List<List<String>> findLadders(String beginWord, String endWord, List<String> wordList) {
    List<List<String>> ans = new ArrayList<>();
    ArrayList<String> temp = new ArrayList<String>();
    //temp 用来保存当前的路径
    temp.add(beginWord);
    findLaddersHelper(beginWord, endWord, wordList, temp, ans);
    return ans;
}

int min = Integer.MAX_VALUE;

private void findLaddersHelper(String beginWord, String endWord, List<String> wordList, 
                               ArrayList<String> temp, List<List<String>> ans) {
    if (beginWord.equals(endWord)) {
        if (min > temp.size()) {
            ans.clear();
            min = temp.size();
            ans.add(new ArrayList<String>(temp));
        } else if (min == temp.size()) {
            ans.add(new ArrayList<String>(temp));
        }
        return;
    }
    //当前的长度到达了 min,还是没有到达结束单词就提前结束
    if (temp.size() >= min) {
        return;
    }
    //遍历当前所有的单词
    for (int i = 0; i < wordList.size(); i++) {
        String curWord = wordList.get(i);
        //路径中已经含有当前单词,如果再把当前单词加到路径,那肯定会使得路径更长,所以跳过
        if (temp.contains(curWord)) {
            continue;
        }
        //符合只有一个单词不同,就进入递归
        if (oneChanged(beginWord, curWord)) {
            temp.add(curWord);
            findLaddersHelper(curWord, endWord, wordList, temp, ans);
            temp.remove(temp.size() - 1);
        }
    }
}
private boolean oneChanged(String beginWord, String curWord) {
    int count = 0;
    for (int i = 0; i < beginWord.length(); i++) {
        if (beginWord.charAt(i) != curWord.charAt(i)) {
            count++;
        }
        if (count == 2) {
            return false;
        }
    }
    return count == 1;
}

但是对于普通的输入可以解决,如果 wordList 过长的话就会造成超时了。

得到下一个的节点,如果采用第二种解法呢?

int min = Integer.MAX_VALUE;
public List<List<String>> findLadders(String beginWord, String endWord, List<String> wordList) {
    List<List<String>> ans = new ArrayList<>();
    ArrayList<String> temp = new ArrayList<String>();
    temp.add(beginWord);
    //temp 用来保存当前的路径
    findLaddersHelper(beginWord, endWord, wordList, temp, ans);
    return ans;
}


private void findLaddersHelper(String beginWord, String endWord, List<String> wordList,
                               ArrayList<String> temp, List<List<String>> ans) {
    if (beginWord.equals(endWord)) {
        if (min > temp.size()) {
            ans.clear();
            min = temp.size();
            ans.add(new ArrayList<String>(temp));
        } else if (min == temp.size()) {
            ans.add(new ArrayList<String>(temp));
        }
        return;
    }

    if (temp.size() >= min) {
        return;
    }
    Set<String> dict = new HashSet<>(wordList);
    //一次性到达所有的下一个的节点
    ArrayList<String> neighbors = getNeighbors(beginWord, dict);
    for (String neighbor : neighbors) {
        if (temp.contains(neighbor)) {
            continue;
        }
        temp.add(neighbor);
        findLaddersHelper(neighbor, endWord, wordList,  temp, ans);
        temp.remove(temp.size() - 1);
    }
}


private ArrayList<String> getNeighbors(String node, Set<String> dict) {
    ArrayList<String> res = new ArrayList<String>();
    char chs[] = node.toCharArray();

    for (char ch = 'a'; ch <= 'z'; ch++) {
        for (int i = 0; i < chs.length; i++) {
            if (chs[i] == ch)
                continue;
            char old_ch = chs[i];
            chs[i] = ch;
            if (dict.contains(String.valueOf(chs))) {
                res.add(String.valueOf(chs));
            }
            chs[i] = old_ch;
        }

    }
    return res;
}

快了一些,但是还是超时。

我们继续来优化,首先想一下为什么会超时,看一下之前的图。

DFS 的过程的话,结合上图,就是先考虑了最左边的路径,然后再回溯一下,继续到达底部。然后回溯回溯,终于到了一条含有结束单词的路径,然而事实上这条并不是最短路径。综上,我们会多判断很多无用的路径。

如果我们事先知道了最短路径长度是 4,那么我们只需要考虑前 4 层就足够了。

怎么知道结束单词在哪一层呢?只能一层层的找了,也就是 BFS。此外,因为上图需要搜索的树提前是没有的,我们需要边找边更新这个树。而在 DFS 中,我们也需要这个树,其实就是需要每个节点的所有相邻节点。

所以我们在 BFS 中,就把每个节点的所有相邻节点保存到 HashMap 中,就省去了 DFS 再去找相邻节点的时间。

此外,BFS 的过程中,把最短路径的高度用 min 也记录下来,在 DFS 的时候到达高度后就可以提前结束。

int min = 0;
public List<List<String>> findLadders(String beginWord, String endWord, List<String> wordList) {
    List<List<String>> ans = new ArrayList<>();
    //如果不含有结束单词,直接结束,不然后边会造成死循环
    if (!wordList.contains(endWord)) {
        return ans;
    }
    //利用 BFS 得到所有的邻居节点
    HashMap<String, ArrayList<String>> map = bfs(beginWord, endWord, wordList);
    ArrayList<String> temp = new ArrayList<String>();
    // temp 用来保存当前的路径
    temp.add(beginWord);
    findLaddersHelper(beginWord, endWord, map, temp, ans);
    return ans;
}

private void findLaddersHelper(String beginWord, String endWord, HashMap<String, ArrayList<String>> map,
                               ArrayList<String> temp, List<List<String>> ans) {
    if (beginWord.equals(endWord)) {
        ans.add(new ArrayList<String>(temp));

        return;
    }
    if(temp.size() - 1==  min){
        return;
    }
    // 得到所有的下一个的节点
    ArrayList<String> neighbors = map.getOrDefault(beginWord, new ArrayList<String>());
    for (String neighbor : neighbors) {
        if (temp.contains(neighbor)) {
            continue;
        }
        temp.add(neighbor);
        findLaddersHelper(neighbor, endWord, map, temp, ans);
        temp.remove(temp.size() - 1);
    }
}

public HashMap<String, ArrayList<String>> bfs(String beginWord, String endWord, List<String> wordList) {
    Queue<String> queue = new LinkedList<>();
    queue.offer(beginWord);
    HashMap<String, ArrayList<String>> map = new HashMap<>();
    boolean isFound = false;

    Set<String> dict = new HashSet<>(wordList);
    while (!queue.isEmpty()) {
        int size = queue.size();
        min++;
        for (int j = 0; j < size; j++) {
            String temp = queue.poll();
            // 一次性得到所有的下一个的节点
            ArrayList<String> neighbors = getNeighbors(temp, dict);
            map.put(temp, neighbors);
            for (String neighbor : neighbors) {
                if (neighbor.equals(endWord)) {
                    isFound = true;
                }
                queue.offer(neighbor);
            }
        }
        if (isFound) {
            break;
        }
    }
    return map;
}
private ArrayList<String> getNeighbors(String node, Set<String> dict) {
    ArrayList<String> res = new ArrayList<String>();
    char chs[] = node.toCharArray();

    for (char ch = 'a'; ch <= 'z'; ch++) {
        for (int i = 0; i < chs.length; i++) {
            if (chs[i] == ch)
                continue;
            char old_ch = chs[i];
            chs[i] = ch;
            if (dict.contains(String.valueOf(chs))) {
                res.add(String.valueOf(chs));
            }
            chs[i] = old_ch;
        }

    }
    return res;
}

然而这个优化,对于 leetcodetests 并没有什么影响。

让我们继续考虑优化方案,回到之前的图。

假如我们在考虑上图中黄色节点的相邻节点,发现第三层的 abc 在第二层已经考虑过了。所以第三层的 abc 其实不用再考虑了,第三层的 abc 后边的结构一定和第二层后边的结构一样,因为我们要找最短的路径,所以如果产生了最短路径,一定是第二层的 abc 首先达到结束单词。

所以其实我们在考虑第 k 层的某一个单词,如果这个单词在第 1k-1 层已经出现过,我们其实就不过继续向下探索了。

在之前的代码中,我们其实已经考虑了部分这个问题。

if (temp.contains(neighbor)) {
    continue;
}

但我们只考虑了当前路径是否含有该单词,而就像上图表示的,其他路径之前已经考虑过了当前单词,我们也是可以跳过的。

根据这个优化思路,有两种解决方案。

第一种,再利用一个 HashMap,记为 distance 变量。在 BFS 的过程中,把第一次遇到的单词当前的层数存起来。之后遇到也不进行更新,就会是下边的效果。

这样我们就可以在 DFS 的时候来判断当前黄色的节点的 distance 是不是比邻接节点的小 1。上图中 distance 都是 1 ,所以不符合,就可以跳过。

此外,在 DFS 中,因为我们每次都根据节点的层数来进行深搜,所以之前保存最短路径的全局变量 min 在这里也就不需要了。

public List<List<String>> findLadders(String beginWord, String endWord, List<String> wordList) {
    List<List<String>> ans = new ArrayList<>();
    // 如果不含有结束单词,直接结束,不然后边会造成死循环
    if (!wordList.contains(endWord)) {
        return ans;
    }
    // 利用 BFS 得到所有的邻居节点,以及每个节点的所在层数
    HashMap<String, Integer> distance = new HashMap<>();
    HashMap<String, ArrayList<String>> map = new HashMap<>();
    bfs(beginWord, endWord, wordList, map, distance);
    ArrayList<String> temp = new ArrayList<String>();
    // temp 用来保存当前的路径
    temp.add(beginWord);
    findLaddersHelper(beginWord, endWord, map, distance, temp, ans);
    return ans;
}

private void findLaddersHelper(String beginWord, String endWord, HashMap<String, ArrayList<String>> map,
                               HashMap<String, Integer> distance, ArrayList<String> temp, List<List<String>> ans) {
    if (beginWord.equals(endWord)) {
        ans.add(new ArrayList<String>(temp));
        return;
    }
    // 得到所有的下一个的节点
    /*
      "a"
      "c"
      ["a","b","c"]*/
    //之所以是 map.getOrDefault 而不是 get,就是上边的情况 get 会出错
    ArrayList<String> neighbors = map.getOrDefault(beginWord, new ArrayList<String>());
    for (String neighbor : neighbors) {
        //判断层数是否符合
        if (distance.get(beginWord) + 1 == distance.get(neighbor)) {
            temp.add(neighbor);
            findLaddersHelper(neighbor, endWord, map, distance, temp, ans);
            temp.remove(temp.size() - 1);
        }
    }
}

public void bfs(String beginWord, String endWord, List<String> wordList, HashMap<String, ArrayList<String>> map,
                HashMap<String, Integer> distance) {
    Queue<String> queue = new LinkedList<>();
    queue.offer(beginWord);
    distance.put(beginWord, 0);
    boolean isFound = false;
    int depth = 0;
    Set<String> dict = new HashSet<>(wordList);
    while (!queue.isEmpty()) {
        int size = queue.size();
        depth++;
        for (int j = 0; j < size; j++) {
            String temp = queue.poll();
            // 一次性得到所有的下一个的节点
            ArrayList<String> neighbors = getNeighbors(temp, dict);
            map.put(temp, neighbors);
            for (String neighbor : neighbors) {
                if (!distance.containsKey(neighbor)) {
                    distance.put(neighbor, depth);
                    if (neighbor.equals(endWord)) {
                        isFound = true;
                    }
                    queue.offer(neighbor);
                }

            }
        }
        if (isFound) {
            break;
        }
    }
}

private ArrayList<String> getNeighbors(String node, Set<String> dict) {
    ArrayList<String> res = new ArrayList<String>();
    char chs[] = node.toCharArray();

    for (char ch = 'a'; ch <= 'z'; ch++) {
        for (int i = 0; i < chs.length; i++) {
            if (chs[i] == ch)
                continue;
            char old_ch = chs[i];
            chs[i] = ch;
             if (dict.contains(String.valueOf(chs))) {
                res.add(String.valueOf(chs));
            }
            chs[i] = old_ch;
        }

    }
    return res;
}

终于,上边的算法 AC 了。上边讲到我们提前存储了 distance ,方便在 DFS 中来判断我们是否继续深搜。

这里再讲一下另一种思路,再回顾一下这个要进行优化的图。

我们就是减少了第三层的 abc 的情况的判断。我们其实可以不用 distance ,在 BFS 中,如果发现有邻接节点在之前已经出现过了,我们直接把这个邻接节点删除不去。这样的话,在 DFS 中就不用再判断了,直接取邻居节点就可以了。

判断之前是否已经处理过,可以用一个 HashSet 来把之前的节点存起来进行判断。

这里删除邻接节点需要用到一个语言特性,java 中遍历 List 过程中,不能对 List 元素进行删除。如果想边遍历边删除,可以借助迭代器。

Iterator<String> it = neighbors.iterator();//把元素导入迭代器
while (it.hasNext()) {
    String neighbor = it.next();
    if (!visited.contains(neighbor)) {
        if (neighbor.equals(endWord)) {
            isFound = true;
        }
        queue.offer(neighbor);
        subVisited.add(neighbor);
    }else{
        it.remove();
    } 
}

此外我们要判断的是当前节点在之前层有没有出现过,当前层正在遍历的节点先加到 subVisited 中。

public List<List<String>> findLadders(String beginWord, String endWord, List<String> wordList) {
    List<List<String>> ans = new ArrayList<>();
    if (!wordList.contains(endWord)) {
        return ans;
    }
    // 利用 BFS 得到所有的邻居节点
    HashMap<String, ArrayList<String>> map = new HashMap<>();
    bfs(beginWord, endWord, wordList, map);
    ArrayList<String> temp = new ArrayList<String>();
    // temp 用来保存当前的路径
    temp.add(beginWord);
    findLaddersHelper(beginWord, endWord, map, temp, ans);
    return ans;
}

private void findLaddersHelper(String beginWord, String endWord, HashMap<String, ArrayList<String>> map,
                               ArrayList<String> temp, List<List<String>> ans) {
    if (beginWord.equals(endWord)) {
        ans.add(new ArrayList<String>(temp));
        return;
    }
    // 得到所有的下一个的节点
    ArrayList<String> neighbors = map.getOrDefault(beginWord, new ArrayList<String>());
    for (String neighbor : neighbors) {
        temp.add(neighbor);
        findLaddersHelper(neighbor, endWord, map, temp, ans);
        temp.remove(temp.size() - 1);

    }
}

public void bfs(String beginWord, String endWord, List<String> wordList, HashMap<String, ArrayList<String>> map) {
    Queue<String> queue = new LinkedList<>();
    queue.offer(beginWord);
    boolean isFound = false;
    int depth = 0;
    Set<String> dict = new HashSet<>(wordList);
    Set<String> visited = new HashSet<>();
    visited.add(beginWord);
    while (!queue.isEmpty()) {
        int size = queue.size();
        depth++;
        Set<String> subVisited = new HashSet<>();
        for (int j = 0; j < size; j++) {
            String temp = queue.poll();
            // 一次性得到所有的下一个的节点
            ArrayList<String> neighbors = getNeighbors(temp, dict); 
            Iterator<String> it = neighbors.iterator();//把元素导入迭代器
            while (it.hasNext()) {
                String neighbor = it.next();
                if (!visited.contains(neighbor)) {
                    if (neighbor.equals(endWord)) {
                        isFound = true;
                    }
                    queue.offer(neighbor);
                    subVisited.add(neighbor);
                }else{
                    it.remove();
                } 
            }
            map.put(temp, neighbors);
        }
        visited.addAll(subVisited);
        if (isFound) {
            break;
        }
    }
}

private ArrayList<String> getNeighbors(String node, Set<String> dict) {
    ArrayList<String> res = new ArrayList<String>();
    char chs[] = node.toCharArray();

    for (char ch = 'a'; ch <= 'z'; ch++) {
        for (int i = 0; i < chs.length; i++) {
            if (chs[i] == ch)
                continue;
            char old_ch = chs[i];
            chs[i] = ch;
            if (dict.contains(String.valueOf(chs))) {
                res.add(String.valueOf(chs));
            }
            chs[i] = old_ch;
        }

    }
    return res;
}

解法二 BFS

如果理解了上边的 DFS 过程,接下来就很好讲了。上边 DFS 借助了 BFS 把所有的邻接关系保存了起来,再用 DFS 进行深度搜索。

我们可不可以只用 BFS,一边进行层次遍历,一边就保存结果。当到达结束单词的时候,就把结果存储。省去再进行 DFS 的过程。

是完全可以的,BFS 的队列就不去存储 String 了,直接去存到目前为止的路径,也就是一个 List

public List<List<String>> findLadders(String beginWord, String endWord, List<String> wordList) {
    List<List<String>> ans = new ArrayList<>();
    // 如果不含有结束单词,直接结束,不然后边会造成死循环
    if (!wordList.contains(endWord)) {
        return ans;
    }
    bfs(beginWord, endWord, wordList, ans);
    return ans;
}

public void bfs(String beginWord, String endWord, List<String> wordList, List<List<String>> ans) {
    Queue<List<String>> queue = new LinkedList<>();
    List<String> path = new ArrayList<>();
    path.add(beginWord);
    queue.offer(path);
    boolean isFound = false;
    Set<String> dict = new HashSet<>(wordList);
    Set<String> visited = new HashSet<>();
    visited.add(beginWord);
    while (!queue.isEmpty()) {
        int size = queue.size();
        Set<String> subVisited = new HashSet<>();
        for (int j = 0; j < size; j++) {
            List<String> p = queue.poll();
            //得到当前路径的末尾单词
            String temp = p.get(p.size() - 1);
            // 一次性得到所有的下一个的节点
            ArrayList<String> neighbors = getNeighbors(temp, dict);
            for (String neighbor : neighbors) {
                //只考虑之前没有出现过的单词
                if (!visited.contains(neighbor)) {
                    //到达结束单词
                    if (neighbor.equals(endWord)) {
                        isFound = true;
                        p.add(neighbor);
                        ans.add(new ArrayList<String>(p));
                        p.remove(p.size() - 1);
                    }
                    //加入当前单词
                    p.add(neighbor);
                    queue.offer(new ArrayList<String>(p));
                    p.remove(p.size() - 1);
                    subVisited.add(neighbor);
                }
            }
        }
        visited.addAll(subVisited);
        if (isFound) {
            break;
        }
    }
}

private ArrayList<String> getNeighbors(String node, Set<String> dict) {
    ArrayList<String> res = new ArrayList<String>();
    char chs[] = node.toCharArray();
    for (char ch = 'a'; ch <= 'z'; ch++) {
        for (int i = 0; i < chs.length; i++) {
            if (chs[i] == ch)
                continue;
            char old_ch = chs[i];
            chs[i] = ch;
            if (dict.contains(String.valueOf(chs))) {
                res.add(String.valueOf(chs));
            }
            chs[i] = old_ch;
        }

    }
    return res;
}

代码看起来简洁了很多。

解法三 DFS + BFS 双向搜索(two-end BFS)

在解法一的思路上,我们还能够继续优化。

解法一中,我们利用了 BFS 建立了每个节点的邻居节点。在之前的示意图中,我们把同一个字符串也画在了不同节点。这里把同一个节点画在一起,再看一下。

我们可以从结束单词反向进行 BFS

这样的话,当两个方向产生了共同的节点,就是我们的最短路径了。

至于每次从哪个方向扩展,我们可以每次选择需要扩展的节点数少的方向进行扩展。

例如上图中,一开始需要向下扩展的个数是 1 个,需要向上扩展的个数是 1 个。个数相等,我们就向下扩展。然后需要向下扩展的个数就变成了 4 个,而需要向上扩展的个数是 1 个,所以此时我们向上扩展。接着,需要向上扩展的个数变成了 6 个,需要向下扩展的个数是 4 个,我们就向下扩展......直到相遇。

双向扩展的好处,我们粗略的估计一下时间复杂度。

假设 beginwordendword 之间的距离是 d。每个节点可以扩展出 k 个节点。

那么正常的时间复杂就是 k^d

双向搜索的时间复杂度就是 k^{d/2} + k^{d/2}

public List<List<String>> findLadders(String beginWord, String endWord, List<String> wordList) {
    List<List<String>> ans = new ArrayList<>();
    if (!wordList.contains(endWord)) {
        return ans;
    }
    // 利用 BFS 得到所有的邻居节点
    HashMap<String, ArrayList<String>> map = new HashMap<>();
    bfs(beginWord, endWord, wordList, map);
    ArrayList<String> temp = new ArrayList<String>();
    // temp 用来保存当前的路径
    temp.add(beginWord);
    findLaddersHelper(beginWord, endWord, map, temp, ans);
    return ans;
}

private void findLaddersHelper(String beginWord, String endWord, HashMap<String, ArrayList<String>> map,
                               ArrayList<String> temp, List<List<String>> ans) {
    if (beginWord.equals(endWord)) {
        ans.add(new ArrayList<String>(temp));
        return;
    }
    // 得到所有的下一个的节点
    ArrayList<String> neighbors = map.getOrDefault(beginWord, new ArrayList<String>());
    for (String neighbor : neighbors) {
        temp.add(neighbor);
        findLaddersHelper(neighbor, endWord, map, temp, ans);
        temp.remove(temp.size() - 1);
    }
}

//利用递归实现了双向搜索
private void bfs(String beginWord, String endWord, List<String> wordList, HashMap<String, ArrayList<String>> map) {
    Set<String> set1 = new HashSet<String>();
    set1.add(beginWord);
    Set<String> set2 = new HashSet<String>();
    set2.add(endWord);
    Set<String> wordSet = new HashSet<String>(wordList);
    bfsHelper(set1, set2, wordSet, true, map);
}

// direction 为 true 代表向下扩展,false 代表向上扩展
private boolean bfsHelper(Set<String> set1, Set<String> set2, Set<String> wordSet, boolean direction,
                          HashMap<String, ArrayList<String>> map) {
    //set1 为空了,就直接结束
    //比如下边的例子就会造成 set1 为空
    /*  "hot"
        "dog"
        ["hot","dog"]*/
    if(set1.isEmpty()){
        return false;
    }
    // set1 的数量多,就反向扩展
    if (set1.size() > set2.size()) {
        return bfsHelper(set2, set1, wordSet, !direction, map);
    }
    // 将已经访问过单词删除
    wordSet.removeAll(set1);
    wordSet.removeAll(set2);

    boolean done = false;

    // 保存新扩展得到的节点
    Set<String> set = new HashSet<String>();

    for (String str : set1) {
        //遍历每一位
        for (int i = 0; i < str.length(); i++) {
            char[] chars = str.toCharArray();

            // 尝试所有字母
            for (char ch = 'a'; ch <= 'z'; ch++) {
                if(chars[i] == ch){
                    continue;
                }
                chars[i] = ch;

                String word = new String(chars);

                // 根据方向得到 map 的 key 和 val
                String key = direction ? str : word;
                String val = direction ? word : str;

                ArrayList<String> list = map.containsKey(key) ? map.get(key) : new ArrayList<String>();
                
                //如果相遇了就保存结果
                if (set2.contains(word)) {
                    done = true;
                    list.add(val);
                    map.put(key, list);
                }

                //如果还没有相遇,并且新的单词在 word 中,那么就加到 set 中
                if (!done && wordSet.contains(word)) {
                    set.add(word);
                    list.add(val);
                    map.put(key, list);
                }
            }
        }
    }

    //一般情况下新扩展的元素会多一些,所以我们下次反方向扩展  set2
    return done || bfsHelper(set2, set, wordSet, !direction, map);

}

最近事情比较多,这道题每天想一想,陆陆续续拖了好几天了。这道题本质上就是在正常的遍历的基础上,去将一些分支剪去,从而提高速度。至于方法的话,除了我上边介绍的实现方式,应该也会有很多其它的方式,但其实本质上是为了实现一样的东西。另外,双向搜索的方法,自己第一次遇到,网上搜了一下,看样子还是比较经典的一个算法。主要就是用于解决已知起点和终点,去求图的最短路径的问题。

题目描述(困难难度)

给定两个单词,一个作为开始,一个作为结束,还有一个单词列表。然后依次选择单词,只有当前单词到下一个单词只有一个字母不同才能被选择,然后新的单词再作为当前单词,直到选到结束的单词。输出这个的最短路径,如果有多组,则都输出。

思路分析

结合了开始自己的想法,又看了 Discuss,这道题有些难讲清楚,一个原因就是解法的代码会很长,这里理一下整个的思路。

如果我们从开始的单词,把与之能够转换的单词连起来,它就会长成下边的样子。

橙色表示结束单词,上图橙色的路线就是我们要找的最短路径。所以我们要做的其实就是遍历上边的树,然后判断当前节点是不是结束单词,找到结束单词后,还要判断当前是不是最短的路径。说到遍历当然就是两种思路了,DFS 或者 BFS

解法一 DFS

利用回溯的思想,做一个 DFS。

首先要解决的问题是怎么找到节点的所有孩子节点。这里有两种方案。

第一种,遍历 wordList 来判断每个单词和当前单词是否只有一个字母不同。

for (int i = 0; i < wordList.size(); i++) {
    String curWord = wordList.get(i);
    //符合只有一个单词不同,就进入递归
    if (oneChanged(beginWord, curWord)) {
        //此时代表可以从 beginWord -> curWord
    }
}

private boolean oneChanged(String beginWord, String curWord) {
    int count = 0;
    for (int i = 0; i < beginWord.length(); i++) {
        if (beginWord.charAt(i) != curWord.charAt(i)) {
            count++;
        }
        if (count == 2) {
            return false;
        }
    }
    return count == 1;
}

这种的时间复杂度的话,如果 wordList 长度为 m,每个单词的长度为 n。那么就是 O(mn)
第二种,将要找的节点单词的每个位置换一个字符,然后看更改后的单词在不在 wordList 中。

//dict 就是 wordList,为了提高速度,从 List 转为 HashSet
//cur 是我们要考虑的单词
private List<String> getNext(String cur, Set<String> dict) {
    List<String> res = new ArrayList<>();
    char[] chars = cur.toCharArray();
    //考虑每一位
    for (int i = 0; i < chars.length; i++) {
        char old = chars[i];
        //考虑变成其他所有的字母
        for (char c = 'a'; c <= 'z'; c++) {
            if (c == old) {
                continue;
            }
            chars[i] = c;
            String next = new String(chars);
            //判断 wordList 是否包含修改后的单词
            if (dict.contains(next)) {
                res.add(next);
            }
        }
        chars[i] = old;
    }
    return res;
}

这种的话,由于用到了 HashSet ,所以 contains 函数就是 O(1)。所以整个计算量就是 26n,所以是 O(n)

还要解决的一个问题是,因为我们要找的是最短的路径。但是事先我们并不知道最短的路径是多少,我们需要一个全局变量来保存当前找到的路径的长度。如果找到的新的路径的长度比之前的路径短,就把之前的结果清空,重新找,如果是最小的长度,就加入到结果中。

看下一递归出口。

//到了结尾单词
if (beginWord.equals(endWord)) {
    //当前长度更小,清空之前的,加新的路径加入到结果中
    if (min > temp.size()) {
        ans.clear();
        min = temp.size();
        ans.add(new ArrayList<String>(temp));
    //相等的话就直接加路径加入到结果中
    } else if (min == temp.size()) {
        ans.add(new ArrayList<String>(temp));
    }
    return;
}
//当前的长度到达了 min,还是没有到达结束单词就提前结束
if (temp.size() >= min) {
    return;
}

得到下一个节点刚才讲了两种思路,我们先采用第一种解法,看一下效果。

public List<List<String>> findLadders(String beginWord, String endWord, List<String> wordList) {
    List<List<String>> ans = new ArrayList<>();
    ArrayList<String> temp = new ArrayList<String>();
    //temp 用来保存当前的路径
    temp.add(beginWord);
    findLaddersHelper(beginWord, endWord, wordList, temp, ans);
    return ans;
}

int min = Integer.MAX_VALUE;

private void findLaddersHelper(String beginWord, String endWord, List<String> wordList, 
                               ArrayList<String> temp, List<List<String>> ans) {
    if (beginWord.equals(endWord)) {
        if (min > temp.size()) {
            ans.clear();
            min = temp.size();
            ans.add(new ArrayList<String>(temp));
        } else if (min == temp.size()) {
            ans.add(new ArrayList<String>(temp));
        }
        return;
    }
    //当前的长度到达了 min,还是没有到达结束单词就提前结束
    if (temp.size() >= min) {
        return;
    }
    //遍历当前所有的单词
    for (int i = 0; i < wordList.size(); i++) {
        String curWord = wordList.get(i);
        //路径中已经含有当前单词,如果再把当前单词加到路径,那肯定会使得路径更长,所以跳过
        if (temp.contains(curWord)) {
            continue;
        }
        //符合只有一个单词不同,就进入递归
        if (oneChanged(beginWord, curWord)) {
            temp.add(curWord);
            findLaddersHelper(curWord, endWord, wordList, temp, ans);
            temp.remove(temp.size() - 1);
        }
    }
}
private boolean oneChanged(String beginWord, String curWord) {
    int count = 0;
    for (int i = 0; i < beginWord.length(); i++) {
        if (beginWord.charAt(i) != curWord.charAt(i)) {
            count++;
        }
        if (count == 2) {
            return false;
        }
    }
    return count == 1;
}

但是对于普通的输入可以解决,如果 wordList 过长的话就会造成超时了。

得到下一个的节点,如果采用第二种解法呢?

int min = Integer.MAX_VALUE;
public List<List<String>> findLadders(String beginWord, String endWord, List<String> wordList) {
    List<List<String>> ans = new ArrayList<>();
    ArrayList<String> temp = new ArrayList<String>();
    temp.add(beginWord);
    //temp 用来保存当前的路径
    findLaddersHelper(beginWord, endWord, wordList, temp, ans);
    return ans;
}


private void findLaddersHelper(String beginWord, String endWord, List<String> wordList,
                               ArrayList<String> temp, List<List<String>> ans) {
    if (beginWord.equals(endWord)) {
        if (min > temp.size()) {
            ans.clear();
            min = temp.size();
            ans.add(new ArrayList<String>(temp));
        } else if (min == temp.size()) {
            ans.add(new ArrayList<String>(temp));
        }
        return;
    }

    if (temp.size() >= min) {
        return;
    }
    Set<String> dict = new HashSet<>(wordList);
    //一次性到达所有的下一个的节点
    ArrayList<String> neighbors = getNeighbors(beginWord, dict);
    for (String neighbor : neighbors) {
        if (temp.contains(neighbor)) {
            continue;
        }
        temp.add(neighbor);
        findLaddersHelper(neighbor, endWord, wordList,  temp, ans);
        temp.remove(temp.size() - 1);
    }
}


private ArrayList<String> getNeighbors(String node, Set<String> dict) {
    ArrayList<String> res = new ArrayList<String>();
    char chs[] = node.toCharArray();

    for (char ch = 'a'; ch <= 'z'; ch++) {
        for (int i = 0; i < chs.length; i++) {
            if (chs[i] == ch)
                continue;
            char old_ch = chs[i];
            chs[i] = ch;
            if (dict.contains(String.valueOf(chs))) {
                res.add(String.valueOf(chs));
            }
            chs[i] = old_ch;
        }

    }
    return res;
}

快了一些,但是还是超时。

我们继续来优化,首先想一下为什么会超时,看一下之前的图。

DFS 的过程的话,结合上图,就是先考虑了最左边的路径,然后再回溯一下,继续到达底部。然后回溯回溯,终于到了一条含有结束单词的路径,然而事实上这条并不是最短路径。综上,我们会多判断很多无用的路径。

如果我们事先知道了最短路径长度是 4,那么我们只需要考虑前 4 层就足够了。

怎么知道结束单词在哪一层呢?只能一层层的找了,也就是 BFS。此外,因为上图需要搜索的树提前是没有的,我们需要边找边更新这个树。而在 DFS 中,我们也需要这个树,其实就是需要每个节点的所有相邻节点。

所以我们在 BFS 中,就把每个节点的所有相邻节点保存到 HashMap 中,就省去了 DFS 再去找相邻节点的时间。

此外,BFS 的过程中,把最短路径的高度用 min 也记录下来,在 DFS 的时候到达高度后就可以提前结束。

int min = 0;
public List<List<String>> findLadders(String beginWord, String endWord, List<String> wordList) {
    List<List<String>> ans = new ArrayList<>();
    //如果不含有结束单词,直接结束,不然后边会造成死循环
    if (!wordList.contains(endWord)) {
        return ans;
    }
    //利用 BFS 得到所有的邻居节点
    HashMap<String, ArrayList<String>> map = bfs(beginWord, endWord, wordList);
    ArrayList<String> temp = new ArrayList<String>();
    // temp 用来保存当前的路径
    temp.add(beginWord);
    findLaddersHelper(beginWord, endWord, map, temp, ans);
    return ans;
}

private void findLaddersHelper(String beginWord, String endWord, HashMap<String, ArrayList<String>> map,
                               ArrayList<String> temp, List<List<String>> ans) {
    if (beginWord.equals(endWord)) {
        ans.add(new ArrayList<String>(temp));

        return;
    }
    if(temp.size() - 1==  min){
        return;
    }
    // 得到所有的下一个的节点
    ArrayList<String> neighbors = map.getOrDefault(beginWord, new ArrayList<String>());
    for (String neighbor : neighbors) {
        if (temp.contains(neighbor)) {
            continue;
        }
        temp.add(neighbor);
        findLaddersHelper(neighbor, endWord, map, temp, ans);
        temp.remove(temp.size() - 1);
    }
}

public HashMap<String, ArrayList<String>> bfs(String beginWord, String endWord, List<String> wordList) {
    Queue<String> queue = new LinkedList<>();
    queue.offer(beginWord);
    HashMap<String, ArrayList<String>> map = new HashMap<>();
    boolean isFound = false;

    Set<String> dict = new HashSet<>(wordList);
    while (!queue.isEmpty()) {
        int size = queue.size();
        min++;
        for (int j = 0; j < size; j++) {
            String temp = queue.poll();
            // 一次性得到所有的下一个的节点
            ArrayList<String> neighbors = getNeighbors(temp, dict);
            map.put(temp, neighbors);
            for (String neighbor : neighbors) {
                if (neighbor.equals(endWord)) {
                    isFound = true;
                }
                queue.offer(neighbor);
            }
        }
        if (isFound) {
            break;
        }
    }
    return map;
}
private ArrayList<String> getNeighbors(String node, Set<String> dict) {
    ArrayList<String> res = new ArrayList<String>();
    char chs[] = node.toCharArray();

    for (char ch = 'a'; ch <= 'z'; ch++) {
        for (int i = 0; i < chs.length; i++) {
            if (chs[i] == ch)
                continue;
            char old_ch = chs[i];
            chs[i] = ch;
            if (dict.contains(String.valueOf(chs))) {
                res.add(String.valueOf(chs));
            }
            chs[i] = old_ch;
        }

    }
    return res;
}

然而这个优化,对于 leetcodetests 并没有什么影响。

让我们继续考虑优化方案,回到之前的图。

假如我们在考虑上图中黄色节点的相邻节点,发现第三层的 abc 在第二层已经考虑过了。所以第三层的 abc 其实不用再考虑了,第三层的 abc 后边的结构一定和第二层后边的结构一样,因为我们要找最短的路径,所以如果产生了最短路径,一定是第二层的 abc 首先达到结束单词。

所以其实我们在考虑第 k 层的某一个单词,如果这个单词在第 1k-1 层已经出现过,我们其实就不过继续向下探索了。

在之前的代码中,我们其实已经考虑了部分这个问题。

if (temp.contains(neighbor)) {
    continue;
}

但我们只考虑了当前路径是否含有该单词,而就像上图表示的,其他路径之前已经考虑过了当前单词,我们也是可以跳过的。

根据这个优化思路,有两种解决方案。

第一种,再利用一个 HashMap,记为 distance 变量。在 BFS 的过程中,把第一次遇到的单词当前的层数存起来。之后遇到也不进行更新,就会是下边的效果。

这样我们就可以在 DFS 的时候来判断当前黄色的节点的 distance 是不是比邻接节点的小 1。上图中 distance 都是 1 ,所以不符合,就可以跳过。

此外,在 DFS 中,因为我们每次都根据节点的层数来进行深搜,所以之前保存最短路径的全局变量 min 在这里也就不需要了。

public List<List<String>> findLadders(String beginWord, String endWord, List<String> wordList) {
    List<List<String>> ans = new ArrayList<>();
    // 如果不含有结束单词,直接结束,不然后边会造成死循环
    if (!wordList.contains(endWord)) {
        return ans;
    }
    // 利用 BFS 得到所有的邻居节点,以及每个节点的所在层数
    HashMap<String, Integer> distance = new HashMap<>();
    HashMap<String, ArrayList<String>> map = new HashMap<>();
    bfs(beginWord, endWord, wordList, map, distance);
    ArrayList<String> temp = new ArrayList<String>();
    // temp 用来保存当前的路径
    temp.add(beginWord);
    findLaddersHelper(beginWord, endWord, map, distance, temp, ans);
    return ans;
}

private void findLaddersHelper(String beginWord, String endWord, HashMap<String, ArrayList<String>> map,
                               HashMap<String, Integer> distance, ArrayList<String> temp, List<List<String>> ans) {
    if (beginWord.equals(endWord)) {
        ans.add(new ArrayList<String>(temp));
        return;
    }
    // 得到所有的下一个的节点
    /*
      "a"
      "c"
      ["a","b","c"]*/
    //之所以是 map.getOrDefault 而不是 get,就是上边的情况 get 会出错
    ArrayList<String> neighbors = map.getOrDefault(beginWord, new ArrayList<String>());
    for (String neighbor : neighbors) {
        //判断层数是否符合
        if (distance.get(beginWord) + 1 == distance.get(neighbor)) {
            temp.add(neighbor);
            findLaddersHelper(neighbor, endWord, map, distance, temp, ans);
            temp.remove(temp.size() - 1);
        }
    }
}

public void bfs(String beginWord, String endWord, List<String> wordList, HashMap<String, ArrayList<String>> map,
                HashMap<String, Integer> distance) {
    Queue<String> queue = new LinkedList<>();
    queue.offer(beginWord);
    distance.put(beginWord, 0);
    boolean isFound = false;
    int depth = 0;
    Set<String> dict = new HashSet<>(wordList);
    while (!queue.isEmpty()) {
        int size = queue.size();
        depth++;
        for (int j = 0; j < size; j++) {
            String temp = queue.poll();
            // 一次性得到所有的下一个的节点
            ArrayList<String> neighbors = getNeighbors(temp, dict);
            map.put(temp, neighbors);
            for (String neighbor : neighbors) {
                if (!distance.containsKey(neighbor)) {
                    distance.put(neighbor, depth);
                    if (neighbor.equals(endWord)) {
                        isFound = true;
                    }
                    queue.offer(neighbor);
                }

            }
        }
        if (isFound) {
            break;
        }
    }
}

private ArrayList<String> getNeighbors(String node, Set<String> dict) {
    ArrayList<String> res = new ArrayList<String>();
    char chs[] = node.toCharArray();

    for (char ch = 'a'; ch <= 'z'; ch++) {
        for (int i = 0; i < chs.length; i++) {
            if (chs[i] == ch)
                continue;
            char old_ch = chs[i];
            chs[i] = ch;
             if (dict.contains(String.valueOf(chs))) {
                res.add(String.valueOf(chs));
            }
            chs[i] = old_ch;
        }

    }
    return res;
}

终于,上边的算法 AC 了。上边讲到我们提前存储了 distance ,方便在 DFS 中来判断我们是否继续深搜。

这里再讲一下另一种思路,再回顾一下这个要进行优化的图。

我们就是减少了第三层的 abc 的情况的判断。我们其实可以不用 distance ,在 BFS 中,如果发现有邻接节点在之前已经出现过了,我们直接把这个邻接节点删除不去。这样的话,在 DFS 中就不用再判断了,直接取邻居节点就可以了。

判断之前是否已经处理过,可以用一个 HashSet 来把之前的节点存起来进行判断。

这里删除邻接节点需要用到一个语言特性,java 中遍历 List 过程中,不能对 List 元素进行删除。如果想边遍历边删除,可以借助迭代器。

Iterator<String> it = neighbors.iterator();//把元素导入迭代器
while (it.hasNext()) {
    String neighbor = it.next();
    if (!visited.contains(neighbor)) {
        if (neighbor.equals(endWord)) {
            isFound = true;
        }
        queue.offer(neighbor);
        subVisited.add(neighbor);
    }else{
        it.remove();
    } 
}

此外我们要判断的是当前节点在之前层有没有出现过,当前层正在遍历的节点先加到 subVisited 中。

public List<List<String>> findLadders(String beginWord, String endWord, List<String> wordList) {
    List<List<String>> ans = new ArrayList<>();
    if (!wordList.contains(endWord)) {
        return ans;
    }
    // 利用 BFS 得到所有的邻居节点
    HashMap<String, ArrayList<String>> map = new HashMap<>();
    bfs(beginWord, endWord, wordList, map);
    ArrayList<String> temp = new ArrayList<String>();
    // temp 用来保存当前的路径
    temp.add(beginWord);
    findLaddersHelper(beginWord, endWord, map, temp, ans);
    return ans;
}

private void findLaddersHelper(String beginWord, String endWord, HashMap<String, ArrayList<String>> map,
                               ArrayList<String> temp, List<List<String>> ans) {
    if (beginWord.equals(endWord)) {
        ans.add(new ArrayList<String>(temp));
        return;
    }
    // 得到所有的下一个的节点
    ArrayList<String> neighbors = map.getOrDefault(beginWord, new ArrayList<String>());
    for (String neighbor : neighbors) {
        temp.add(neighbor);
        findLaddersHelper(neighbor, endWord, map, temp, ans);
        temp.remove(temp.size() - 1);

    }
}

public void bfs(String beginWord, String endWord, List<String> wordList, HashMap<String, ArrayList<String>> map) {
    Queue<String> queue = new LinkedList<>();
    queue.offer(beginWord);
    boolean isFound = false;
    int depth = 0;
    Set<String> dict = new HashSet<>(wordList);
    Set<String> visited = new HashSet<>();
    visited.add(beginWord);
    while (!queue.isEmpty()) {
        int size = queue.size();
        depth++;
        Set<String> subVisited = new HashSet<>();
        for (int j = 0; j < size; j++) {
            String temp = queue.poll();
            // 一次性得到所有的下一个的节点
            ArrayList<String> neighbors = getNeighbors(temp, dict); 
            Iterator<String> it = neighbors.iterator();//把元素导入迭代器
            while (it.hasNext()) {
                String neighbor = it.next();
                if (!visited.contains(neighbor)) {
                    if (neighbor.equals(endWord)) {
                        isFound = true;
                    }
                    queue.offer(neighbor);
                    subVisited.add(neighbor);
                }else{
                    it.remove();
                } 
            }
            map.put(temp, neighbors);
        }
        visited.addAll(subVisited);
        if (isFound) {
            break;
        }
    }
}

private ArrayList<String> getNeighbors(String node, Set<String> dict) {
    ArrayList<String> res = new ArrayList<String>();
    char chs[] = node.toCharArray();

    for (char ch = 'a'; ch <= 'z'; ch++) {
        for (int i = 0; i < chs.length; i++) {
            if (chs[i] == ch)
                continue;
            char old_ch = chs[i];
            chs[i] = ch;
            if (dict.contains(String.valueOf(chs))) {
                res.add(String.valueOf(chs));
            }
            chs[i] = old_ch;
        }

    }
    return res;
}

解法二 BFS

如果理解了上边的 DFS 过程,接下来就很好讲了。上边 DFS 借助了 BFS 把所有的邻接关系保存了起来,再用 DFS 进行深度搜索。

我们可不可以只用 BFS,一边进行层次遍历,一边就保存结果。当到达结束单词的时候,就把结果存储。省去再进行 DFS 的过程。

是完全可以的,BFS 的队列就不去存储 String 了,直接去存到目前为止的路径,也就是一个 List

public List<List<String>> findLadders(String beginWord, String endWord, List<String> wordList) {
    List<List<String>> ans = new ArrayList<>();
    // 如果不含有结束单词,直接结束,不然后边会造成死循环
    if (!wordList.contains(endWord)) {
        return ans;
    }
    bfs(beginWord, endWord, wordList, ans);
    return ans;
}

public void bfs(String beginWord, String endWord, List<String> wordList, List<List<String>> ans) {
    Queue<List<String>> queue = new LinkedList<>();
    List<String> path = new ArrayList<>();
    path.add(beginWord);
    queue.offer(path);
    boolean isFound = false;
    Set<String> dict = new HashSet<>(wordList);
    Set<String> visited = new HashSet<>();
    visited.add(beginWord);
    while (!queue.isEmpty()) {
        int size = queue.size();
        Set<String> subVisited = new HashSet<>();
        for (int j = 0; j < size; j++) {
            List<String> p = queue.poll();
            //得到当前路径的末尾单词
            String temp = p.get(p.size() - 1);
            // 一次性得到所有的下一个的节点
            ArrayList<String> neighbors = getNeighbors(temp, dict);
            for (String neighbor : neighbors) {
                //只考虑之前没有出现过的单词
                if (!visited.contains(neighbor)) {
                    //到达结束单词
                    if (neighbor.equals(endWord)) {
                        isFound = true;
                        p.add(neighbor);
                        ans.add(new ArrayList<String>(p));
                        p.remove(p.size() - 1);
                    }
                    //加入当前单词
                    p.add(neighbor);
                    queue.offer(new ArrayList<String>(p));
                    p.remove(p.size() - 1);
                    subVisited.add(neighbor);
                }
            }
        }
        visited.addAll(subVisited);
        if (isFound) {
            break;
        }
    }
}

private ArrayList<String> getNeighbors(String node, Set<String> dict) {
    ArrayList<String> res = new ArrayList<String>();
    char chs[] = node.toCharArray();
    for (char ch = 'a'; ch <= 'z'; ch++) {
        for (int i = 0; i < chs.length; i++) {
            if (chs[i] == ch)
                continue;
            char old_ch = chs[i];
            chs[i] = ch;
            if (dict.contains(String.valueOf(chs))) {
                res.add(String.valueOf(chs));
            }
            chs[i] = old_ch;
        }

    }
    return res;
}

代码看起来简洁了很多。

解法三 DFS + BFS 双向搜索(two-end BFS)

在解法一的思路上,我们还能够继续优化。

解法一中,我们利用了 BFS 建立了每个节点的邻居节点。在之前的示意图中,我们把同一个字符串也画在了不同节点。这里把同一个节点画在一起,再看一下。

我们可以从结束单词反向进行 BFS

这样的话,当两个方向产生了共同的节点,就是我们的最短路径了。

至于每次从哪个方向扩展,我们可以每次选择需要扩展的节点数少的方向进行扩展。

例如上图中,一开始需要向下扩展的个数是 1 个,需要向上扩展的个数是 1 个。个数相等,我们就向下扩展。然后需要向下扩展的个数就变成了 4 个,而需要向上扩展的个数是 1 个,所以此时我们向上扩展。接着,需要向上扩展的个数变成了 6 个,需要向下扩展的个数是 4 个,我们就向下扩展......直到相遇。

双向扩展的好处,我们粗略的估计一下时间复杂度。

假设 beginwordendword 之间的距离是 d。每个节点可以扩展出 k 个节点。

那么正常的时间复杂就是 k^d

双向搜索的时间复杂度就是 k^{d/2} + k^{d/2}

public List<List<String>> findLadders(String beginWord, String endWord, List<String> wordList) {
    List<List<String>> ans = new ArrayList<>();
    if (!wordList.contains(endWord)) {
        return ans;
    }
    // 利用 BFS 得到所有的邻居节点
    HashMap<String, ArrayList<String>> map = new HashMap<>();
    bfs(beginWord, endWord, wordList, map);
    ArrayList<String> temp = new ArrayList<String>();
    // temp 用来保存当前的路径
    temp.add(beginWord);
    findLaddersHelper(beginWord, endWord, map, temp, ans);
    return ans;
}

private void findLaddersHelper(String beginWord, String endWord, HashMap<String, ArrayList<String>> map,
                               ArrayList<String> temp, List<List<String>> ans) {
    if (beginWord.equals(endWord)) {
        ans.add(new ArrayList<String>(temp));
        return;
    }
    // 得到所有的下一个的节点
    ArrayList<String> neighbors = map.getOrDefault(beginWord, new ArrayList<String>());
    for (String neighbor : neighbors) {
        temp.add(neighbor);
        findLaddersHelper(neighbor, endWord, map, temp, ans);
        temp.remove(temp.size() - 1);
    }
}

//利用递归实现了双向搜索
private void bfs(String beginWord, String endWord, List<String> wordList, HashMap<String, ArrayList<String>> map) {
    Set<String> set1 = new HashSet<String>();
    set1.add(beginWord);
    Set<String> set2 = new HashSet<String>();
    set2.add(endWord);
    Set<String> wordSet = new HashSet<String>(wordList);
    bfsHelper(set1, set2, wordSet, true, map);
}

// direction 为 true 代表向下扩展,false 代表向上扩展
private boolean bfsHelper(Set<String> set1, Set<String> set2, Set<String> wordSet, boolean direction,
                          HashMap<String, ArrayList<String>> map) {
    //set1 为空了,就直接结束
    //比如下边的例子就会造成 set1 为空
    /*  "hot"
        "dog"
        ["hot","dog"]*/
    if(set1.isEmpty()){
        return false;
    }
    // set1 的数量多,就反向扩展
    if (set1.size() > set2.size()) {
        return bfsHelper(set2, set1, wordSet, !direction, map);
    }
    // 将已经访问过单词删除
    wordSet.removeAll(set1);
    wordSet.removeAll(set2);

    boolean done = false;

    // 保存新扩展得到的节点
    Set<String> set = new HashSet<String>();

    for (String str : set1) {
        //遍历每一位
        for (int i = 0; i < str.length(); i++) {
            char[] chars = str.toCharArray();

            // 尝试所有字母
            for (char ch = 'a'; ch <= 'z'; ch++) {
                if(chars[i] == ch){
                    continue;
                }
                chars[i] = ch;

                String word = new String(chars);

                // 根据方向得到 map 的 key 和 val
                String key = direction ? str : word;
                String val = direction ? word : str;

                ArrayList<String> list = map.containsKey(key) ? map.get(key) : new ArrayList<String>();
                
                //如果相遇了就保存结果
                if (set2.contains(word)) {
                    done = true;
                    list.add(val);
                    map.put(key, list);
                }

                //如果还没有相遇,并且新的单词在 word 中,那么就加到 set 中
                if (!done && wordSet.contains(word)) {
                    set.add(word);
                    list.add(val);
                    map.put(key, list);
                }
            }
        }
    }

    //一般情况下新扩展的元素会多一些,所以我们下次反方向扩展  set2
    return done || bfsHelper(set2, set, wordSet, !direction, map);

}

最近事情比较多,这道题每天想一想,陆陆续续拖了好几天了。这道题本质上就是在正常的遍历的基础上,去将一些分支剪去,从而提高速度。至于方法的话,除了我上边介绍的实现方式,应该也会有很多其它的方式,但其实本质上是为了实现一样的东西。另外,双向搜索的方法,自己第一次遇到,网上搜了一下,看样子还是比较经典的一个算法。主要就是用于解决已知起点和终点,去求图的最短路径的问题。

更多详细通俗题解详见 leetcode.wang

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,732评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,496评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,264评论 0 338
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,807评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,806评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,675评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,029评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,683评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 41,704评论 1 299
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,666评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,773评论 1 332
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,413评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,016评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,978评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,204评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,083评论 2 350
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,503评论 2 343

推荐阅读更多精彩内容