题目

类型：二叉树、DFS

image.png

解题思路

二叉树的直径，就是左右子树的最大深度之和，那么直接的想法是对每个节点计算左右子树的最大高度，得出每个节点的直径，从而得出最大的那个直径。
但是由于 maxDepth 也是递归函数，所以上述方式时间复杂度较高。
需要灵活运用二叉树的后序遍历，在 maxDepth 的后序遍历位置顺便计算最大直径。

代码

  1. /**
  2.  * Definition for a binary tree node.
  3.  * public class TreeNode {
  4.  *     int val;
  5.  *     TreeNode left;
  6.  *     TreeNode right;
  7.  *     TreeNode() {}
  8.  *     TreeNode(int val) { this.val = val; }
  9.  *     TreeNode(int val, TreeNode left, TreeNode right) {
  10.  *         this.val = val;
  11.  *         this.left = left;
  12.  *         this.right = right;
  13.  *     }
  14.  * }
  15.  */
  16. class Solution {
  17.     int maxDiameter = 0;
  19.     public int diameterOfBinaryTree(TreeNode root) {
  20.         maxDepth(root);
  21.         return maxDiameter;
  22.     }
  24.     int maxDepth(TreeNode root) {
  25.         if (root == null) {
  26.             return 0;
  27.         }
  28.         int leftMax = maxDepth(root.left);
  29.         int rightMax = maxDepth(root.right);
  30.         // 后序遍历位置顺便计算最大直径
  31.         maxDiameter = Math.max(maxDiameter, leftMax + rightMax);
  32.         return 1 + Math.max(leftMax, rightMax);
  33.     }
  34. }