oj.uz

Submission #1001768

# Submission time Handle Problem Language Result Execution time Memory
1001768 2024-06-19T07:37:03 Z Unforgettablepl Capital City (JOI20_capital_city) C++17
11 / 100
 622 ms 52228 KB 
#include <bits/stdc++.h>
using namespace std;

#define int long long

void strat1(int n){
    int k;
    cin >> k;
    vector<vector<int>> adj(n+1);
    for(int i=1;i<n;i++){
        int a,b;cin>>a>>b;
        adj[a].emplace_back(b);
        adj[b].emplace_back(a);
    }
    vector<int> col(n+1);
    for(int i=1;i<=n;i++)cin>>col[i];
    vector<int> nodes(k+1);
    function<void(int,int,int)> dfs = [&](int x,int par,int dep){
        nodes[col[x]] = x;
        for(int&i:adj[x])if(i!=par)dfs(i,x,dep+1);
    };
    dfs(1,0,1);
    vector<vector<int>> compoadj(k+1);
    for(int i=1;i<=k;i++){
        set<int> bad;
        function<bool(int,int)> pdfs = [&](int x,int p){
            bool found = col[x]==i;
            for(int&i:adj[x])if(i!=p)if(pdfs(i,x))found=true;
            if(found)bad.insert(col[x]);
            return found;
        };
        pdfs(nodes[i],0);
        for(int x:bad)compoadj[i].emplace_back(x);
    }
    vector<bool> visited(k+1);
    int siz = 0;
    function<void(int)> compodfs = [&](int x){
        if(visited[x])return;
        siz++;
        visited[x]=true;
        for(int&i:compoadj[x])compodfs(i);
    };
    int ans = n;
    for(int i=1;i<=k;i++){
        siz = 0;
        visited = vector<bool>(k+1);
        compodfs(i);
        ans = min(ans,siz-1);
    }
    cout << ans << '\n';
}

int32_t main(){
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    int n;
    cin >> n;
    if(n<=2000){
        strat1(n);
        return 0;
    }
    int k;
    cin >> k;
    vector<vector<int>> adj(n+1);
    for(int i=1;i<n;i++){
        int a,b;cin>>a>>b;
        adj[a].emplace_back(b);
        adj[b].emplace_back(a);
    }
    vector<int> colcor(n+1);
    vector<int> col(n+1);
    for(int i=1;i<=n;i++)cin>>colcor[i];
    function<void(int,int,int)> dfs = [&](int x,int par,int idx){
        col[idx] = colcor[x];
        for(int&i:adj[x])if(i!=par)dfs(i,x,idx+1);
    };
    for(int i=1;i<=n;i++)if(adj[i].size()==1){
        dfs(i,0,1);
        break;
    }
    vector<int> minidx(k+1);
    vector<int> maxidx(k+1);
    for(int i=1;i<=n;i++)maxidx[col[i]]=i;
    for(int i=n;i;i--)minidx[col[i]]=i;
    vector<vector<int>> compoadj(k+1);
    vector<vector<int>> comporevadj(k+1);
    for(int i=1;i<=k;i++){
        for(int x=minidx[i];x<=maxidx[i];x++){
            if(col[x]!=i){
                compoadj[i].emplace_back(col[x]);
                comporevadj[col[x]].emplace_back(i);
                x = maxidx[col[x]];
            }
        }
    }
    vector<bool> visited(k+1);
    vector<int> order;
    vector<int> siz(k+1);
    vector<int> mycompo(k+1);
    function<void(int)> topo = [&](int x){
        if(visited[x])return;
        visited[x]=true;
        for(int&i:compoadj[x])topo(i);
        order.emplace_back(x);
    };
    for(int i=1;i<=k;i++)if(!visited[i])topo(i);
    reverse(order.begin(),order.end());
    function<void(int,int)> cr_dfs = [&](int x,int compo){
        if(mycompo[x]){
            if(mycompo[x]!=compo)siz[mycompo[x]]=n+1;
            return;
        }
        siz[compo]++;
        mycompo[x] = compo;
        for(int&i:comporevadj[x])cr_dfs(i,compo);
    };
    int components = 0;
    int ans = n+1;
    for(int i=0;i<k;i++)if(mycompo[order[i]]==0)cr_dfs(order[i],++components);
    for(int i=1;i<=components;i++){
        ans = min(ans,siz[components]-1);
    }
    cout << ans << '\n';
}

