给定一棵二叉树,二叉树每个节点的值唯一,从根节点开始找出路径上的所有节点的节点值之和等于规定值的路径

思路:以先序遍历(根节点-左子树-右子树)的方式访问二叉树的每一个节点,记录根节点到遍历到这个节点的所有节点值之和,同时用一个list存储遍历的路径,若节点和等于给定值则返回路径,直到叶子节点,结束递归。

用到的数据结构:ArrayList,stack

代码如下:

import java.util.ArrayList;
import java.util.Stack;
/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;
    public TreeNode(int val) {
        this.val = val;
    }
}
*/
public class Solution {
    public ArrayList> FindPath(TreeNode root,int target) {
        ArrayList> result = new ArrayList>();
        if(root==null){
            return result;
        }
        Stack stack = new Stack<>();
        FindPath(root, target, stack, result);
        return result;
    }
    
    private void FindPath(TreeNode root, int target, Stack stack, ArrayList> result) {
        if(root==null){
            return;
        }
        if(root.left==null&&root.right==null){
            if(root.val==target){
                ArrayList list = new ArrayList<>();
                for(int i:stack){
                    list.add(i);
                }
                list.add(root.val);
                result.add(list);
            }
        }else{
            stack.push(root.val);
            FindPath(root.left, target-root.val, stack, result);
            FindPath(root.right, target-root.val, stack, result);
            stack.pop();
        }
    }
}

你可能感兴趣的:(给定一棵二叉树,二叉树每个节点的值唯一,从根节点开始找出路径上的所有节点的节点值之和等于规定值的路径)