题解 | #合并二叉树#
合并二叉树
https://www.nowcoder.com/practice/7298353c24cc42e3bd5f0e0bd3d1d759
import java.util.*; /* * public class TreeNode { * int val = 0; * TreeNode left = null; * TreeNode right = null; * public TreeNode(int val) { * this.val = val; * } * } */ // 空间复杂度为O(1),所以就是一个inplace的操作了 public class Solution { /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param t1 TreeNode类 * @param t2 TreeNode类 * @return TreeNode类 */ public TreeNode mergeTrees (TreeNode t1, TreeNode t2) { // write code here t1.val = t1.val + t2.val; if (t1.left != null && t2.left != null) { mergeTrees(t1.left, t2.left); } if (t1.left == null && t2.left != null) { t1.left = t2.left; } if (t1.right != null && t2.right != null) { mergeTrees(t1.right, t2.right); } if (t1.right == null && t2.right != null) { t1.right = t2.right; } return t1; } }
直接用先序遍历的策略,然后把第一棵树当成主树,第一棵树没有的,就把指针指向第二棵树的位置,若都有,那么就把值相加
这个函数可以记下来日后用,因为合并二叉树的操作应该很常见