二叉树和递归的关系在算法题中比较紧密,使用递归的方法是为了在算法的运行过程中,一次的遍历情况下利用到更多的信息。
前序位置的代码可以获取进入每个节点之前的信息
后序位置的代码可以利用离开节点的信息或者子树的信息
以求一棵树的最大直径为例:
给你一棵二叉树的根节点,返回该树的 直径 。
二叉树的 直径 是指树中任意两个节点之间最长路径的 长度 。这条路径可能经过也可能不经过根节点 root 。
两节点之间路径的 长度 由它们之间边数表示。
通过对这个最大直径的概念经过分析其实很容易就能知道这个直径,其实可以通过一个非叶子结点的最大左子树深度加右子树深度得到(root.left的最大深度和root.right的最大深度之和),所以使用递归方法时,仍然是求每个节点的子树深度,只是需要通过这个来更新最终需要的"直径",这里需要注意的是,代码中的递归部分既可以通过返回值更新节点的最大深度,也可以通过更新私有变量res来确保最终得到的直径是最大的。
class Solution {
private int res = 0;
public int diameterOfBinaryTree(TreeNode root) {
maxdiameter(root);
return res;
}
public int maxdiameter (TreeNode root) {
if (root == null) {
return 0;
}
int leftmax = maxdiameter(root.left);
int rightmax = maxdiameter(root.right);
int thediameter = leftmax + rightmax;
res = Math.max(thediameter,res);
return Math.max(leftmax, rightmax) + 1;
}
}