Submission #1292052

#	Time	Username	Problem	Language	Result	Execution time	Memory
1292052		eliminator_101	Regions (IOI09_regions)	C++20	20 / 100	4520 ms	196608 KiB

regions

#include <bits/stdc++.h>
using namespace std;

#define int long long int

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int n, r, q;
    cin >> n >> r >> q;
    vector<vector<int>> cnt(n+1, vector<int>(r+1,0));
    vector<int> p(n+1),re(n+1);
    vector<vector<int>> reg(r+1);
    cin >> re[1];
    reg[re[1]].push_back(1);
    for (int i =2; i <= n; i++) {
        cin >> p[i]>>re[i];
        reg[re[i]].push_back(i);
    }
    for (int i = n; i >= 1;i--) {
        cnt[i][re[i]]++;
        for (int j = 1; j <= r; j++) {
            cnt[p[i]][j]+=cnt[i][j];
        }
    }
    map<pair<int, int>,int>mp;
    for (int i =0; i < q; i++) {
        int r1, r2;
        cin >> r1 >> r2;
        int ans = 0;
        for (int j:reg[r1]) {
            ans += cnt[j][r2];
        }
        cout << ans << "\n";
        cout.flush();
    }

    //fclose(stdin);
    //fclose(stdout);
    return 0;
}

• Subtask 1: 15 / 15
• Subtask 2: 5 / 85