結果
問題 | No.399 動的な領主 |
ユーザー |
|
提出日時 | 2020-10-13 09:10:53 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 183 ms / 2,000 ms |
コード長 | 1,055 bytes |
コンパイル時間 | 663 ms |
コンパイル使用メモリ | 73,268 KB |
実行使用メモリ | 23,552 KB |
最終ジャッジ日時 | 2024-07-20 18:22:29 |
合計ジャッジ時間 | 3,444 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 19 |
コンパイルメッセージ
main.cpp:24:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type] 24 | main() | ^~~~
ソースコード
#include<iostream> #include<vector> using namespace std; int N; vector<int>G[1<<17]; int pr[17][1<<17],dep[1<<17]; void dfs(int u,int p,int d) { dep[u]=d; pr[0][u]=p; for(int v:G[u])if(v!=p)dfs(v,u,d+1); } int cnt[1<<17]; long ans; int dfs2(int u,int p) { for(int v:G[u])if(v!=p) { cnt[u]+=dfs2(v,u); } ans+=(long)cnt[u]*(cnt[u]+1)/2; return cnt[u]; } main() { cin>>N; for(int i=1;i<N;i++) { int u,v;cin>>u>>v; u--,v--; G[u].push_back(v); G[v].push_back(u); } dfs(0,-1,0); for(int k=1;k<17;k++)for(int i=0;i<N;i++) { if(pr[k-1][i]==-1)pr[k][i]=-1; else pr[k][i]=pr[k-1][pr[k-1][i]]; } int Q;cin>>Q; for(;Q--;) { int a,b;cin>>a>>b; a--,b--; int lca; int u=a,v=b; if(dep[u]>dep[v]) { int t=u;u=v;v=t; } for(int k=0;k<17;k++)if(dep[v]-dep[u]>>k&1)v=pr[k][v]; if(u==v)lca=u; else { for(int k=17;k--;)if(pr[k][u]!=pr[k][v]) { u=pr[k][u]; v=pr[k][v]; } lca=pr[0][u]; } cnt[a]++; cnt[b]++; cnt[lca]--; if(pr[0][lca]>=0)cnt[pr[0][lca]]--; } dfs2(0,-1); cout<<ans<<endl; }