一、思路:
对二叉树进行深度遍历,涉及到递归的思想。在写递归时,不要深入模拟,不要深入模拟,不要深入模拟!
对于当前这个节点我们有三条路可以走:
1、当前节点和左子节点形成最长路径;
2、当前节点和右子节点形成最长路径;
3、当前节点作为中间站,连接左子节点和右子节点形成最长路径。
注意:对于这三种情况,1、2中情况可以作为递归函数的返回值,第三种情况则需要额外进行记录和比较。
二、代码以及注释
/**
* Definition for a binary tree node.
* public 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 Solution {
private int ans = 0;
public int diameterOfBinaryTree(TreeNode root) {
maxPath(root);
return ans;
}
public int maxPath(TreeNode root){
/*
* 递归三要素:
* 1、end 递归结束条件
* 2、do 本次递归要进行的操作
* 3、return 递归的返回值
*/
//end
if(root == null){
//这时候来到了空节点,实际不存在,所以返回-1,用以抵消后边加一边的操作
return -1;
}
//do
//左边节点最长路径
int left = maxPath(root.left);
//右边节点最长路径
int right = maxPath(root.right);
//该节点从左或从右最大,再加上子节点到父节点的一条边
int ret = Math.max(left, right) + 1;
//比较一下从左往右走的情况,这时候要加上两条边
ans = Math.max(ans, Math.max(ret, left + right + 2));
//return
return ret;
}
}
三、注意事项
1、本文将于叶节点连接的虚节点看作可以访问的,而实际上是不可达到的,这时候需要返回-1进行抵消。
2、建议画图,只关心当前节点、左子节点和右子节点的情况,避免陷入递归。
3、当递归条件不清晰的时候,可以画最小层的二叉树进行模拟,推出递归结束条件和递归需要进行的操作。