剑指 Offer - 24:二叉树中和为某一值的路径

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/AnselLyy/article/details/84999963

题目描述

输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)

题目链接:https://www.nowcoder.com/practice/b736e784e3e34731af99065031301bca

public class Solution {
	ArrayList<ArrayList<Integer>> result = new ArrayList<>();
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root, int target) {
        
        if (root == null) {
            return result;
        }
        int root_val = root.val;
        ArrayList<Integer> list = new ArrayList<>();
        list.add(root_val);
        rec(root, list, root_val, target);
        Collections.sort(result, (o1, o2) -> o2.size() - o1.size());
        return result;
    }
    
    private void rec(TreeNode node, ArrayList<Integer> list, int sum, int target) {
        if (node.left == null && node.right == null) {
            if (sum == target) {
                result.add(new ArrayList<>(list));
            }
            return;
        }
        
        // 递归左子树
        if (node.left != null) {
            Integer value = node.left.val;
            list.add(value);
            rec(node.left, list, sum+value, target);
            list.remove(value);
        }
        // 递归右子树
        if (node.right != null) {
            Integer value = node.right.val;
            list.add(value);
            rec(node.right, list, sum+value, target);
            list.remove(value);
        }
    }
}

猜你喜欢

转载自blog.csdn.net/AnselLyy/article/details/84999963