剑指offer-题25:二叉树中和为某一值的路径

本文介绍了一种使用Java和Python实现的算法,该算法能够找出一棵二叉树中所有节点值之和等于指定整数的路径。通过递归地遍历二叉树,并维护一个路径栈来跟踪当前路径,当到达叶子节点且路径和等于目标值时,将路径加入到结果列表中。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目描述

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

实验平台:牛客网


解决思路:

这里写图片描述
这里写图片描述
这里写图片描述
这里写图片描述
这里写图片描述

java:

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 {
    int curSum = 0;
    Stack<Integer> pathStack = new Stack<>();
    ArrayList<ArrayList<Integer>> pathList = new ArrayList<>();
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
        if (root == null) {
            return pathList;
        }
        curSum += root.val;
        pathStack.push(root.val);

        boolean isLeafNode = root.left == null && root.right == null;
        if (curSum == target && isLeafNode) {
            ArrayList<Integer> path = new ArrayList<Integer>();
            for (Integer value : pathStack) {
                path.add(value);
            }
            pathList.add(path);
        }
        if (root.left != null) {
            FindPath(root.left, target);
        }
        if (root.right != null) {
            FindPath(root.right, target);
        }
        pathStack.pop();
        curSum -= root.val;
        return pathList;
    }
}

python:

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    path_list = []
    stack = []
    cur_sum = 0
    # 返回二维列表,内部每个列表表示找到的路径
    def FindPath(self, root, expectNumber):
        self.path_list = []
        self.stack = []
        self.cur_sum = 0
        return self.get_path(root, expectNumber)

    def get_path(self, root, expectNumber):
        if root is None:
            return self.path_list
        self.cur_sum += root.val
        self.stack.append(root.val)

        is_leaf = root.right is None and root.left is None
        if self.cur_sum == expectNumber and is_leaf:
            temp_path = self.stack[:]
            self.path_list.append(temp_path)
        if root.left:
            self.get_path(root.left, expectNumber)
        if root.right:
            self.get_path(root.right, expectNumber)
        self.stack.pop()
        self.cur_sum -= root.val
        return self.path_list
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值