oj.uz

Submission #298957

# Submission time Handle Problem Language Result Execution time Memory
298957 2020-09-14T11:14:14 Z jovan_b Regions (IOI09_regions) C++17
0 / 100
 170 ms 79516 KB 
#include <bits/stdc++.h>
using namespace std;

typedef long long ll;

int LIM = 500;
int in[200005];
int out[200005];
int region[200005];
int parent[200005];
int cnt[200005];
vector <int> graf[200005];

int gore[25005][505];
int dole[25005][505];

int sada[25005];
int rev[25005];
int compress[25005];
int ccomp;

int r;

int tajm;

vector <pair <int, int>> svi[25005];
vector <int> ins[25005];

void dfs(int v){

    in[v] = ++tajm;
    sada[region[v]]++;
    /*if(compress[region[v]]){
        for(int i=1; i<=r; i++){
            dole[i][compress[region[v]]] += sada[region[v]];
        }
        dole[region[v]][compress[region[v]]]--;
        gore[region[v]][compress[region[v]]]--;
    }*/
    for(int i=1; i<=ccomp; i++){
        gore[region[v]][i] += sada[rev[i]];
    }
    for(auto c : graf[v]){
        dfs(c);
    }
    sada[region[v]]--;
    out[v] = tajm;
    svi[region[v]].push_back({in[v], 1});
    svi[region[v]].push_back({out[v]+1, -1});
    ins[region[v]].push_back(in[v]);
}

int main(){
    ios_base::sync_with_stdio(false), cin.tie(0);

    int n, q;
    cin >> n >> r >> q;
    cin >> region[1];
    cnt[region[1]]++;
    for(int i=2; i<=n; i++){
        cin >> parent[i] >> region[i];
        cnt[region[i]]++;
        graf[parent[i]].push_back(i);
    }
    for(int i=1; i<=r; i++){
        if(cnt[i] <= LIM) continue;
        compress[i] = ++ccomp;
        rev[ccomp] = i;
    }
    dfs(1);
    return 0;
    for(int i=1; i<=r; i++) sort(svi[i].begin(), svi[i].end());
    for(int i=1; i<=r; i++) sort(ins[i].begin(), ins[i].end());
    while(q--){
        int a, b;
        cin >> a >> b;
        if(cnt[a] > LIM){
            cout << gore[b][compress[a]] << endl;
            continue;
        }
        if(cnt[b] > LIM){
            cout << dole[a][compress[b]] << endl;
            continue;
        }
        int k = svi[a].size();
        int j = -1;
        int tren = 0;
        ll res = 0;
        for(auto c : ins[b]){
            while(j < k-1 && c >= svi[a][j+1].first){
                j++;
                tren += svi[a][j].second;
            }
            res += tren;
        }
        cout << res << endl;
    }

    return 0;
}

Subtask #1
0 / 15.0

# Verdict Execution time Memory Grader output
1 Incorrect 4 ms 6272 KB Unexpected end of file - int32 expected
2 Incorrect 5 ms 6272 KB Unexpected end of file - int32 expected
3 Incorrect 5 ms 6272 KB Unexpected end of file - int32 expected
4 Incorrect 5 ms 6272 KB Unexpected end of file - int32 expected
5 Incorrect 4 ms 6272 KB Unexpected end of file - int32 expected
6 Incorrect 5 ms 6272 KB Unexpected end of file - int32 expected
7 Incorrect 5 ms 6400 KB Unexpected end of file - int32 expected
8 Incorrect 5 ms 6400 KB Unexpected end of file - int32 expected
9 Incorrect 7 ms 6912 KB Unexpected end of file - int32 expected
10 Incorrect 8 ms 7040 KB Unexpected end of file - int32 expected
11 Incorrect 11 ms 7424 KB Unexpected end of file - int32 expected
12 Incorrect 12 ms 8064 KB Unexpected end of file - int32 expected
13 Incorrect 13 ms 7936 KB Unexpected end of file - int32 expected
14 Incorrect 18 ms 8704 KB Unexpected end of file - int32 expected
15 Incorrect 21 ms 11392 KB Unexpected end of file - int32 expected

Subtask #2
0 / 85.0

# Verdict Execution time Memory Grader output
1 Incorrect 34 ms 12764 KB Unexpected end of file - int32 expected
2 Incorrect 36 ms 11760 KB Unexpected end of file - int32 expected
3 Incorrect 35 ms 14964 KB Unexpected end of file - int32 expected
4 Incorrect 21 ms 8576 KB Unexpected end of file - int32 expected
5 Incorrect 20 ms 10240 KB Unexpected end of file - int32 expected
6 Incorrect 29 ms 10112 KB Unexpected end of file - int32 expected
7 Incorrect 40 ms 11640 KB Unexpected end of file - int32 expected
8 Incorrect 49 ms 16896 KB Unexpected end of file - int32 expected
9 Incorrect 85 ms 19320 KB Unexpected end of file - int32 expected
10 Incorrect 80 ms 23544 KB Unexpected end of file - int32 expected
11 Incorrect 108 ms 19832 KB Unexpected end of file - int32 expected
12 Incorrect 141 ms 52004 KB Unexpected end of file - int32 expected
13 Incorrect 121 ms 52080 KB Unexpected end of file - int32 expected
14 Incorrect 170 ms 59624 KB Unexpected end of file - int32 expected
15 Incorrect 133 ms 74732 KB Unexpected end of file - int32 expected
16 Incorrect 127 ms 79516 KB Unexpected end of file - int32 expected
17 Incorrect 128 ms 68332 KB Unexpected end of file - int32 expected