import static sun.swing.MenuItemLayoutHelper.max;
class TreeNode {
int val;
TreeNode left;
TreeNode right;
// 构造函数
TreeNode() {}
TreeNode(int val)
{ this.val = val; }
TreeNode(int val, TreeNode left, TreeNode right) {
this.val = val;
this.left = left;
this.right = right;
}
}
// 自底向上解法
class MaxDepth{
public int maxDepth(TreeNode root) {
// 自底向上的思路就是从空节点开始算起
if (root == null) return 0;
int leftNodeDepth = maxDepth(root.left);
int rightNodeDepth = maxDepth(root.right);
return max(leftNodeDepth,rightNodeDepth)+1;
}
}
// 自顶向下解法
class MaxDepth2{
// 自顶向下的思路是以值根节点的深度为1,每往下一层深度加1,看能加到多少层
int maxDepth = 1;
public void generateDepth(TreeNode root,int depth){
if(root == null)return;
if(root.left == null && root.right == null){
maxDepth = max(maxDepth,depth);
}
generateDepth(root.left,depth+1);
generateDepth(root.right,depth+1);
}
}
class test{
public static void main(String[] args) {
TreeNode node1 = new TreeNode(1);
TreeNode node2 = new TreeNode(2);
TreeNode node3 = new TreeNode(3);
TreeNode node4 = new TreeNode(4);
TreeNode node5 = new TreeNode(5);
TreeNode node6 = new TreeNode(6);
TreeNode node7 = new TreeNode(7);
node1.left = node2;node1.right = node3;
node2.left = node4;node2.right = node5;
node3.left = node6;node3.right = node7;
System.out.println(new MaxDepth().maxDepth(node1));
MaxDepth2 mx = new MaxDepth2();
mx.generateDepth(node1,1);
System.out.println(mx.maxDepth);
}
}
总结
了解递归并利用递归解决问题并不容易。
当遇到树问题时,请先思考一下两个问题:
你能确定一些参数,从该节点自身解决出发寻找答案吗?
你可以使用这些参数和节点本身的值来决定什么应该是传递给它子节点的参数吗?
如果答案都是肯定的,那么请尝试使用 “自顶向下” 的递归来解决此问题。
或者你可以这样思考:对于树中的任意一个节点,如果你知道它子节点的答案,你能计算出该节点的答案吗? 如果答案是肯定的,那么 “自底向上” 的递归可能是一个不错的解决方法。