剑指Offer_24

发表于 | 分类于

题目

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

解题思路

从树的根点开始，直至叶节点的路径和为target，打印路径。

采用带记忆的深度优先搜索。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
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 {
    private ArrayList<ArrayList<Integer>> AA = new ArrayList<>();
    private ArrayList<Integer> A = new ArrayList<>();
    
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
        if(root == null){
            return AA;
        }
        //将被试探的节点加入
        A.add(root.val);
        target -= root.val;
        if(target == 0 && root.left == null && root.right == null){
            AA.add(new ArrayList<Integer>(A));
        }
        //试探节点的左右子树节点
        FindPath(root.left, target);
        FindPath(root.right, target);
        
        //该节点被试探完，就从记忆数组中丢弃
        A.remove(A.size()-1);
        
        return AA;
    }
}