剑指offer — 二叉树中的某一路径

来源:互联网 发布:内置软件卸载工具 编辑:程序博客网 时间:2024/06/13 23:02

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

使用dfs可以轻松解决

java

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 {    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {ArrayList<ArrayList<Integer>> result = new ArrayList<>();        if (root == null) {            return result;        }        ArrayList<Integer> path = new ArrayList<>();        path.add(root.val);        dfs(root, target - root.val, result, path);        return result;    }    private void dfs(TreeNode root,                      int target,                      ArrayList<ArrayList<Integer>> result,                     ArrayList<Integer> path) {        if (root == null) {            return;        }        if (root.left == null && root.right == null && target == 0) {            result.add(new ArrayList<Integer>(path));            return;        }        if (root.left != null) {            path.add(root.left.val);dfs(root.left, target - root.left.val, result, path);            path.remove(path.size() - 1);        }        if (root.right != null) {            path.add(root.right.val);            dfs(root.right, target - root.right.val, result, path);            path.remove(path.size() - 1);        }    }}