题解 | #合并二叉树#

合并二叉树

https://www.nowcoder.com/practice/7298353c24cc42e3bd5f0e0bd3d1d759

//!recurience &c++
//step 1:首先判断t1与t2是否为空,若为则用另一个代替,若都为空,返回的值也是空。
//step 2:使用三个辅助队列,第一个队列q用于暂存合并后的二叉树的层次遍历节点,第二个队列q1用于暂存t1的层次遍历节点,第三个队列q2用于暂存t2的层次遍历节点。
//step 3:两棵树同步层次遍历,先将根节点加入队列中,同时根节点优先合并。
//step 4:每次从队列分别弹出一个元素,判断分别二者的左右子节点是否存在,若是都存在,则相加合并,若是只存在一个则连接该存在的节点,若是都不存在则连接null。

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 * };
 */

#include <queue>
class Solution {
public:
    /**
     * 
     * @param t1 TreeNode类 
     * @param t2 TreeNode类 
     * @return TreeNode类
     */
    TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
        // write code here
        //非递归
        if (t1==NULL) {
            return t2;
        }
        if (t2==NULL) {
            return t1;
        }
        TreeNode* head=new TreeNode(t1->val+t2->val);
        queue<TreeNode*> q;
        queue<TreeNode*> q1;
        queue<TreeNode*> q2;
        q.push(head);
        q1.push(t1);
        q2.push(t2);
        while(!q1.empty()||!q2.empty()){
            TreeNode *node=q.front(),*node1=q1.front(),*node2=q2.front();
            q.pop();
            q1.pop();
            q2.pop();
            TreeNode *left1=node1->left,*left2=node2->left,*right1=node1->right,*right2=node2->right;
            if (left1&&left2) {
                TreeNode *left=new TreeNode(left1->val+left2->val);
                node->left=left;
                q.push(left);
                q1.push(left1);
                q2.push(left2);
            
            }else if (left1) {
                node->left=left1;
                
            }else if (left2) {
                node->left=left2;
            }

            if (right1&&right2) {
                TreeNode *right=new TreeNode(right1->val+right2->val);
                node->right=right;
                q.push(right);
                q1.push(right1);
                q2.push(right2);
            
            }else if (right1) {
                node->right=right1;
                
            }else if (right2) {
                node->right=right2;
            }

        }
        return head;

        

    }
};

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务