oj.uz

Submission #537575

# Submission time Handle Problem Language Result Execution time Memory
537575 2022-03-15T08:51:20 Z Vladth11 Capital City (JOI20_capital_city) C++14
11 / 100
 3000 ms 38184 KB 
/// Asta nu ar trebui sa mearga, dar ca idee
#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];
    }
    total = sz[node];
}

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;
            }
            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[centroid]){
        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 5 ms 9748 KB Output is correct
2 Correct 7 ms 9684 KB Output is correct
3 Correct 7 ms 9684 KB Output is correct
4 Correct 7 ms 9684 KB Output is correct
5 Correct 5 ms 9684 KB Output is correct
6 Correct 6 ms 9684 KB Output is correct
7 Correct 5 ms 9684 KB Output is correct
8 Correct 7 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 5 ms 9748 KB Output is correct
2 Correct 7 ms 9684 KB Output is correct
3 Correct 7 ms 9684 KB Output is correct
4 Correct 7 ms 9684 KB Output is correct
5 Correct 5 ms 9684 KB Output is correct
6 Correct 6 ms 9684 KB Output is correct
7 Correct 5 ms 9684 KB Output is correct
8 Correct 7 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 39 ms 9812 KB Output is correct
12 Correct 31 ms 9812 KB Output is correct
13 Correct 31 ms 9812 KB Output is correct
14 Correct 32 ms 9812 KB Output is correct
15 Correct 17 ms 9920 KB Output is correct
16 Correct 13 ms 9812 KB Output is correct
17 Correct 13 ms 9948 KB Output is correct
18 Correct 11 ms 9940 KB Output is correct
19 Correct 10 ms 9960 KB Output is correct
20 Correct 11 ms 9940 KB Output is correct
21 Correct 14 ms 9956 KB Output is correct
22 Correct 17 ms 10036 KB Output is correct
23 Correct 28 ms 9940 KB Output is correct
24 Correct 19 ms 9916 KB Output is correct
25 Correct 24 ms 9980 KB Output is correct
26 Correct 18 ms 10020 KB Output is correct
27 Correct 22 ms 9996 KB Output is correct
28 Correct 18 ms 9940 KB Output is correct
29 Correct 19 ms 9948 KB Output is correct
30 Correct 14 ms 9940 KB Output is correct

Subtask #3
0 / 30.0

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

Subtask #4
0 / 59.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 9748 KB Output is correct
2 Correct 7 ms 9684 KB Output is correct
3 Correct 7 ms 9684 KB Output is correct
4 Correct 7 ms 9684 KB Output is correct
5 Correct 5 ms 9684 KB Output is correct
6 Correct 6 ms 9684 KB Output is correct
7 Correct 5 ms 9684 KB Output is correct
8 Correct 7 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 39 ms 9812 KB Output is correct
12 Correct 31 ms 9812 KB Output is correct
13 Correct 31 ms 9812 KB Output is correct
14 Correct 32 ms 9812 KB Output is correct
15 Correct 17 ms 9920 KB Output is correct
16 Correct 13 ms 9812 KB Output is correct
17 Correct 13 ms 9948 KB Output is correct
18 Correct 11 ms 9940 KB Output is correct
19 Correct 10 ms 9960 KB Output is correct
20 Correct 11 ms 9940 KB Output is correct
21 Correct 14 ms 9956 KB Output is correct
22 Correct 17 ms 10036 KB Output is correct
23 Correct 28 ms 9940 KB Output is correct
24 Correct 19 ms 9916 KB Output is correct
25 Correct 24 ms 9980 KB Output is correct
26 Correct 18 ms 10020 KB Output is correct
27 Correct 22 ms 9996 KB Output is correct
28 Correct 18 ms 9940 KB Output is correct
29 Correct 19 ms 9948 KB Output is correct
30 Correct 14 ms 9940 KB Output is correct
31 Execution timed out 3071 ms 38184 KB Time limit exceeded
32 Halted 0 ms 0 KB -