Submission #912719

#	Time	Username	Problem	Language	Result	Execution time	Memory
912719		anarch_y	Regions (IOI09_regions)	C++17	14 / 100	8096 ms	27896 KiB

regions

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define all(x) begin(x), end(x)
#define sz(x) (int)x.size()
#define pb push_back

const int maxn = 200001;
vector<int> adj[maxn];
int home[maxn], dp[maxn];
int ans;

void dfs(int s, int p, int r1, int r2){
    dp[s] = 0;
    if(home[s] == r2) dp[s] = 1;
    for(auto u: adj[s]){
        if(u == p) continue;
        dfs(u, s, r1, r2);
        dp[s] += dp[u];
    }
    if(home[s] == r1) ans += dp[s];
}

int main(){
    ios::sync_with_stdio(0);
    cin.tie(0);

    int N, R, Q; cin >> N >> R >> Q;
    int h; cin >> h; home[1] = h;
    for(int i=2; i<=N; i++){
        int p, g; cin >> p >> g;
        adj[i].pb(p); adj[p].pb(i);
        home[i] = g; 
    }

    while(Q--){
        int r1, r2; cin >> r1 >> r2;
        ans = 0;
        dfs(1, 0, r1, r2);
        cout << ans << "\n";
        cout.flush();
    }
}

• Subtask 1: 14 / 15
• Subtask 2: 0 / 85