算法 |《剑指offer》面试题34.二叉树中和为某一值的路径

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

示例:

给定如下二叉树,以及目标和 sum = 22,

          5
         / \
        4   8
       /   / \
      11  13  4
     /  \    / \
    7    2  5   1
返回:

[
[5,4,11,2],
[5,8,4,5]
]

提示:

节点总数 <= 10000

题解:
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    List<List<Integer>> res = new ArrayList();
    public List<List<Integer>> pathSum(TreeNode root, int sum) {
        traceSum(root, sum, new ArrayList());
        return res;
    }

    private void traceSum(TreeNode node, int target, List<Integer> collector) {
        if(node == null) return;
        collector.add(node.val);
        target -= node.val;
        if(target == 0 && node.left == null && node.right == null) {
            res.add(new ArrayList(collector));
        } else {
            traceSum(node.left, target, collector);
            traceSum(node.right, target, collector);
        }
        //回溯 后退,因为整个过程用的同一个收集器
        collector.remove(collector.size() - 1);
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章