LeetCode刷题笔记：543.二叉树的直径

1. 问题描述

给定一棵二叉树，你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。

2. 解题思路

⭐本题关键在于所求路径可能不穿过根节点
比方说下面这棵二叉树，最大路径不经过根节点 1：

			1
		  /
		2
	  /  \
	 3    4
    /		\
   5         6

① 遍历每一个节点
② 以每一个节点为中心点计算其 左子树边长 与 右子树边长 之和
③ 每次遍历判断是否更新全局变量 max_d

3. 代码实现

/**
 * 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 {
	int max_d = 0;
	public int diameterOfBinaryTree(TreeNode root) {
		if (root == null) {
			return 0;
		}
		dfs(root);
		return max_d;
	}
	
	private int dfs(TreeNode root) {
		if (root.left == null && root.right == null) {
			return 0;
		}
		// 若当前节点左子树为空，则记它的左子树边长为0
		int leftLength = root.left == null ? 0 : dfs(root.left) + 1;
		// 若当前节点右子树为空，则记它的右子树边长为1
		int rightLength = root.right == null ? 0 : dfs(root.right) + 1;
		// 判断是否需要更新最大直径
		max_d = Math.max(max_d, leftLength + rightLength);
		return Math.max(leftLength, rightLength);
	}
}