合并二叉树。
方法1递归:以前序遍历的方式处理
两棵树的节点。如果树1为空,则合并的结果是树2,反之亦然。
剩下的情况就是参数中的两个节点都不是NULL。
将节点2的值合并到节点1中。
处理左节点:递归两棵树的左节点。
处理右节点:递归两棵树的右节点。
C++:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
if(t1==NULL) return t2;
if(t2==NULL) return t1;
t1->val+=t2->val;
t1->left=mergeTrees(t1->left,t2->left);
t1->right=mergeTrees(t1->right,t2->right);
return t1;
}
};
方法2栈:
以前序处理节点。
栈元素是
一个节点数值对,节点1和节点2的数值。
一开始
把两棵树的根节点压栈。
只要栈不空,就一直执行下列操作:
若节点2为空,忽略这个数值对。
修改节点1的值。
如果节点1的左节点为空,把节点2的左节点搬过来。若不为空,把节点1和节点2的左节点都压栈。
右边的处理相同。
栈空之后,返回节点1。
java:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public TreeNode mergeTrees(TreeNode t1, TreeNode t2) {
if (t1 == null)
return t2;
Stack < TreeNode[] > stack = new Stack < > ();
stack.push(new TreeNode[] {t1, t2});
while (!stack.isEmpty()) {
TreeNode[] t = stack.pop();
if (t[0] == null || t[1] == null) {
continue;
}
t[0].val += t[1].val;
if (t[0].left == null) {
t[0].left = t[1].left;
} else {
stack.push(new TreeNode[] {t[0].left, t[1].left});
}
if (t[0].right == null) {
t[0].right = t[1].right;
} else {
stack.push(new TreeNode[] {t[0].right, t[1].right});
}
}
return t1;
}
}
后一种方法的时间复杂度和两棵树节点数的最小值有关,效率更高。