제출 #1324593

#제출 시각아이디문제언어결과실행 시간메모리
1324593JungPSRegions (IOI09_regions)C++20
40 / 100
8070 ms39132 KiB
#include<bits/stdc++.h>
using namespace std;

int h[200007];
vector<int> home[25007];
vector<int> vec[200007];
vector<int> tour;
int timer=1;
pair<int,int> inout[200007];

vector<int> touridx[25007];
void dfs(int x){
    tour.push_back(h[x]);
    inout[x].first=timer;
    ++timer;
    for(auto i:vec[x]){
        if(inout[i].first) continue;
        dfs(i);
    }
    tour.push_back(0);
    inout[x].second=timer;
    ++timer;
}
int n,r,q;
map<pair<int,int>,int> ans;
void precompute(){
    for(int rg=1;rg<=r;++rg){
        if(touridx[rg].size()<sqrt(n)) continue;
        // S * n
        for(auto j:touridx[rg]){
            for(int i=inout[j].first+1;i<=inout[j].second-1;++i){
                ++ans[{rg,tour[i]}];
            }
        }
    }
}

int dynamic(int r1,int r2){
    //if(touridx[r1].size()>=sqrt(n)) return ans[{r1,r2}];

    // S log n
    int ans=0;
    for(auto i:home[r1]){
        ans+=upper_bound(touridx[r2].begin(),touridx[r2].end(),inout[i].second)-lower_bound(touridx[r2].begin(),touridx[r2].end(),inout[i].first);
    }
    return ans;
}
signed main(){
    cin >> n >> r >> q;
    cin >> h[1];
    for(int i=2;i<=n;++i){
        int s; cin >> s >> h[i];
        vec[i].push_back(s);
        vec[s].push_back(i); 
    }
    for(int i=1;i<=n;++i){
        home[h[i]].push_back(i);
    }
    tour.push_back(0);
    dfs(1);
    for(int i=0;i<tour.size();++i) touridx[tour[i]].push_back(i);
    precompute();
    while(q--){
        int r1,r2; cin >> r1 >> r2;
        /*
        int ans=0;
        
            */
        // Size log n - per r1 r2
        // Size * n - per r1 xx
        cout << dynamic(r1,r2) << endl;
    }
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...