二叉树遍历【先序】 617. 合并二叉树c++

来自:leecode
在这里插入图片描述

 TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
    if(t1==nullptr)
    return t2;
    if(t2==nullptr)
    return t1;
    //同时先序遍历两个二叉树
    stack<TreeNode*> stack1;
    stack<TreeNode*> stack2;
    //根节点入栈
    stack1.push(t1);
    stack2.push(t2);

    while (!stack1.empty())        
    {
        //节点出栈
        if(stack2.empty()) return t1;
        TreeNode* ta=stack1.top();
        stack1.pop();
        TreeNode* tb=stack2.top();
        stack2.pop();
        //先序遍历根
        //val相加
        ta->val+=tb->val;
        //先序遍历左子树
        if(ta->left==nullptr) {ta->left=tb->left;}
        else {
            stack1.push(ta->left);
            stack2.push(tb->left);
        }
        //先序遍历右子树
        if(ta->right==nullptr) {ta->right=tb->right;}
        else {
            stack1.push(ta->right);
            stack2.push(tb->right);
        }
    }
    return t1;
    
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值