剑指 Offer II 047. 二叉树剪枝

92 阅读1分钟

剑指 Offer II 047. 二叉树剪枝

递归思想:

将整个树看成是两个子树组成,如果两个子树经过修建后,都不存在了,且自身的值为0了,则说明自己也该删除了,这时候,返回null,否则返回本身

var pruneTree = function (root) {
  if (root == null) {
    return root;
  }
  root.left = pruneTree(root.left);
  root.right = pruneTree(root.right);
  if (!root.left && !root.right && root.val == 0) {
    return null;
  }
  return root;
};