Submission #1215542

#TimeUsernameProblemLanguageResultExecution timeMemory
1215542dostsKeys (IOI21_keys)C++20
37 / 100
3115 ms431152 KiB
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx2")
//#define int long long
#define pii pair<int,int>
#define vi vector<int>
#define ff first
#define ss second
#define sp << " " <<
#define all(x) x.begin(),x.end()
#define big(x) ((int)(x.size()))
using namespace std;
using namespace __gnu_pbds;
const int MOD = 1e9+7, LIM = 3e5+1, inf = 2e9;
vi edges[LIM],egdes[LIM], vis(LIM,0), col(LIM);
vi order;

void topo(int node) {
	if (vis[node]) return;
	vis[node] = 1;
	for (auto it : edges[node]) {
		topo(it);
	}
	order.push_back(node);
}

int color(int node,int c) {
	if (vis[node]) return 0;
	vis[node] = 1;
	col[node] = c;
	int ans = 1;
	for (auto it : egdes[node]) ans+=color(it,c);
	return ans;
}

std::vector<int> find_reachable(std::vector<int> r, std::vector<int> u, std::vector<int> v, std::vector<int> c) {
	int n = r.size(),m = u.size();
	vi canreach[n];
	vi cols[n];
	gp_hash_table<int,vi> edg[n];
	for (int i = 0;i<m;i++) {
		if (r[u[i]] == c[i]) canreach[v[i]].push_back(u[i]);
		if (r[v[i]] == c[i]) canreach[u[i]].push_back(v[i]);
		edg[u[i]][c[i]].push_back(v[i]);
		edg[v[i]][c[i]].push_back(u[i]);
	}
	queue<pii> q;
	for (int i = 0;i<n;i++) q.push({i,r[i]}),cols[i].push_back(r[i]);
	set<pii> done;
	while (!q.empty()) {
		pii f = q.front();
		q.pop();
		int city = f.ff,col = f.ss;
		if (done.count({city,col})) continue;
		done.insert({city,col});
		for (auto it : canreach[city]) {
			if (big(edg[it][col])) {
				cols[it].push_back(col);
				q.push({it,col});
			}
		}
		for (auto it : edg[city][col]) {
			edges[city].push_back(it);
			egdes[it].push_back(city);
			canreach[it].push_back(city);
			for (auto it : cols[it]) {
				cols[city].push_back(it);
				q.push({city,it});
			}
		}
	}
	vi ans(n,0);
	int fnd = 0;
	for (int i = 0;i<n;i++) {
		if (edges[i].empty()) {
			fnd = 1;
			ans[i] = 1;
		}
	}
	if (fnd) return ans;
	for (int i = 0;i<n;i++) topo(i);
	reverse(all(order));
	vis.assign(n,0);
	int cc = 0;
	vi szs;
	for (int i = 0;i<n;i++) {
		if (!vis[order[i]]) {
			szs.push_back(color(order[i],cc++));
		}
	}
	reverse(all(order));
	for (auto i : order) {
		for (auto it : edges[i]) {
			if (col[i] != col[it]) {
				szs[col[i]]+=szs[col[it]];
			}
		}
	}
	int x = inf;
	for (int i=0;i<n;i++) {
		if (szs[col[i]] < x) {
			x = szs[col[i]];
		}
	}
	for (int i = 0;i<n;i++) {
		if (szs[col[i]] == x) ans[i] = 1;
	}
	return ans;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...