Interview100-4 二叉搜索树中寻找元素和为给定值的路径

最新推荐文章于 2023-02-25 22:41:55 发布

Ethan_pika

最新推荐文章于 2023-02-25 22:41:55 发布

阅读量208

点赞数

分类专栏：面试算法100题文章标签： algorithm interview

本文链接：https://blog.csdn.net/yitian_z/article/details/88084140

版权

面试算法100题专栏收录该内容

50 篇文章 0 订阅

订阅专栏

题目

输入一个整数n和一个二叉搜索树。从数的根节点开始向下访问子节点直到叶子结点所经过的所有节点形成一条路径。在给定的二叉搜索树中寻找和为给定值n的所有路径。

例如输入整数22，如下的二叉搜索树：

则打印出两条满足条件的路径：10-5-7 和 10-12

解法

思想：由于还是树形结构，因此可以想到递归+回溯的方法。手动维护一个stack，存储从根结点到叶子节点的遍历的路径，如果和为指定的数字，则打印路径。如果不是则返回上层节点继续寻找到其他叶子节点。伪代码如下;

二叉搜索树的节点结构：

BSTreeNode {
    int value;
    BSTreeNode left;
    BSTreeNode right;
}

算法伪代码：

FIND-SUM-PATH(BSTreeNode root, int sum):
    int path[Tree.height]; // 使用数组模拟stack的功能
    BACK-TRACK-PATH(root, sum, path, 0);

BACK-TRACK-PATH(BSTreeNode root, int sum, int[] path, top):
    path[top++] = root.value;
    sum -= root.value;
    if root.left == null and root.right == null // 如果已经到达叶子节点
        if sum == 0
            PRINT-PATH(path, top);

    else // 否则进行递归
        if root.left != null
            BACK-TRACK-PATH(root.left, sum, path, top);
        else 
            BACK-TRACK-PATH(root.right, sum, path, top);
    top--; 
    sum += root.value;

PRINT-PATH(int[] path, int top):
    for i = 0 to top:
        print path[i]