結果
問題 | No.399 動的な領主 |
ユーザー | kotatsugame |
提出日時 | 2020-10-13 09:10:53 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 213 ms / 2,000 ms |
コード長 | 1,055 bytes |
コンパイル時間 | 672 ms |
コンパイル使用メモリ | 72,708 KB |
実行使用メモリ | 23,456 KB |
最終ジャッジ日時 | 2023-09-27 23:59:20 |
合計ジャッジ時間 | 4,101 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 4 ms
14,084 KB |
testcase_01 | AC | 5 ms
14,120 KB |
testcase_02 | AC | 5 ms
14,172 KB |
testcase_03 | AC | 4 ms
14,104 KB |
testcase_04 | AC | 5 ms
14,096 KB |
testcase_05 | AC | 18 ms
14,556 KB |
testcase_06 | AC | 209 ms
18,892 KB |
testcase_07 | AC | 212 ms
18,796 KB |
testcase_08 | AC | 203 ms
18,856 KB |
testcase_09 | AC | 206 ms
18,976 KB |
testcase_10 | AC | 6 ms
14,232 KB |
testcase_11 | AC | 17 ms
14,616 KB |
testcase_12 | AC | 168 ms
19,296 KB |
testcase_13 | AC | 156 ms
19,276 KB |
testcase_14 | AC | 122 ms
23,428 KB |
testcase_15 | AC | 132 ms
23,456 KB |
testcase_16 | AC | 150 ms
20,932 KB |
testcase_17 | AC | 213 ms
18,972 KB |
testcase_18 | AC | 208 ms
19,092 KB |
コンパイルメッセージ
main.cpp:24:1: 警告: ISO C++ では型の無い ‘main’ の宣言を禁止しています [-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; }