oj.uz

Submission #537567

# Submission time Handle Problem Language Result Execution time Memory
537567 2022-03-15T08:47:12 Z Vladth11 Capital City (JOI20_capital_city) C++14
11 / 100
 3000 ms 38152 KB 
#include <bits/stdc++.h>
#define debug(x) cerr << #x << " " << x << "\n"
#define debugs(x) cerr << #x << " " << x << " "

using namespace std;
typedef long long ll;
typedef pair <ll, ll> pii;
typedef pair <long double, pii> muchie;

const ll NMAX = 200001;
const ll VMAX = 1000001;
const ll INF = (1LL << 60);
const ll MOD = 1000000007;
const ll BLOCK = 1000000;
const ll nr_of_bits = 21;

vector <int> v[NMAX], colors[NMAX];

int c[NMAX];
int viz[NMAX];
int instance[NMAX];
int stamp;
int added[NMAX];
int p[NMAX];
int sz[NMAX];
int minim = 2e9;
int total = 0;

void getSize(int node, int p){
    sz[node] = 1;
    for(auto x : v[node]){
        if(viz[x] || x == p) continue;
        getSize(x, node);
        sz[node] += sz[x];
    }
}

int findCentroid(int node, int p){
    for(auto x : v[node]){
        if(x == p || viz[x]) continue;
        if(sz[x] > total / 2){
            return findCentroid(x, node);
        }
    }
    return node;
}

int getCentroid(int node){
    getSize(node, 0);
    return findCentroid(node, 0);
}

void DFS(int node, int parent){
    p[node] = parent;
    instance[node] = stamp;
    for(auto x : v[node]){
        if(x == parent) continue;
        if(viz[x]) continue;
        DFS(x, node);
    }
}

void Centroid(int node){
    int centroid = getCentroid(node);
    viz[centroid] = 1;
    stamp++;
    instance[centroid] = stamp;
    for(auto x : v[centroid]){
        if(viz[x])
            continue;
        DFS(x, centroid);
    }
    queue <int> q;
    int ok = 1;
    q.push(c[centroid]);
    added[c[centroid]] = stamp;
    int cnt = 1;
    while(q.size()){
        int color = q.front();
        q.pop();
        for(auto x : colors[color]){
            if(instance[x] != stamp){
                ok = 0;
                break;
            }
            if(x != centroid && added[c[p[x]]] != stamp){
                q.push(c[p[x]]);
                added[c[p[x]]] = stamp;
                cnt++;
            }
        }
    }
    if(ok){
        minim = min(minim, cnt - 1);
    }
    for(auto x : v[node]){
        if(viz[x]) continue;
        Centroid(x);
    }
}

int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    int n, i, k;
    cin >> n >> k;
    for(i = 2; i <= n; i++){
        int a, b;
        cin >> a >> b;
        v[a].push_back(b);
        v[b].push_back(a);
    }
    for(i = 1; i <= n; i++){
        cin >> c[i];
        colors[c[i]].push_back(i);
    }
    Centroid(1);
    cout << minim;
    return 0;
}

Subtask #1
1.0 / 1.0

# Verdict Execution time Memory Grader output
1 Correct 6 ms 9684 KB Output is correct
2 Correct 8 ms 9744 KB Output is correct
3 Correct 5 ms 9684 KB Output is correct
4 Correct 5 ms 9684 KB Output is correct
5 Correct 7 ms 9684 KB Output is correct
6 Correct 6 ms 9752 KB Output is correct
7 Correct 6 ms 9700 KB Output is correct
8 Correct 6 ms 9684 KB Output is correct
9 Correct 7 ms 9684 KB Output is correct
10 Correct 6 ms 9684 KB Output is correct

Subtask #2
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 6 ms 9684 KB Output is correct
2 Correct 8 ms 9744 KB Output is correct
3 Correct 5 ms 9684 KB Output is correct
4 Correct 5 ms 9684 KB Output is correct
5 Correct 7 ms 9684 KB Output is correct
6 Correct 6 ms 9752 KB Output is correct
7 Correct 6 ms 9700 KB Output is correct
8 Correct 6 ms 9684 KB Output is correct
9 Correct 7 ms 9684 KB Output is correct
10 Correct 6 ms 9684 KB Output is correct
11 Correct 74 ms 11372 KB Output is correct
12 Correct 73 ms 11416 KB Output is correct
13 Correct 70 ms 11336 KB Output is correct
14 Correct 72 ms 11332 KB Output is correct
15 Correct 47 ms 11340 KB Output is correct
16 Correct 62 ms 11324 KB Output is correct
17 Correct 60 ms 11484 KB Output is correct
18 Correct 57 ms 11408 KB Output is correct
19 Correct 55 ms 11392 KB Output is correct
20 Correct 51 ms 11340 KB Output is correct
21 Correct 53 ms 11408 KB Output is correct
22 Correct 47 ms 11340 KB Output is correct
23 Correct 73 ms 11408 KB Output is correct
24 Correct 53 ms 11392 KB Output is correct
25 Correct 77 ms 11340 KB Output is correct
26 Correct 80 ms 11332 KB Output is correct
27 Correct 82 ms 11392 KB Output is correct
28 Correct 74 ms 11340 KB Output is correct
29 Correct 76 ms 11416 KB Output is correct
30 Correct 69 ms 11404 KB Output is correct

Subtask #3
0 / 30.0

# Verdict Execution time Memory Grader output
1 Execution timed out 3070 ms 38152 KB Time limit exceeded
2 Halted 0 ms 0 KB -

Subtask #4
0 / 59.0

# Verdict Execution time Memory Grader output
1 Correct 6 ms 9684 KB Output is correct
2 Correct 8 ms 9744 KB Output is correct
3 Correct 5 ms 9684 KB Output is correct
4 Correct 5 ms 9684 KB Output is correct
5 Correct 7 ms 9684 KB Output is correct
6 Correct 6 ms 9752 KB Output is correct
7 Correct 6 ms 9700 KB Output is correct
8 Correct 6 ms 9684 KB Output is correct
9 Correct 7 ms 9684 KB Output is correct
10 Correct 6 ms 9684 KB Output is correct
11 Correct 74 ms 11372 KB Output is correct
12 Correct 73 ms 11416 KB Output is correct
13 Correct 70 ms 11336 KB Output is correct
14 Correct 72 ms 11332 KB Output is correct
15 Correct 47 ms 11340 KB Output is correct
16 Correct 62 ms 11324 KB Output is correct
17 Correct 60 ms 11484 KB Output is correct
18 Correct 57 ms 11408 KB Output is correct
19 Correct 55 ms 11392 KB Output is correct
20 Correct 51 ms 11340 KB Output is correct
21 Correct 53 ms 11408 KB Output is correct
22 Correct 47 ms 11340 KB Output is correct
23 Correct 73 ms 11408 KB Output is correct
24 Correct 53 ms 11392 KB Output is correct
25 Correct 77 ms 11340 KB Output is correct
26 Correct 80 ms 11332 KB Output is correct
27 Correct 82 ms 11392 KB Output is correct
28 Correct 74 ms 11340 KB Output is correct
29 Correct 76 ms 11416 KB Output is correct
30 Correct 69 ms 11404 KB Output is correct
31 Execution timed out 3070 ms 38152 KB Time limit exceeded
32 Halted 0 ms 0 KB -