LeetCode 543. 二叉树的直径

题目传送门
简单的递归题目。

class Solution {
public:
	int ans = 0;
	int depth(TreeNode* rt)
	{
		if (rt == NULL)
		{
			return 0;
		}
		int l = depth(rt->left);
		int r = depth(rt->right);
		ans = max(ans, l + r );
		return max(l, r) + 1;
	}
	int diameterOfBinaryTree(TreeNode* root) {
		depth(root);
		return ans;
	}
};