工作生活

输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入

2019-07-02  本文已影响0人  小明今晚加班

解题思路:首先判断根结点是否为空,不为空的话判断当前结点是否是叶子结点;如果当前存入集合的值小于target,且有左右子树,则递归;如果当前存入集合的值等于target,且为叶子结点,则打印路径;如果当前存入集合的值大于target,则不用递归了。(题目中,默认所有结点的值都为整数)
代码如下:

import java.util.ArrayList;
/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
    ArrayList<ArrayList<Integer>> list = new ArrayList<>();
    ArrayList<Integer> arr = new ArrayList<>();
    int num  =0;
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
        if(root==null)
            return list;
        boolean leaf = root.left==null && root.right==null;
        num+=root.val;
        arr.add(root.val);
        if(num==target && leaf){
            ArrayList<Integer> path = new ArrayList<>();
            for(int i=0;i<arr.size();i++){
                path.add(arr.get(i));
            }
            list.add(path);
        }
        if(num<target && root.left!=null){
            FindPath(root.left, target);
        }
        if(num<target && root.right!=null){
            FindPath(root.right, target);
        }
        //当前节点值的和大于target,那就没必要递归,同时把最后添加的值给移除
        num-=root.val;
        arr.remove(arr.size()-1);
        return list;
    }
}
上一篇 下一篇

猜你喜欢

热点阅读