给定一棵二叉树,二叉树每个节点的值唯一,从根节点开始找出路径上的

2019-01-07  本文已影响0人  白驹过隙_a

思路:以先序遍历(根节点-左子树-右子树)的方式访问二叉树的每一个节点,记录根节点到遍历到这个节点的所有节点值之和,同时用一个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<ArrayList<Integer>> FindPath(TreeNode root,int target) {
        ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();
        if(root==null){
            return result;
        }
        Stack<Integer> stack = new Stack<>();
        FindPath(root, target, stack, result);
        return result;
    }
    
    private void FindPath(TreeNode root, int target, Stack<Integer> stack, ArrayList<ArrayList<Integer>> result) {
        if(root==null){
            return;
        }
        if(root.left==null&&root.right==null){
            if(root.val==target){
                ArrayList<Integer> 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();
        }
    }
}

上一篇下一篇

猜你喜欢

热点阅读