Submission #1279754

#TimeUsernameProblemLanguageResultExecution timeMemory
1279754phuocrucppCapital City (JOI20_capital_city)C++20
11 / 100
3096 ms66232 KiB
/*ㅤ∧_∧
 ( ・∀・)
 ( つ┳⊃
ε (_)へ⌒ヽフ
 (  ( ・ω・)
 ◎―◎   ⊃  ⊃
BePhuongSuperSuwi
From TK4 - CHT
ㅤㅤ/ ⌒\____
  /・   )  \
 /ノへ ノ    /|
ノ    \\ |/_/_/*/


#include<bits/stdc++.h>
#define task "main"
//#include <ext/pb_ds/assoc_container.hpp>
//#include <ext/pb_ds/tree_policy.hpp>
#define endl '\n'
#define int long long
#define pb push_back
#define fi first
#define se second
#define ii pair<int,int>
#define iii pair<int,ii>
#define iv pair<ii, ii>
#define base 341
#define MASK(i) (1ll << i)
#define oo 1e18
#define isOn(x,i) ((x) & MASK(i))
#define bitOn(x,i) ((x) | MASK(i))
#define bitOff(x,i) ((x) & ~MASK(i))
#define ordered_set tree<int, null_type,less<int>, rb_tree_tag,tree_order_statistics_node_update>
#define __lcm(a,b) (1ll * ((a) / __gcd((a), (b))) * (b))
using namespace std;
//using namespace __gnu_pbds;
const int maxn = 2e5 + 5;
const int lg = 18;
int n, k, h[maxn], par[maxn][lg + 2];
vector <int> g[maxn];

void dfs(int u, int p) {
	for (int x : g[u]) {
		if (x == p) continue;
		h[x] = h[u] + 1;
		par[x][0] = u;
		dfs(x, u);
	}
}
int lca(int u, int v) {
	if (h[u] < h[v]) swap(u,v);
	for (int i = lg; i >= 0; i--) {
		if ((h[u] - h[v]) >= MASK(i)) u = par[u][i];
	}
	if (u == v) return u;
	for (int i = lg; i >= 0; i--) {
		if (par[u][i] != par[v][i]) {
			u = par[u][i];
			v = par[v][i];
		}
	}
	return par[u][0];
}
void prelca() {
	dfs(1, -1);
	for (int j = 1; j <= lg; j++) {
		for (int i = 1; i <= n; i++) {
			par[i][j] = par[par[i][j-1]][j-1];
		}
	}
}

bool ok[maxn];
int parent[maxn], val[maxn];
vector <int> cur, c[maxn];

int acs(int u) {
    if (parent[u] == u) return u;
    else return parent[u] = acs(parent[u]);
}

void join(int u, int v) {
    int x = acs(u);
    int y = acs(v);
    if (x != y) {
        parent[x] = y;
    }
}

void connect(int u, int v) {
    int p = lca(u, v);
    while(h[u] > h[p]) {
        if (!ok[val[u]]) ok[val[u]] = 1, cur.pb(val[u]);
        join(u, par[u][0]);
        u = acs(u);
    }

    while(h[v] > h[p]) {
        if (!ok[val[v]]) ok[val[v]] = 1, cur.pb(val[v]);
        join(v, par[v][0]);
        v = acs(v);
    }

    if (!ok[val[p]]) ok[val[p]] = 1, cur.pb(val[p]);
}

int mini = oo;
void tinh(int id) {
    for (int i = 1; i <= n; i++) parent[i] = i, ok[i] = 0;
    cur.clear();
    ok[id] = 1;
    cur.pb(id);

    for (int i = 0; i < cur.size(); i++) {
        int x = cur[i];
        for (int i = 1; i < c[x].size(); i++) {
            connect(c[x][0], c[x][i]);
        }
//        cout << "!" << x << " ";
    }


//    cout << id << " " << cur.size() << endl;
//   cout << "hi ";
//    for (int x : cur)
//        cout << x << " ";
//    cout << endl;
    mini = min(mini, (int)cur.size() - 1);
}

main() {
    ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    if(fopen(task".inp","r")) {
        freopen(task".inp","r",stdin);
        freopen(task".out","w",stdout);
    }
    cin >> n >> k;
    for (int i = 1; i < n; i++) {
        int l, r;
        cin >> l >> r;
        g[l].pb(r); g[r].pb(l);
    }

    prelca();

    for (int i = 1; i <= n; i++) {
        cin >> val[i];
        c[val[i]].pb(i);
    }


    for (int i = 1; i <= k; i++) {
        tinh(i);
      //  break;
    }
    cout << mini;
}


Compilation message (stderr)

capital_city.cpp:131:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
  131 | main() {
      | ^~~~
capital_city.cpp: In function 'int main()':
capital_city.cpp:134:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  134 |         freopen(task".inp","r",stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~
capital_city.cpp:135:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  135 |         freopen(task".out","w",stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...