結果
問題 | No.399 動的な領主 |
ユーザー | autumn-eel |
提出日時 | 2018-01-11 21:23:31 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 152 ms / 2,000 ms |
コード長 | 1,189 bytes |
コンパイル時間 | 1,628 ms |
コンパイル使用メモリ | 169,400 KB |
実行使用メモリ | 32,808 KB |
最終ジャッジ日時 | 2023-08-25 04:05:21 |
合計ジャッジ時間 | 4,495 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge15 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 6 ms
22,320 KB |
testcase_01 | AC | 6 ms
22,324 KB |
testcase_02 | AC | 6 ms
22,304 KB |
testcase_03 | AC | 6 ms
22,312 KB |
testcase_04 | AC | 7 ms
22,448 KB |
testcase_05 | AC | 15 ms
22,752 KB |
testcase_06 | AC | 146 ms
26,652 KB |
testcase_07 | AC | 152 ms
26,660 KB |
testcase_08 | AC | 140 ms
26,704 KB |
testcase_09 | AC | 141 ms
26,576 KB |
testcase_10 | AC | 7 ms
22,400 KB |
testcase_11 | AC | 13 ms
22,744 KB |
testcase_12 | AC | 98 ms
27,104 KB |
testcase_13 | AC | 93 ms
26,960 KB |
testcase_14 | AC | 63 ms
32,740 KB |
testcase_15 | AC | 74 ms
32,808 KB |
testcase_16 | AC | 81 ms
29,284 KB |
testcase_17 | AC | 135 ms
26,684 KB |
testcase_18 | AC | 149 ms
26,544 KB |
ソースコード
#include <bits/stdc++.h> #define rep(i,n)for(int i=0;i<(n);i++) using namespace std; typedef long long ll; #define MAX_LOG 18 int d[200000],par[MAX_LOG][200000]; vector<int>E[200000]; ll sum[200000]; int n; void dfs(int v,int p,int depth){ d[v]=depth;par[0][v]=p; for(int u:E[v]){ if(u==p)continue; dfs(u,v,depth+1); } } void lca_init(){ for(int i=1;i<MAX_LOG;i++)rep(j,n){ if(par[i-1][j]==-1)par[i][j]=-1; else par[i][j]=par[i-1][par[i-1][j]]; } } int lca(int u,int v){ if(d[u]<d[v])swap(u,v); rep(i,MAX_LOG){ if((d[u]-d[v])>>i&1)u=par[i][u]; } if(u==v)return u; for(int i=MAX_LOG-1;i>=0;i--){ if(par[i][u]!=par[i][v]){ u=par[i][u]; v=par[i][v]; } } return par[0][u]; } ll ans=0; void dfs2(int v,int p){ for(int u:E[v]){ if(u==p)continue; dfs2(u,v); sum[v]+=sum[u]; } ans+=(sum[v]+1)*sum[v]/2; } int main(){ scanf("%d",&n); rep(i,n-1){ int u,v;scanf("%d%d",&u,&v);u--;v--; E[u].push_back(v); E[v].push_back(u); } dfs(0,-1,0); lca_init(); int q;scanf("%d",&q); rep(i,q){ int a,b;scanf("%d%d",&a,&b);a--;b--; sum[a]++;sum[b]++; int p=lca(a,b); sum[p]--; if(par[0][p]!=-1)sum[par[0][p]]--; } dfs2(0,-1); cout<<ans<<endl; }