Submission #1369152

#TimeUsernameProblemLanguageResultExecution timeMemory
1369152clemmy14Mergers (JOI19_mergers)C++20
0 / 100
63 ms22108 KiB
#include<bits/stdc++.h>
using namespace std;

vector<int> t, depth;
vector<bool> vis;
vector<vector<int>> adj, col, par;

void dfs(int node, int prev, int d) {
    depth[node]=d;
    par[node][0]=prev;
    col[t[node]].push_back(node);
    for(auto child : adj[node]) if(child != prev) {
        dfs(child, node, d+1);
    }
}

int ancestor(int x, int k) {
    int cnt=0;
    while(k&&x) {
        if(k&1) x=par[x][cnt];
        k=k>>1; cnt++;
    }
    return x;
}

int lca(int a, int b) {
    if(depth[a] < depth[b]) swap(a, b);
    a=ancestor(a, depth[a]-depth[b]);
    if(a == b) return a;
    for(int k=19; k>=0; k--) {
        int aa=par[a][k];
        int bb=par[b][k];
        if(aa != bb) {
            a=aa; b=bb;
        }
    }
    return par[a][0];
}

struct DSU{
    vector<int> e;
    DSU(int N) {e=vector<int>(N, -1);}
    int get(int x) {return e[x] < 0 ? x : e[x]=get(e[x]);}
    int size(int x) {return -e[get(x)];}
    bool unite(int a, int b) {
        a=get(a); b=get(b);
        if(a == b) return false;
        if(e[a] > e[b]) swap(a, b);
        e[a]+=e[b]; e[b]=a;
        return true;
    }
};

void up(int node, int tar, DSU&dsu) {
    if(node == tar) return;
    if(!dsu.unite(node, par[node][0])) return; 
    up(par[node][0], tar, dsu);
}

signed main() {
    int n, k; cin >> n >> k;
    adj=vector<vector<int>>(n);
    for(int i=1; i<n; i++) {
        int a, b; cin >> a >> b; a--; b--;
        adj[a].push_back(b);
        adj[b].push_back(a);
    }
    t=depth=vector<int>(n);
    for(int i=0; i<n; i++) {
        cin >> t[i]; t[i]--;
    }
    col=vector<vector<int>>(k);
    par=vector<vector<int>>(n, vector<int>(20, 0));
    dfs(0, 0, 0);
    for(int j=1; j<20; j++) {
        for(int i=0; i<n; i++) {
            if(par[i][j-1]) par[i][j]=par[par[i][j-1]][j-1];
        }
    }

    DSU dsu(n);
    for(int i=0; i<k; i++) {
        int top=col[i][0];
        for(int j=1; j<col[i].size(); j++) top=lca(top, col[i][j]);
        for(int j=1; j<col[i].size(); j++) {
            int child=col[i][j];
            // int top=lca(node, child);
            up(child, top, dsu);
        }       
    }

    vector<vector<int>> newAdj(n);
    // for(int i=0; i<n; i++) {
    //     cout << dsu.get(i) << endl;
    // }
    // cout << endl;

    for(int i=0; i<n; i++) {
        for(auto x : adj[i]) {
            int ii=dsu.get(i), xx=dsu.get(x);
            if(ii != xx) {
                newAdj[ii].push_back(xx);
            }
        }
    }

    // for(int i=0; i<n; i++) {
    //     for(int j=0; j<newAdj[i].size(); j++) cout << newAdj[i][j] << ' ';
    //     cout << endl;
    // }
    // cout << endl;

    int leaves=0;
    for(int i=0; i<n; i++) {
        if(newAdj[i].size() == 1) leaves++;
    }
    if(leaves == 1) cout << "1";
    else cout << (leaves+1)/2;
    return 0;
}
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...