Submission #763777

# Submission time Handle Problem Language Result Execution time Memory
763777 2023-06-22T20:57:25 Z izanbf Regions (IOI09_regions) C++14
100 / 100
4757 ms 99964 KB
#include <bits/stdc++.h>
using namespace std;

using vi = vector<int>;
using vvi = vector<vi>;

const int BIG = 400;
vi r, tin, tout, big_id;
vvi adj, tins, touts, by_r, dp;
int timer;

void euler(int u, int p) {
    tin[u] = timer++;
    tins[r[u]].push_back(tin[u]);
    for (int v : adj[u]) {
        if (v != p) euler(v, u);
    }
    tout[u] = timer++;
    touts[r[u]].push_back(tout[u]);
}

int count_in_range(const vi& v, int l, int r) {
    if (v.empty()) return 0;
    int ri = upper_bound(v.begin(), v.end(), r) - v.begin();
    int li = upper_bound(v.begin(), v.end(), l) - v.begin();
    return ri - li;
}

void precompute(int rg, int p, int u) {
    int rid = big_id[rg];
    dp[rid][u] = (r[u] == rg);
    for (int v : adj[u]) {
        if (v != p) {
            precompute(rg, u, v);
            dp[rid][u] += dp[rid][v];
        }
    }
}

int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    
    int N, R, Q;
    cin >> N >> R >> Q;

    r = tin = tout = vi(N);
    big_id = vi(R);
    adj = vvi(N);
    tins = touts = by_r = vvi(R);

    vi p(N, -1);

    cin >> r[0]; --r[0];
    by_r[r[0]].push_back(0);

    for (int i = 1; i < N; ++i) {
        cin >> p[i] >> r[i]; --p[i], --r[i];
        by_r[r[i]].push_back(i);
        adj[i].push_back(p[i]);
        adj[p[i]].push_back(i);
    }

    int bigs = 0;
    for (int rg = 0; rg < R; ++rg) {
        if (by_r[rg].size() >= BIG) big_id[rg] = bigs++;
    }

    dp = vvi(bigs, vi(N)); // O(N/BIG N)
    for (int rg = 0; rg < R; ++rg) {
        if (by_r[rg].size() >= BIG) precompute(rg, 0, 0);
    }

    timer = 1;
    euler(0, 0);

    map<pair<int,int>,long long> mem;

    while (Q--) {
        int r1, r2;
        cin >> r1 >> r2; --r1, --r2;
        // cerr << 1+r1 << " " << 1+r2 << endl;
        if (by_r[r1].size() < BIG and by_r[r2].size() >= BIG) { // r1 small r2 big  O(Q BIG)
            long long cnt = 0;
            int rid = big_id[r2];
            for (int u : by_r[r1]) {
                cnt += dp[rid][u];
            }
            cout << cnt << endl;
        }
        else { //O(Q BIG log N)
            if (not mem.count({r1,r2})) {
                long long cnt = 0;
                for (int u : by_r[r2]) {
                    cnt += count_in_range(tins[r1], 0, tin[u])
                         - count_in_range(touts[r1], 0, tin[u]);
                }
                mem[{r1,r2}] = cnt;
            }
            cout << mem[{r1,r2}] << endl;
        }
    }
}




# Verdict Execution time Memory Grader output
1 Correct 1 ms 208 KB Output is correct
2 Correct 1 ms 208 KB Output is correct
3 Correct 1 ms 208 KB Output is correct
4 Correct 4 ms 384 KB Output is correct
5 Correct 8 ms 336 KB Output is correct
6 Correct 14 ms 564 KB Output is correct
7 Correct 28 ms 524 KB Output is correct
8 Correct 33 ms 712 KB Output is correct
9 Correct 30 ms 1432 KB Output is correct
10 Correct 98 ms 1708 KB Output is correct
11 Correct 137 ms 2248 KB Output is correct
12 Correct 147 ms 3204 KB Output is correct
13 Correct 196 ms 3256 KB Output is correct
14 Correct 292 ms 3788 KB Output is correct
15 Correct 290 ms 7348 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 1087 ms 15092 KB Output is correct
2 Correct 1287 ms 13196 KB Output is correct
3 Correct 1972 ms 18240 KB Output is correct
4 Correct 272 ms 5152 KB Output is correct
5 Correct 292 ms 7832 KB Output is correct
6 Correct 484 ms 20288 KB Output is correct
7 Correct 809 ms 16836 KB Output is correct
8 Correct 1121 ms 57704 KB Output is correct
9 Correct 2231 ms 25132 KB Output is correct
10 Correct 3551 ms 99964 KB Output is correct
11 Correct 4757 ms 33376 KB Output is correct
12 Correct 1344 ms 24088 KB Output is correct
13 Correct 1878 ms 27172 KB Output is correct
14 Correct 2273 ms 44464 KB Output is correct
15 Correct 3024 ms 37352 KB Output is correct
16 Correct 3257 ms 46360 KB Output is correct
17 Correct 3141 ms 60008 KB Output is correct