facebook 面经

FB 面经 Delete subtree sum zero

2018-09-14  本文已影响0人  Anseis
static TreeNode delete(TreeNode root) { 
    if (root == null) {
      return null;
    }
    int sum = deleteSum(root);
    if (sum == 0) {
      return null;
    }
    return root;
  } 
  int deleteSum(TreeNode root) {
    if (root == null) {
      return 0;
    }
    int left = deleteSum(root.left);
    int right = deleteSum(root.right);
    if (left == 0) {
      root.left = null;
    }
    if (right == 0) {
      root.right = 0;
    }
    return root.val+left+right;
  }
上一篇 下一篇

猜你喜欢

热点阅读