結果

問題 No.399 動的な領主
コンテスト
ユーザー ikd
提出日時 2018-02-12 16:04:56
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 281 ms / 2,000 ms
コード長 1,712 bytes
コンパイル時間 619 ms
コンパイル使用メモリ 99,180 KB
実行使用メモリ 51,996 KB
最終ジャッジ日時 2024-06-12 23:52:56
合計ジャッジ時間 5,275 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

class Tree{
  import std.algorithm;
  int[][] g;
  int[][] par;
  int[] dep;
  int n;
  const int len=17; // n <= 1e5
  long[] sub;
  this(int n){
    this.n=n;
    g.length=n;
    par=new int[][](n, len);
    foreach(i; 0..n) fill(par[i], -1);
    dep.length=n;
    sub.length=n;
  }
  void addEdge(int u, int v){
    g[u]~=v; g[v]~=u;
  }
  void dfs(int i=0, int j=-1){
    par[i][0]=j;
    if(j>=0) dep[i]=dep[j]+1;
    foreach(k; g[i]){
      if(j!=k) dfs(k, i); 
    }
  }
  void init(){
    dfs();
    for(int j=0; j<len-1; j++)foreach(i; 0..n){
      if(par[i][j]>=0) par[i][j+1]=par[par[i][j]][j];
    }
  }
  int getLca(int s, int t){
    if(dep[s]<dep[t]) swap(s, t);
    foreach(i; 0..len){
      if((dep[s]-dep[t])&(1<<i)) s=par[s][i];
    }
    if(s==t) return s;
    for(auto i=len-1; i>=0; i--){
      if(par[s][i]!=par[t][i]) s=par[s][i], t=par[t][i];
    }
    return par[s][0];
  }
  void set(int s, int t){
    sub[s]++; sub[t]++;
    auto lca=getLca(s, t);
    sub[lca]--;
    if(lca>0) sub[par[lca][0]]--;
  }
  void add(int i=0, int p=-1){
    foreach(j; g[i])if(j!=p){
      add(j, i);
      sub[i]+=sub[j];
    }
  }
  long acc(){
    return reduce!((r,e)=>(r+e*(e+1)/2))(0L, sub);
  }
}

void main(){
  import std.stdio, std.string, std.conv, std.algorithm;

  int n; rd(n);
  auto tree=new Tree(n);
  foreach(_; 0..(n-1)){
    int u, v; rd(u, v);
    tree.addEdge(u-1, v-1);
  }
  tree.init;
  int q; rd(q);
  while(q--){
    int u, v; rd(u, v);
    tree.set(u-1, v-1);
  }

  tree.add;
  writeln(tree.acc);
}

void rd(T...)(ref T x){
  import std.stdio, std.string, std.conv;
  auto l=readln.split;
  assert(l.length==x.length);
  foreach(i, ref e; x){
    e=l[i].to!(typeof(e));
  }
}
0