微软面试100题系列---在二叉树中找出和为某一定值的所有路径

题目

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

实现

递归+回溯
思路：保存搜索路径，记录当前路径上所有元素的和。如果当前节点为叶子节点，并且当前节点值和sum和=data，则满足条件，打印后递归返回到父节点；如果不是叶子节点，则继续访问子节点；

• 当访问到某一节点时，把该节点的值添加到当前变量和sum中，且把该节点压入栈中；
• 若节点为叶子节点，且sum==期望的和，则打印出栈内容；
• 若节点不是叶子节点，继续访问左孩子节点和右孩子节点；
• 从栈中移除该节点，并从sum和中减去该节点的值（回溯）；

代码实现：

import java.util.Stack;


public class Num4 {

    public static int sum=0;

    public static Stack stack=new Stack<BinaryTreeNode>();
    public static Stack stack_2=new Stack<BinaryTreeNode>();

    public static void main(String[] args) {
        BinaryTreeNode root=null;

        //create a binaryTree to test
        root=createTree(root);

        pathTree(root,22);
    }

    public static BinaryTreeNode createTree(BinaryTreeNode root){
        root=new BinaryTreeNode(8);
        BinaryTreeNode node=new BinaryTreeNode(5);
        root.left=node;
        node=new BinaryTreeNode(12);
        root.right=node;
        node=new BinaryTreeNode(7);
        root.left.left=node;

        return root;
    }

    public static void pathTree(BinaryTreeNode root,int value){
        sum+=root.value;
        stack.push(root);
        if(root.left==null && root.right==null && sum==value){
            printPath();
        }
        if(root.left!=null){
            pathTree(root.left,value);
        }
        if(root.right!=null){
            pathTree(root.right,value);
        }
        if(!stack.isEmpty())
        {
            stack.pop();
            sum-=root.value;
        }

    }

    private static void printPath() {
        while(!stack.isEmpty()){
            BinaryTreeNode node= (BinaryTreeNode) stack.pop();
            stack_2.push(node);
        }
        while(!stack_2.isEmpty()){
            BinaryTreeNode node=(BinaryTreeNode) stack_2.pop();
            stack.push(node);
            System.out.print(node.value+"     ");
        }
        System.out.println();
    }
}

class BinaryTreeNode{
    int value;
    BinaryTreeNode left;
    BinaryTreeNode right;

    public BinaryTreeNode(int value){
        this.value=value;
        this.left=null;
        this.right=null;
    }
}