微软算法100题04 二叉树中查找和为某值的所有路径

4.在二元树中找出和为某一值的所有路径
题目：输入一个整数和一棵二元树。
从树的根结点开始往下访问一直到叶结点所经过的所有结点形成一条路径。
打印出和与输入整数相等的所有路径。
例如输入整数22 和如下二元树
10
/ \
5 12
/ \
4 7
则打印出两条路径：10, 12 和10, 5, 7

 

思路: 递归  

为了记录路径 需要采用一个辅助数据结构记录正确路径上的节点 这里采用list

1. 从根节点开始 将根节点放入List 将总额sum减去根节点的值 如果等于零 说明找到正确路径上的节点 则打印list

2. 如果该节点有左子树 则用1的逻辑递归处理左子树

3. 如果该节点有右子树 则用1的逻辑递归处理右子树

4. 删除list里的最后一个节点 因为该节点要么为非正确节点 要么为已访问过的正确节点

 

package com.rui.microsoft;

import java.util.ArrayList;
import java.util.List;

public class Test04 {
    
    
    public static void main(String[] args) {
        Test04 test = new Test04();
        Node root = test.init();
        
        List<Node> list = new ArrayList<Node>();
        test.findRoute(root, 22, list);
        
    }
    
    private void findRoute(Node node, int sum, List<Node> list) {
        list.add(node);
        sum -= node.value;
        
        if(isLeaf(node)){
            //if sum is zero, it means current node is the last node on the path
            if(sum == 0) print(list);
        }else{
            if(null != node.left){
                findRoute(node.left, sum, list);
            }
            
            if(null != node.right){
                findRoute(node.right, sum, list);
            }
        }
        
        //if we reach here, it means current node is not on the path or have already been displayed
        //remove it
        //current node is on the top of the list
        list.remove(list.size()  - 1);
    }
    
    private void print(List<Node> nodes){
        for(Node node: nodes){
            System.out.print(" " + node.value);
        }
        System.out.println("");
    }

    private boolean isLeaf(Node node) {
        if(null != node.left || null != node.right) return false;
        return true;
    }

    private Node init(){
        Node left2 = new Node(4,null,null);
        Node right2 = new Node(7,null,null);
        Node left = new Node(5,left2,right2);
        Node right = new Node(12, null, null);
        Node root = new Node(10,left,right);
        return root;
    }
    
    class Node {
        int value;
        Node left;
        Node right;
        
        public Node(int value){
            this.value = value;
        }
        
        public Node(int value, Node left, Node right){
            this.value = value;
            this.left = left;
            this.right = right;
        }
    }

}