又到了金九银十找实习,找工作的时机,兄弟们都拿到自己心仪的offer了吗?下面我来分享一下面试过程中,出现频率非常高的面试题之一的-------剑指 Offer II 047. 二叉树剪枝
一:题目背景:
给定一个二叉树 根节点 root ,树的每个节点的值要么是 0,要么是 1。请剪除该二叉树中所有节点的值为 0 的子树。节点 node 的子树为 node 本身,以及所有 node 的后代。
二:题目深度解析:
经过题目的说明以及通过给的示例来分析,该特殊的二叉树的节点的值为0或者1组成,要求我们删除节点为0的节点,有两种情况,第一种就是实例一:节点为0的节点都在叶子节点,那么直接进行删除即可,还有一种就是实例二的那种情况,父节点和孩子节点都为0,都需要删除。
三:解答:
通过递归来判断,首先如果遍历的当前节点为空,那么直接返回空,然后第一次判断,来解决叶子节点需要删除的,判断该节点的左孩子为空,右孩子为空,且该节点为0,那么直接返回null,为什么这里直接返回空呢,因为我们在下面直接接受,把这个空直接赋值给它的左或者右孩子,是不是就代表把这个节点删除了。然后一次遍历左子树,右子树,最终,还要进行一次判断,因为只处理了它的左右孩子,但是父节点还没有处理,所以再次判断,如果该父节点为0,并且左右孩子为空,那么直接返回null,否则直接返回该节点就行。
四:代码展示:
class Solution {
public TreeNode pruneTree(TreeNode root) {
return deletetree(root);
}
public TreeNode deletetree(TreeNode root) {
if(root == null || (root.val == 0 && root.left == null && root.right == null)) {
return null;
}
root.left = deletetree(root.left);
root.right = deletetree(root.right);
if(root.val == 0 && root.left == null && root.right == null) return null;
return root;
}
}