在二叉树中查找和为某一值的所有路径

  1. 在二叉树中,我们求根节点到叶子节点和为某一数值的路径,并打印出来。
  2. 如下图所示:
  3. 在这里插入图片描述
  4. 我们求解和为16的路径,那么我们可以求解出两条路径,一条为5-4-7,一条为5-8-3。那么在处理二叉树的时候我们会用递归的算法。由于涉及到存储路径,我们可以用堆栈的思想,并且在进行下一步计算之前,对现场进行恢复。整体的实现代码如下所示:
  5. #include
    #include
    using namespace std;
    typedef struct node
    {
    int val;
    node *lChild;
    node *rChild;
    node(int val)
    {
    this->val = val;
    this->lChild = nullptr;
    this->rChild = nullptr;
    }
    node()
    {
    this->lChild = nullptr;
    this->rChild = nullptr;
    }
    }TreeNode;
    TreeNode *createtree(int i, int intarr[], int length)//根据数组创建二叉树
    {
    TreeNode *node = nullptr;
    if (i == 0)
    {
    node = new TreeNode(intarr[i]);
    node->lChild = createtree(i * 2 + 1, intarr, length);
    node->rChild = createtree(i * 2 + 2, intarr, length);
    }
    else if (i < length)
    {
    node = new TreeNode(intarr[i]);
    node->lChild = createtree(i * 2 + 1, intarr, length);
    node->rChild = createtree(i * 2 + 2, intarr, length);
    }
    return node;
    }
    void display(vector path)//打印存储路径
    {
    for (auto x : path)
    {
    cout << x << " ";
    }
    cout << endl;
    }
    bool isLeaf(TreeNode *node)//判断是否为叶子节点
    {
    return (!node->lChild) && (!node->rChild);
    }
    void findPath(TreeNode *node, int expext, int &cursum, vector &path)
    //node是需要计算的节点,expext是期望值,cursum是当前的值,path是存储的路径
    {
    if (node==nullptr)
    {
    return;
    }
    if (cursum+node->val == expext && isLeaf(node))//当且仅当cursum和node->val和为expext,且node为叶子节点的时候,执行打印路径操作。
    {
    path.push_back(node->val);
    display(path);
    path.pop_back();//执行完打印路径的操作,需要把node节点pop出来,一遍进行下一次计算的时候不出错。
    return;
    }
    else
    {
    cursum = cursum + node->val;//上述条件不满足的时候,把node->val加到cursum中,并且把node push到path里面。对其两个节点进行计算。
    path.push_back(node->val);
    if (node->lChild)
    {
    findPath(node->lChild, expext, cursum, path);//计算左节点是否满足条件。
    }
    if (node->rChild)
    {
    findPath(node->rChild, expext, cursum, path);//计算右节点是否满足条件。
    }
    cursum = cursum - node->val;//以下两个步骤都是恢复现场,以便进行下一个节点计算的时候,不会出错。
    path.pop_back();
    return;
    }
    }
    int main()
    {
    int intarr[6] = { 5,4,8,7,1,3 };
    TreeNode *root = createtree(0, intarr, 6);
    int cursum = 0;
    int expectsum = 16;
    vector path;
    findPath(root, expectsum, cursum, path);
    system(“pause”);
    return 0;
    }
    执行结果:
    在这里插入图片描述
    在本算法中使用了回溯的思想,回溯的思想的最主要的就是要保存现场和恢复现场,以便进行下一次计算的时候不会把当前的计算结果杂糅到下一次的计算结果中,导致计算结果不正确。
以下是使用C语言实现在二叉树找出和为某一值的所有路径的示例代码: ```c #include <stdio.h> #include <stdlib.h> typedef struct TreeNode { int val; struct TreeNode* left; struct TreeNode* right; } TreeNode; void findPath(TreeNode* root, int sum, int* path, int depth) { if (root == NULL) { return; } path[depth] = root->val; int i, curSum = 0; for (i = depth; i >= 0; --i) { curSum += path[i]; if (curSum == sum) { int j; for (j = i; j < depth; ++j) { printf("%d->", path[j]); } printf("%d\n", path[depth]); } } findPath(root->left, sum, path, depth + 1); findPath(root->right, sum, path, depth + 1); } int main() { // 构造二叉树 TreeNode* root = (TreeNode*)malloc(sizeof(TreeNode)); root->val = 10; root->left = (TreeNode*)malloc(sizeof(TreeNode)); root->left->val = 5; root->right = (TreeNode*)malloc(sizeof(TreeNode)); root->right->val = 12; root->left->left = (TreeNode*)malloc(sizeof(TreeNode)); root->left->left->val = 4; root->left->right = (TreeNode*)malloc(sizeof(TreeNode)); root->left->right->val = 7; // 在二叉树查找和为22的路径 int path[100]; findPath(root, 22, path, 0); // 释放二叉树空间 free(root->left->left); free(root->left->right); free(root->left); free(root->right); free(root); return 0; } ``` 该示例代码,我们先构造了一棵二叉树,并在其查找和为22的路径。在查找路径的过程,我们使用了深度优先遍历的方法,通过一个数组来保存当前已经走过的路径,每次遍历到一个节点,则将该节点的值加入路径。在加入值之后,我们检查路径是否存在和为目标值的子路径,如果存在,则输出该路径。最后,我们递归遍历该节点的左子树和右子树,以便查找所有可能的路径。 需要注意的是,该示例代码并未考虑二叉树存在负数的情况,如有需要,可以根据实际情况进行修改。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值