【剑指offer第24题】二叉树中和为某一值的路径（java实现）

解答思路

输入一棵二叉树和一个整数，打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。

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

          5

         / \

        4   8

       /   / \

      11  13  4

     /  \    / \

    7    2  5   1

返回:

[

[5,4,11,2],

[5,8,4,5]

]

题意很清楚，就是遍历所有从根结点到叶子结点的所有情况，只要这些结点值的和符合题意就是结果之一。关键就是这个遍历过程中，还要计算这些值的和，就是到了每个结点时候都要计算从根结点到此结点的和，用一个数组存储这些路径值。这里就要用到回溯的思想。遍历所有根结点到叶子结点情况就是深度遍历，也就是前序遍历。在这个递归的过程中，就需要回溯这一步。

到了叶子结点只需判断值和是否和目标值相等，而递归回到它的父亲结点时候，因为此时这个点不符合题意，下一步会继续回去它的父亲结点，所以此时需要删除路径数组这个结点的值，因为它已经不符合题意了。

代码实现

	/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;
    }
}
*/
import java.util.ArrayList;
public class Solution {
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        ArrayList<Integer> temp = new ArrayList<>();
        int tempData = 0;
        preOrderRecursive(res , temp , tempData , root , target);
        return res;
    }
    //前序遍历递归函数
    private void preOrderRecursive(ArrayList<ArrayList<Integer>> res , ArrayList<Integer> temp , int tempData , TreeNode root , int sum) {
        if (root == null) return;
        //若是叶子结点
        if (root.left == null && root.right == null) {
            tempData += root.val;
            if (tempData == sum) {
                temp.add(root.val);
                res.add(new ArrayList(temp));
                int count = temp.size();
                temp.remove(count - 1);
                tempData -=  root.val;
            } else tempData -=  root.val;
            return;
        }
        temp.add(root.val);
        tempData += root.val;
        //前序遍历
        preOrderRecursive(res , temp , tempData , root.left , sum);
        preOrderRecursive(res , temp , tempData , root.right , sum);
        //回溯，删除结点值。
        int count = temp.size();
        temp.remove(count - 1);
        tempData -=  root.val;
    }
}

总结

重点就是回溯过程中，回到非叶子结点时就要删除路径中这个结点值，然后，有个地方需要注意就是，当寻找了一条合适路径，也就是结果之一，把它放进res数组中是，不要直接就把temp数组放进去了，应该新建一个数组存储，在放进去。因为这里res和temp数组都是全局变量。不能把temp放进res中，temp是一直变化的。正确写法：

res.add(new ArrayList(temp));

时间复杂度：O(n)需要遍历每一个结点。空间复杂度：O(n),最坏情况下，二叉树退化成链表，temp需要存储所有的结点。