在二叉树中有两个结点m和n,若m是n的祖先,则使用后序遍历可以找到从m到n的路径。

非递归:

#include <iostream>
#include <stack>

using namespace std;

// 二叉树结点定义
struct TreeNode 
{
    int val;
    TreeNode* left;
    TreeNode* right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

// 非递归后序遍历查找路径m->n
void postorder(TreeNode* root, TreeNode* n) 
{
    stack<TreeNode*> stk;
    TreeNode* cur = root;
    TreeNode* prev = NULL;

    while(cur || !stk.empty())
    {
        // 将当前节点及其左节点全部入栈
        while(cur)
        {
            stk.push(cur);
            cur = cur->left;
        }

        cur = stk.top();
        // 如果栈顶没有右子节点或右子节点已经访问完
        if(!cur->right || cur->right == prev)
        {
            if (cur->val == n->val) // 访问到目标结点n时
            {
                while(!stk.empty()) // 输出栈中元素
                {
                    cout << stk.top()->val << ' ';
                    stk.pop();
                }
                return ;
            }
            stk.pop();
            prev = cur;
            cur = nullptr;
        }
        else // 有右子节点且还未访问
            cur = cur->right;
    }
}

int main() 
{
    // 构建一个二叉树
    TreeNode* root = new TreeNode(1);
    root->left = new TreeNode(2);
    root->right = new TreeNode(3);
    root->left->left = new TreeNode(4);
    root->left->right = new TreeNode(5);
    root->right->left = new TreeNode(6);
    root->right->right = new TreeNode(7);

    TreeNode* m = root;         // 选择m和n
    TreeNode* n = root->right->left;

    // 采用非递归后序遍历,最后访问根结点,访问到目标结点n时,
    // 栈中所有元素均为该结点的祖先,依次出栈打印即可。
    postorder(m, n);            // 调用后序遍历查找路径m->n

    return 0;
}

递归:

#include <iostream>

using namespace std;

// 二叉树结点定义
struct TreeNode 
{
    int val;
    TreeNode* left;
    TreeNode* right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

// 后序遍历查找路径m->n
bool postorder(TreeNode* root, TreeNode* n) 
{
    // 边界情况:空结点 -> 查找失败
    if (!root) 
        return false;

    // 递归遍历左右子树
    if (root == n || postorder(root->left, n) || postorder(root->right, n)) 
    {
        // 如果当前结点是目标结点n,或者在左右子树中找到了目标结点
        // 输出当前结点的值,并返回true
        cout << root->val << ' ';
        return true;
    }
}

int main() 
{
    // 构建一个二叉树
    TreeNode* root = new TreeNode(1);
    root->left = new TreeNode(2);
    root->right = new TreeNode(3);
    root->left->left = new TreeNode(4);
    root->left->right = new TreeNode(5);
    root->right->left = new TreeNode(6);
    root->right->right = new TreeNode(7);

    TreeNode* m = root;         // 选择m和n
    TreeNode* n = root->right->left;

    // 在后续遍历退回时访问根节点,就可以从下向上把从m到n的路径上的节点输出
    postorder(m, n);            // 调用后序遍历查找路径m->n

    return 0;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

程序garbage

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值