思路
因为树这个东西的定义就是递归的,所以关于树的很多问题我们都可以利用递归的思想来解决~
- 首先需要知道树的BFS和DFS
- BFS也叫做层序遍历,用双队列可以实现~
- DFS根据遍历根节点的顺序不同可分为前序、中序、后序遍历 三者的递归算法十分简单~ 关键的是非递归算法的书写
- 其次是递归思想的运用
- 要写递归函数,第一步先写递归的终止条件
- 然后按照处理问题的逻辑,借助于自身来解决当前问题
- 返回当前结果
- 总而言之,最重要的一句话就是: 对于递归函数 don’t look down 我们在代码中间调用到自身时,一定不要过分纠结到底是怎么执行的,可以假设程序已经正确执行完毕~ 然后继续我们后续的书写即可
对于此题而言,应该用后序遍历~ 因为如果我们要删掉一颗树,必须要他的所有左子树、右子树和自己的值都是null,所以用后序遍历最方便,对于符合上述条件的子树,我们直接返回null
class Solution {
public TreeNode pruneTree(TreeNode root) {
if (null == root)
return null;
root.left = pruneTree(root.left);
root.right = pruneTree(root.right);
if(null == root.left && null == root.right && 0 == root.val)
return null;
else
return root;
}
}