#include <bits/stdc++.h>
using namespace std;
//#define int int64_t
#define rep(Begin,End) for (Begin=0;Begin<(End);Begin++)
#define all(x) x.begin(), x.end()
#define all1(x) x.rbegin(), x.rend()
#define sz(x) (int32_t)(x.size())
const int N = 3e5+4, mod =998244353;
const int64_t N1=1e18;
const double eps=1e-6,cmp=1e-3;
int tin[N],tout[N],c[N];
int tmr=1;
vector<int> adj[N];
vector<int> ctin[N],ctout[N],col[25001];
unordered_map<int,int> mp[25001];
int euler[N];
void dfs(int u){
tin[u]=tmr;
euler[tmr]=c[u];
//cerr<<u<< " "<<c[u]<<"\n";
ctin[c[u]].push_back(tmr++);
col[c[u]].push_back(u);
for (const int& it:adj[u]){
dfs(it);
}
tout[u]=tmr++;
ctout[c[u]].push_back(tout[u]);
}
int child(int p,int chld){
int ans=0;
//cerr<<"chld";
for (const int& it:col[chld]){
//cerr<<it<<" ";
int cur=0,cur1=0;
for (int lg=(1ll<<18);lg;lg>>=1){
if (sz(ctin[p])>=cur+lg && ctin[p][cur+lg-1]<tin[it]){
cur+=lg;
}
if (sz(ctout[p])>=cur1+lg && ctout[p][cur1+lg-1]<=tin[it]){
cur1+=lg;
}
}
//cerr<<cur<<" "<<cur1;
ans+=cur-cur1;
}
return ans;
}
int parent(int p,int chld){
int ans=0;
//cerr<<"gh";
//cout<<tin[1]<<" "<<tout[1]<<"\n";
for (const int& it:col[p]){
ans+=upper_bound(all(ctin[chld]),tout[it])-ctin[chld].begin();
ans-=upper_bound(all(ctin[chld]),tin[it])-ctin[chld].begin();
}
return ans;
}
void solve(){
int n,r,q;
cin>>n>>r>>q;
vector<int>f(r+1);
cin>>c[1];
f[c[1]]++;
for (int i=2,x,p;i<=n;i++){
cin>>p>>x;
c[i]=x;
f[x]++;
adj[p].push_back(i);
}
for (int i=0;i<=r;i++){
ctin[i].reserve(f[i]+2);
ctout[i].reserve(f[i]+2);
col[i].reserve(f[i]+2);
}
dfs(1);
for (int i=1;i<=r;i++){
if (sz(col[i])>=512){
vector<int> ok(tmr+1);
for (const int& it:col[i]){
ok[tin[it]]++;
ok[tout[it]]--;
}
for (int j=1;j<tmr;j++){
ok[j]+=ok[j-1];
if (euler[j]){
mp[euler[j]][i]+=ok[j]-!!(euler[j]==i);
}
}
}
}
for (int r,r1;q;q--){
cin>>r>>r1;
if (sz(col[r])>=512){cout<<mp[r1][r]<<endl;continue;}
else if (sz(col[r])<sz(col[r1])){
cout<<parent(r,r1)<<endl;
}
else{
cout<<child(r,r1)<<endl;
}
}
}
int32_t main(){
ios::sync_with_stdio(false),cin.tie(nullptr),cout.tie(nullptr);
int tq = 1;
//cin>>tq;
for (;tq;tq--){solve();}
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
12 ms |
23376 KB |
Output is correct |
2 |
Correct |
12 ms |
23408 KB |
Output is correct |
3 |
Correct |
14 ms |
23384 KB |
Output is correct |
4 |
Correct |
16 ms |
23388 KB |
Output is correct |
5 |
Correct |
19 ms |
23376 KB |
Output is correct |
6 |
Correct |
26 ms |
23504 KB |
Output is correct |
7 |
Correct |
41 ms |
23504 KB |
Output is correct |
8 |
Correct |
50 ms |
23476 KB |
Output is correct |
9 |
Correct |
60 ms |
23992 KB |
Output is correct |
10 |
Correct |
106 ms |
24048 KB |
Output is correct |
11 |
Correct |
127 ms |
24344 KB |
Output is correct |
12 |
Correct |
140 ms |
24912 KB |
Output is correct |
13 |
Correct |
176 ms |
24528 KB |
Output is correct |
14 |
Correct |
254 ms |
25168 KB |
Output is correct |
15 |
Correct |
311 ms |
28256 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1670 ms |
29628 KB |
Output is correct |
2 |
Correct |
1565 ms |
28304 KB |
Output is correct |
3 |
Correct |
2254 ms |
31880 KB |
Output is correct |
4 |
Correct |
251 ms |
25392 KB |
Output is correct |
5 |
Correct |
355 ms |
27592 KB |
Output is correct |
6 |
Correct |
1070 ms |
27216 KB |
Output is correct |
7 |
Correct |
1137 ms |
28344 KB |
Output is correct |
8 |
Correct |
1059 ms |
34376 KB |
Output is correct |
9 |
Correct |
1798 ms |
34632 KB |
Output is correct |
10 |
Runtime error |
3081 ms |
83564 KB |
Execution killed with signal 11 |
11 |
Runtime error |
3072 ms |
70708 KB |
Execution killed with signal 11 |
12 |
Runtime error |
120 ms |
76148 KB |
Execution killed with signal 11 |
13 |
Runtime error |
125 ms |
77188 KB |
Execution killed with signal 11 |
14 |
Runtime error |
126 ms |
76748 KB |
Execution killed with signal 11 |
15 |
Runtime error |
111 ms |
86980 KB |
Execution killed with signal 11 |
16 |
Runtime error |
125 ms |
101988 KB |
Execution killed with signal 11 |
17 |
Runtime error |
115 ms |
98256 KB |
Execution killed with signal 11 |