Facebook 2/28电面面经

uestcxzk 发表于 2017-3-3 14:02
我觉得要在root上方再加个dummy，否则root如果满足条件的话删不掉

        public static void remove(TreeNode root) {
                if(root == null) return;
                if(removeHelper(root) == 0) {
                        root = null;
                }
        }
       
        public static int removeHelper(TreeNode root) {
                if(root == null) return 0;
               
                int sum = root.val;
                Iterator<TreeNode> iter = root.children.iterator();
                while(iter.hasNext()) {
                        TreeNode child = iter.next();
                        int childSum = removeHelper(child);
                        if(childSum == 0) {
                                iter.remove();
                        }
                        sum += childSum;
                }
                return sum;
        }



补充内容 (2017-3-3 14:25):
Good point. 这样可以吗？

uestcxzk 发表于 2017-3-3 14:02
我觉得要在root上方再加个dummy，否则root如果满足条件的话删不掉

可以直接返回null

mingzhou1987 发表于 2017-3-3 13:41
写了个二叉树的， 这样可以否？
TreeNode* delete(TreeNode* root)
{

条件不是二叉树。有n个子节点

mingzhou1987 发表于 2017-3-3 13:41
写了个二叉树的， 这样可以否？
TreeNode* delete(TreeNode* root)
{

这个需要后续遍历吧~~

houqingniao 发表于 2017-3-3 15:07
这个需要后续遍历吧~~

这样？
TreeNode* delete(TreeNode* root)
{
  if(!root)return NULL;
  root->left = delete(root->left);
  root->right = delete(root->right);
  int leftSum = sumTree(root->left);
  int rightSum = sumTree(root->right);
  if(leftSum + rightSum + root->val == 0)return NULL;
  return root;
}