二叉树：二叉树的直径

二叉树直径的定义：二叉树中路径的最大长度
二叉树中路径的最大长度，可以理解所有节点的左右子树高度之和的最大值。假设二叉树有n个节点，编号为{a1,a2,…,an}, 其对应的左右子树的高度之和为H = {h1,h2,h3,…,hn}, 则该二叉树的直径为max(H)。

struct node {
	int val;
	node *left, *right;
};

int ans  = 0;// 用于维护任意节点左右子树高度和的最大值
//dfs()函数的返回值是当前节点的高度

int dfs(node *root) {
	if (!root) return 0;
	int left = dfs(root->left);
	int right = dfs(root->right);
	ans = max(ans, left+right);
	return max(right, left) + 1;
}


int diameterOfBinaryTree(node *root) {
	dfs(root);
	return ans;
}