| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 647347 | Karuk | Regions (IOI09_regions) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include<bits/stdc++.h>
using namespace std;
const int bucket=450;
int parent[200001];
vector<int>children[200001];
int reg[200001];///regions
vector<int>index[25001];
vector<int>regions[25001];
int ticker=0;
int beg[200001],en[200001];
void dfs(int n) {
index[reg[n]].push_back(ticker);
beg[n]=ticker++;
for(int i:children[n]) {
dfs(i);
}
en[n]=ticker-1;
}
vector<int>large;
int num[25001];
map<pair<int,int>,int>ans;
void dfs1(int n) {
for(int i:large) {
ans[{i,reg[n]}]+=num[i];
}
num[reg[n]]++;
for(int i:children[n]) {
dfs1(i);
}
num[reg[n]]--;
}
int main() {
int n,r,q;cin>>n>>r>>q;
cin>>reg[1];
regions[reg[1]].push_back(1);
for(int i=2;i<=n;i++) {
cin>>parent[i]>>reg[i];
children[parent[i]].push_back(i);
regions[reg[i]].push_back(i);
}
dfs(1);
for(int i=1;i<=r;i++) {
if(index[i].size()>=bucket)large.push_back(i);
}
dfs1(1);
for(int i=0;i<q;i++) {
int x,y;cin>>x>>y;
if(index[x].size()>=bucket)cout<<ans[{x,y}]<<endl;
else {
int anss=0;
for(int ind:regions[x]) {
int pos1=upper_bound(index[y].begin(),index[y].end(),beg[ind])-index[y].begin();
int pos2=upper_bound(index[y].begin(),index[y].end(),en[ind])-index[y].begin();
anss+=pos2-pos1;
}
cout<<anss<<endl;
}
}
return 0;
}