Subtask #1
1.0 / 1.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Correct 0 ms 392 KB Output is correct
9 Correct 0 ms 348 KB Output is correct
10 Correct 0 ms 348 KB Output is correct

Subtask #2
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Correct 0 ms 392 KB Output is correct
9 Correct 0 ms 348 KB Output is correct
10 Correct 0 ms 348 KB Output is correct
11 Correct 19 ms 708 KB Output is correct
12 Correct 14 ms 740 KB Output is correct
13 Correct 14 ms 748 KB Output is correct
14 Correct 20 ms 604 KB Output is correct
15 Correct 20 ms 736 KB Output is correct
16 Correct 37 ms 1436 KB Output is correct
17 Correct 11 ms 600 KB Output is correct
18 Correct 11 ms 716 KB Output is correct
19 Correct 11 ms 600 KB Output is correct
20 Correct 10 ms 604 KB Output is correct
21 Correct 11 ms 600 KB Output is correct
22 Correct 622 ms 6176 KB Output is correct
23 Correct 575 ms 5716 KB Output is correct
24 Correct 18 ms 604 KB Output is correct
25 Correct 19 ms 860 KB Output is correct
26 Correct 15 ms 864 KB Output is correct
27 Correct 19 ms 860 KB Output is correct
28 Correct 16 ms 868 KB Output is correct
29 Correct 15 ms 860 KB Output is correct
30 Correct 13 ms 872 KB Output is correct

Subtask #3
0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 100 ms 52168 KB Output is correct
2 Correct 71 ms 52112 KB Output is correct
3 Correct 122 ms 52228 KB Output is correct
4 Correct 83 ms 52168 KB Output is correct
5 Correct 95 ms 51152 KB Output is correct
6 Correct 70 ms 51924 KB Output is correct
7 Correct 92 ms 50128 KB Output is correct
8 Correct 67 ms 49608 KB Output is correct
9 Incorrect 85 ms 43584 KB Output isn't correct
10 Halted 0 ms 0 KB -

Subtask #4
0 / 59.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Correct 0 ms 392 KB Output is correct
9 Correct 0 ms 348 KB Output is correct
10 Correct 0 ms 348 KB Output is correct
11 Correct 19 ms 708 KB Output is correct
12 Correct 14 ms 740 KB Output is correct
13 Correct 14 ms 748 KB Output is correct
14 Correct 20 ms 604 KB Output is correct
15 Correct 20 ms 736 KB Output is correct
16 Correct 37 ms 1436 KB Output is correct
17 Correct 11 ms 600 KB Output is correct
18 Correct 11 ms 716 KB Output is correct
19 Correct 11 ms 600 KB Output is correct
20 Correct 10 ms 604 KB Output is correct
21 Correct 11 ms 600 KB Output is correct
22 Correct 622 ms 6176 KB Output is correct
23 Correct 575 ms 5716 KB Output is correct
24 Correct 18 ms 604 KB Output is correct
25 Correct 19 ms 860 KB Output is correct
26 Correct 15 ms 864 KB Output is correct
27 Correct 19 ms 860 KB Output is correct
28 Correct 16 ms 868 KB Output is correct
29 Correct 15 ms 860 KB Output is correct
30 Correct 13 ms 872 KB Output is correct
31 Correct 100 ms 52168 KB Output is correct
32 Correct 71 ms 52112 KB Output is correct
33 Correct 122 ms 52228 KB Output is correct
34 Correct 83 ms 52168 KB Output is correct
35 Correct 95 ms 51152 KB Output is correct
36 Correct 70 ms 51924 KB Output is correct
37 Correct 92 ms 50128 KB Output is correct
38 Correct 67 ms 49608 KB Output is correct
39 Incorrect 85 ms 43584 KB Output isn't correct
40 Halted 0 ms 0 KB -