結果

問題 No.806 木を道に
ユーザー 👑 CleyLCleyL
提出日時 2022-05-02 18:33:21
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 135 ms / 2,000 ms
コード長 1,642 bytes
コンパイル時間 848 ms
コンパイル使用メモリ 82,880 KB
実行使用メモリ 17,024 KB
最終ジャッジ日時 2024-07-01 22:34:45
合計ジャッジ時間 3,297 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 24 ms
6,016 KB
testcase_11 AC 22 ms
5,632 KB
testcase_12 AC 113 ms
13,312 KB
testcase_13 AC 79 ms
10,624 KB
testcase_14 AC 108 ms
12,800 KB
testcase_15 AC 118 ms
13,440 KB
testcase_16 AC 37 ms
6,944 KB
testcase_17 AC 99 ms
12,032 KB
testcase_18 AC 9 ms
5,376 KB
testcase_19 AC 27 ms
6,016 KB
testcase_20 AC 102 ms
12,032 KB
testcase_21 AC 50 ms
8,192 KB
testcase_22 AC 135 ms
15,232 KB
testcase_23 AC 132 ms
15,104 KB
testcase_24 AC 50 ms
12,288 KB
testcase_25 AC 74 ms
17,024 KB
testcase_26 AC 24 ms
6,912 KB
testcase_27 AC 78 ms
15,540 KB
testcase_28 AC 4 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
using namespace std;
template<typename T>
struct TreeDiameter{
  struct edge{
    int to;
    T cost;
  };
  int n;
  T generated;
  vector<vector<edge>> tree; 
  vector<int> path;
  vector<int> C;
  
  TreeDiameter(int n_) : n(n_),tree(n_) {
    generated = -1;
  }

  void add(int s,int v){
    tree[s].push_back({v,1});
    tree[v].push_back({s,1});
    generated = -1;
  }

  void add(int s,int v,T c){
    tree[s].push_back({v,c});
    tree[v].push_back({s,c});
    generated = -1;
  }

  T build(){
    if(generated != -1)return generated;
    auto x = DFS(0);
    auto y = DFS(x.first);
    int nw = x.first;
    while(y.first != nw){
      path.push_back(nw);
      nw = C[nw];
    }
    path.push_back(nw);
    return generated = y.second;
  }

private:
  //DFS(int, T, (int)) -> pair<int,int>
  pair<int,T> DFS(int nw,T dist=0,int initnal=1){
    if(initnal){
      C.assign(n,-1);
      path.clear();
      initnal = 0;
    }
    C[nw] = 0;
    pair<int,T> ret = make_pair(nw,dist);
    for(int i = 0; tree[nw].size() > i; i++){
      if(C[tree[nw][i].to] == -1){
        pair<int,T> x = DFS(tree[nw][i].to,dist+tree[nw][i].cost,0);
        if(ret.second < x.second){
          ret = x;
          C[nw] = tree[nw][i].to;
        }
      }
    }
    return ret;
  }
};


int main(){
  int n;cin>>n;
  TreeDiameter<int> A(n);
  vector<int> B[n];
  for(int i = 0; n-1 > i; i++){
    int s,v;cin>>s>>v;
    A.add(--s,--v);
    B[s].push_back(v);
    B[v].push_back(s);
  }
  A.build();
  int ans = 0;
  for(int i = 0; n > i; i++){
    ans += max(0,(int)B[i].size()-2);
  }
  cout << ans << endl;
}
0