結果
問題 | No.2337 Equidistant |
ユーザー | umezo |
提出日時 | 2023-06-03 04:10:10 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 553 ms / 4,000 ms |
コード長 | 1,941 bytes |
コンパイル時間 | 2,535 ms |
コンパイル使用メモリ | 204,400 KB |
実行使用メモリ | 45,568 KB |
最終ジャッジ日時 | 2024-12-29 02:31:33 |
合計ジャッジ時間 | 9,597 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 5 ms
8,064 KB |
testcase_01 | AC | 5 ms
8,064 KB |
testcase_02 | AC | 5 ms
8,320 KB |
testcase_03 | AC | 5 ms
8,448 KB |
testcase_04 | AC | 5 ms
8,192 KB |
testcase_05 | AC | 5 ms
8,064 KB |
testcase_06 | AC | 7 ms
8,320 KB |
testcase_07 | AC | 7 ms
8,576 KB |
testcase_08 | AC | 7 ms
8,448 KB |
testcase_09 | AC | 6 ms
8,320 KB |
testcase_10 | AC | 6 ms
8,576 KB |
testcase_11 | AC | 259 ms
30,208 KB |
testcase_12 | AC | 247 ms
30,332 KB |
testcase_13 | AC | 248 ms
30,336 KB |
testcase_14 | AC | 260 ms
30,212 KB |
testcase_15 | AC | 274 ms
30,080 KB |
testcase_16 | AC | 260 ms
30,208 KB |
testcase_17 | AC | 260 ms
30,080 KB |
testcase_18 | AC | 271 ms
30,208 KB |
testcase_19 | AC | 262 ms
30,208 KB |
testcase_20 | AC | 262 ms
30,336 KB |
testcase_21 | AC | 376 ms
45,568 KB |
testcase_22 | AC | 175 ms
30,400 KB |
testcase_23 | AC | 202 ms
30,848 KB |
testcase_24 | AC | 448 ms
40,508 KB |
testcase_25 | AC | 219 ms
30,848 KB |
testcase_26 | AC | 553 ms
40,324 KB |
testcase_27 | AC | 232 ms
30,976 KB |
testcase_28 | AC | 223 ms
30,908 KB |
ソースコード
#define rep(i, n) for (int i = 0; i < (int)(n); i++) #define ALL(v) v.begin(), v.end() typedef long long ll; #include <bits/stdc++.h> using namespace std; const int MAX=200010; const int MAX_LOG=18; vector<int> G[MAX]; int root; int parent[MAX_LOG][MAX]; int len[MAX]; int siz[MAX]; void dfs(int v,int p,int d){ parent[0][v]=p; len[v]=d; int cnt=1; for(auto x:G[v]){ if(x==p) continue; dfs(x,v,d+1); cnt+=siz[x]; } siz[v]=cnt; } void init(int u){ dfs(root,-1,0); for(int k=0;k+1<MAX_LOG;k++){ for(int v=0;v<u;v++){ if(parent[k][v]<0) parent[k+1][v]=-1; else parent[k+1][v]=parent[k][parent[k][v]]; } } } int lca(int u,int v){ if(len[u]>len[v]) swap(u,v); for(int k=0;k<MAX_LOG;k++){ if((len[v]-len[u])&(1<<k)) v=parent[k][v]; } if(u==v) return u; for(int k=MAX_LOG-1;k>=0;k--){ if(parent[k][u]!=parent[k][v]){ u=parent[k][u]; v=parent[k][v]; } } return parent[0][u]; } pair<int,int> lca2(int u,int v){ for(int k=MAX_LOG-1;k>=0;k--){ if(parent[k][u]!=parent[k][v]){ u=parent[k][u]; v=parent[k][v]; } } return {u,v}; } pair<int,int> lca3(int u,int l){ for(int k=MAX_LOG-1;k>=0;k--){ if(len[parent[k][u]]>l){ u=parent[k][u]; } } return {u,parent[0][u]}; } int dist(int a,int b){ return len[a]+len[b]-2*len[lca(a,b)]; } int main(){ ios::sync_with_stdio(false); std::cin.tie(nullptr); int n,q; cin>>n>>q; rep(i,n-1){ int a,b; cin>>a>>b; a--,b--; G[a].push_back(b); G[b].push_back(a); } root=0; init(n); rep(i,q){ int s,t; cin>>s>>t; s--,t--; if(len[s]==len[t]){ auto p=lca2(s,t); cout<<n-siz[p.first]-siz[p.second]<<'\n'; } else if(dist(s,t)%2) cout<<0<<'\n'; else{ if(len[s]<len[t]) swap(s,t); auto p=lca3(s,len[s]-dist(s,t)/2); cout<<siz[p.second]-siz[p.first]<<'\n'; } } return 0; }