이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include<bits/stdc++.h>
using namespace std;
#define all(fl) fl.begin(),fl.end()
#define pb push_back
#define fi first
#define se second
#define for1(i,j,k) for(int i=j;i<=k;i++)
#define for2(i,j,k) for(int i=j;i>=k;i--)
#define for3(i,j,k,l) for(int i=j;i<=k;i+=l)
#define lb lower_bound
#define ub upper_bound
#define sz(a) (int)a.size()
#define pii pair<int,int>
#define pli pair<long long,int>
#define gcd __gcd
#define lcm(x,y) x*y/__gcd(x,y)
#define pil pair<int,long long>
const int maxn=2e5+9;
const int block=300;
int x[maxn/block][25009],y[25009][maxn/block],h[maxn];
vector<int>value[25009];
vector<int>a[maxn];
vector<int>euler[25009];
vector<int>only[2509];
int large[maxn],rep[maxn];
int in[maxn],out[maxn],tme=0;
void dfs(int u){
in[u]=++tme;
for (auto v:a[u]){
dfs(v);
}
out[u]=tme;
}
int sum[maxn];
int nw[maxn];
int cnt;
void redfs(int u){
if (rep[h[u]]){
nw[rep[h[u]]]++;
}
for1(j,1,cnt){
x[j][h[u]]+=nw[j];
}
for (auto v:a[u]){
redfs(v);
}
if (rep[h[u]]){
nw[rep[h[u]]]--;
}
}
int dd[maxn];
void cal(int x1,int y1){
if (rep[y1]){
cout<<y[x1][rep[y1]]<<endl;
return;
}
if (rep[x1]){
cout<<x[rep[x1]][y1]<<endl;
return;
}
int ans=0;
int now=0;
int p=-1;
for(auto v:euler[x1]){
for1(j,p+1,sz(only[y1])-1){
if (only[y1][j]<=v)p=j,now++;
else break;
}
dd[v]=now;
}
for (auto v:value[x1]){
ans+=dd[out[v]]-dd[in[v]-1];
}
for (auto v:euler[x1]){
dd[v]=0;
}
cout<<ans<<endl;
}
signed main(){
ios_base::sync_with_stdio(0);
cin.tie(0);
//freopen("temp.INP","r",stdin);
//freopen("temp.OUT","w",stdout);
int n,r,q;
cin>>n>>r>>q;
for1(i,1,n){
cin>>h[i];
if (i==1){
value[h[i]].pb(i);
continue;
}
int p;
cin>>p;
swap(p,h[i]);
value[h[i]].pb(i);
a[p].pb(i);
}
dfs(1);
cnt=0;
for1(i,1,r){
if (sz(value[i])>=block){
cnt++;
large[cnt]=i;
rep[i]=cnt;
}
}
for1(i,1,cnt){
for1(j,1,n)sum[j]=0;
for(auto v:value[large[i]]){
sum[in[v]]++;
}
for1(j,1,n)sum[j]+=sum[j-1];
for1(j,1,r){
if (j==large[i])continue;
for (auto v:value[j]){
y[j][i]+=sum[out[v]]-sum[in[v]-1];
}
}
}
redfs(1);
for1(i,1,n){
if (in[i]>1)euler[h[i]].pb(in[i]-1);
euler[h[i]].pb(out[i]);
only[h[i]].pb(in[i]);
}
for1(i,1,r){
sort(all(euler[i]));
euler[i].resize(distance(euler[i].begin(),unique(all(euler[i]))));
sort(all(only[i]));
only[i].resize(distance(only[i].begin(),unique(all(only[i]))));
}
for1(i,1,q){
int x1,y1;
cin>>x1>>y1;
cal(x1,y1);
}
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |