JZ24 二叉树中和为某一值的路径

原题链接

描述

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

示例

输入：{10,5,12,4,7},22
返回值：[[10,5,7],[10,12]]

思路

从根结点开始向下遍历，每经过一个结点就把 target 的值减去该结点的 val，并把该结点的 val 加进 arrayList<Integer>当中 ，直到该结点是叶子结点。此时如果 target 为0说明路径上各结点之和就是 target，此时把 arrayList 添加到arrayList<arrayList<>>，否则说明不行，删除该元素。

解答

import java.util.ArrayList;

public class Solution {
    
    private ArrayList<ArrayList<Integer>> listAll = new ArrayList<ArrayList<Integer>>();
    private ArrayList<Integer> list = new ArrayList<Integer>();

    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root, int target) {
        if (root == null) return listAll;
        
        list.add(root.val);
        target -= root.val;
        
        // 叶结点，并且路径和为 target
        if (target == 0 && root.left == null && root.right == null)
            listAll.add(new ArrayList<Integer>(list));
        
        FindPath(root.left, target);
        FindPath(root.right, target);
        
        list.remove(list.size() - 1);
        
        return listAll;
    }
}