LeetCode学习笔记——路径之和（深度优先搜索）

个人博客：The Blog Of WaiterXiaoYY 欢迎来互相交流学习。

前言

深度优先搜索算法（英语：Depth-First-Search，DFS）

是一种用于遍历或搜索树或图的算法。

沿着树的深度遍历树的节点，尽可能深的搜索树的分支。

当节点v的所在边都己被探寻过，搜索将回溯到发现节点v的那条边的起始节点。

这一过程一直进行到已发现从源节点可达的所有节点为止。

如果还存在未被发现的节点，则选择其中一个作为源节点并重复以上过程，

整个进程反复进行直到所有节点都被访问为止。属于盲目搜索。

1. 路径之和 I

给定一个二叉树和一个目标和，判断该树中是否存在根节点到叶子节点的路径，这条路径上所有节点值相加等于目标和。

说明: 叶子节点是指没有子节点的节点。

示例:
给定如下二叉树，以及目标和 sum = 22，

          5
         / \
        4   8
       /   / \
      11  13  4
     /  \      \
    7    2      1

返回 true, 因为存在目标和为 22 的根节点到叶子节点的路径 5->4->11->2。

解题思路

注意这道题的要求，是从根节点到叶子结点，

那么我们就可以采用深度优先搜索，利用递归，从根节点开始，一直搜索到叶子结点，

每向下递归一次，sum就减掉该结点的值，

如果到叶子结点，sum的值等于叶子结点的值，

说明这条路径是正确的，就返回true。

代码

class Solution {
    public boolean hasPathSum(TreeNode root, int sum) {
        if(root == null)
            return false;
        if(root.left == null && root.right == null){
            if(sum == root.val)   //sum等于叶子结点的值
                return true;
        }
        return hasPathSum(root.left, sum - root.val) || hasPathSum(root.right, sum - root.val);  //这里是||，左子树或者右子树正确都可以
    }
}

2. 路径之和 II

给定一个二叉树和一个目标和，找到所有从根节点到叶子节点路径总和等于给定目标和的路径。

说明: 叶子节点是指没有子节点的节点。

示例:
给定如下二叉树，以及目标和 sum = 22，

          5
         / \
        4   8
       /   / \
      11  13  4
     /  \    / \
    7    2  5   1

返回:

[
   [5,4,11,2],
   [5,8,4,5]
]

解题思路

这道题和路径总和I是类似的，

但是这道题的难度在于，

一棵树中，不只有一条正确的路径，而且还要把所有正确的路径保存在列表中，

同样要注意的是，还是从根节点到叶子结点，所以我们还是用深度优先搜索，利用递归来完成，

当我们从根节点搜索到叶子结点的过程中，我们将每一个结点的值保存在一个数组中，

当到叶子结点的时候，判断这条路径是否正确，

如果正确，则将数组记录的结点值放入到一个列表数组中，

如果不正确，返回上个结点时，将该数组删除。

代码

class Solution {
    List<List<Integer>> list = new ArrayList<>();
    ArrayList<Integer> res = new ArrayList<>();
    public List<List<Integer>> pathSum(TreeNode root, int sum) {
        if(root == null)
            return list;
        sum -= root.val;
        res.add(root.val); //将结点值加入到数组中
        if(root.left == null && root.right == null){
            if(sum == 0)
                 list.add(new  ArrayList<>(res)); //复制一份，将数组加入到列表中
        }
        if(root.left != null)
            pathSum(root.left, sum);
        if(root.right != null)
            pathSum(root.right, sum);
        res.remove(res.size()-1);   //返回上个结点前，要将数组删除
        return list;
    }
}

3. 路径之和 III

给定一个二叉树，它的每个结点都存放着一个整数值。

找出路径和等于给定数值的路径总数。

路径不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的（只能从父节点到子节点）。

二叉树不超过1000个节点，且节点数值范围是 [-1000000,1000000] 的整数。

示例：

root = [10,5,-3,3,2,null,11,3,-2,null,1], sum = 8

      10
     /  \
    5   -3
   / \    \
  3   2   11
 / \   \
3  -2   1

返回 3。和等于 8 的路径有:

1.  5 -> 3
2.  5 -> 2 -> 1
3.  -3 -> 11

解题思路

这道题相比前面两道题，难点在于，不一定是从根节点开始，也不一定是叶子结点结束，

所以哪个结点都有可能是路径的起始点，

这里就涉及到双重递归， 双重递归就是，先对结点进行递归遍历，这里也是采用深度优先搜索，

这样子，每一个结点都作为根节点去深度搜索判断该路径是否正确，

现在的问题就是，在不到叶子结点的情况下，如何判断是否正确，

我们可以和之前一样，每递归一次，就将sum值减去结点值，

当sum==0时，就说明正确，计数count++，到最后的时候，返回count；

代码

class Solution {
    int count = 0;
    public int pathSum(TreeNode root, int sum) {
        if(root == null)
            return 0;
        dfs(root,sum);
        //先对结点进行遍历 ，让每一个结点都作为根节点
        pathSum(root.left,sum);
        pathSum(root.right,sum);
        return count;
    }

    public void dfs(TreeNode root, int sum){
        if(root == null)
            return ;
        sum -= root.val;
        if(sum == 0)   //sum==0，则说明路径正确
            count++;
        dfs(root.left,sum);
        dfs(root.right,sum);

    }
}

个人博客：The Blog Of WaiterXiaoYY 欢迎来互相交流学习。

---

整理于2020.3.15

END