leetcode 刷题（求树的直径）

树的直径的定义：

The diameter of a binary tree is the length of thelongest path between any two nodes in a tree. This path may or may not pass through the root.

我首先想到的思路是： 直径必然是两个叶子节点之间的距离的最大值， 怎么求这个最大值呢？

一开始想的是采用穷举， 计算出任何两个叶子节点之间的距离， 求出其中的最大值， 但是这种方法似乎实现起来特别困难； 

不得不转换思路， 转而想， 对于一个节点， 经过这个节点的直径，应该是该节点的左子树的最大深度加上右子树的最大深度。

所以可以依次计算出经过根节点的直径， 经过根节点左孩子的最大直径......., 然后计算出最大值即可。这种实现起来应该比较简单， 毕竟求树的最大深度比较容易。

class Solution {
public:
    int diameterOfBinaryTree(TreeNode* root) {
        list<TreeNode* > nodes;
        if (root != NULL) {
            nodes.push_back(root);
        }
        
        int diameter = 0;
        while(!nodes.empty()) {
            TreeNode* node = nodes.front();
            int nd = diameterOfNode(node);
            
            if (nd > diameter) {
                diameter = nd;
            }
            
            nodes.pop_front();
            if (node->left != NULL) {
                nodes.push_back(node->left);
            }
            if (node->right != NULL) {
                nodes.push_back(node->right);
            }
        }
        
        return diameter;
    }
private:
    int depth(TreeNode* root) {			// 计算树的深度
        if (!root) {
            return 0;
        }
        
        if (!root->left && !root->right) {
            return 1;
        }
        
        int ldp = depth(root->left);
        int rdp = depth(root->right);
        
        return (ldp < rdp ? rdp : ldp) + 1;
    }
private:
    int diameterOfNode(TreeNode* root) {	// 计算某个经过某个节点的直径
        if (!root) {
            return 0;
        }
        
        if (!root->left && !root->right) {
            return 0;
        }
        
        return depth(root->left) + depth(root->right);
    }
};