题解 | #合并二叉树#
合并二叉树
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;
}
};
