*[topcoder]TheTree

http://community.topcoder.com/stat?c=problem_statement&pm=12746&rd=15703

这道题有意思。给了树的根和每层节点的个数，求树的直径。做法是如果该层有两个节点，那么可能是有上有下，直径加二；如果该层只有一个节点，那么从这层开始，往下都只能加一。同时，该直径不一定经过根。

#include <vector>
using namespace std;

class TheTree {
public:
    int maximumDiameter(vector <int> cnt);
};

int TheTree::maximumDiameter(vector <int> cnt) {
    int D = cnt.size();
    int ans = 0;
    for (int i = 0; i < D; i++) {
        int res = 0;
        int flag = false;
        for (int j = i; j < D; j++) {
            if (cnt[j] == 1)
                flag = true;
            res += (flag ? 1 : 2);
            ans = max(res, ans);
        }
    }
    return ans;
};