PigyChan_LeetCode 814. 二叉树剪枝

814. 二叉树剪枝

难度中等

给定二叉树根结点 root ，此外树的每个结点的值要么是 0，要么是 1。
返回移除了所有不包含 1 的子树的原二叉树。
( 节点 X 的子树为 X 本身，以及所有 X 的后代。)

示例1:输入: [1,null,0,0,1]
输出: [1,null,0,null,1]
解释:
只有红色节点满足条件“所有不包含 1 的子树”。
右图为返回的答案。

示例2:输入: [1,0,1,0,0,0,1]
输出: [1,null,1,null,1]

示例3:输入: [1,1,0,1,1,0,1,0]
输出: [1,1,0,1,1,null,1]

说明:
(1)给定的二叉树最多有 100 个节点。
(2)每个节点的值只会为 0 或 1 。

思路1.0：

观察了几个示例，发现若要移除某个节点，需要得到该节点左右子树的信息（即是否全为0），所以采用后序遍历的方法。

代码1.0（已完成）：

  class Solution {
  public:
      TreeNode* posOrder(TreeNode* TNode)
      {
          if (TNode != NULL)
          {
              TNode->left = posOrder(TNode->left);
              TNode->right = posOrder(TNode->right);
              if (TNode->left == NULL && TNode->right == NULL && TNode->val==0)
                  return NULL;
          }


          return TNode;
      }


      TreeNode* pruneTree(TreeNode* root) {
          root = posOrder(root);
          return root;
      }
  };