SRM591 Div1 Easy

275

最長距離をなす2頂点の祖先の深さを決め打ちする

class TheTree {
public:
  int maximumDiameter(vector <int> cnt)
  {
    int mx = 0;
    const int N = cnt.size();
    for (int i = 0; i < N; ++i) {
      int n = (N - i) * 2;
      for (int j = i; j < N; ++j) {
        if (cnt[j] == 1) {
          n = (N - i) + (j - i);
          break;
        }
      }
      mx = max(mx, n);
    }
    return mx;
  }