Submission #810434

#TimeUsernameProblemLanguageResultExecution timeMemory
810434NothingXDSplit the Attractions (IOI19_split)C++17
40 / 100
66 ms17516 KiB
#include "split.h"
#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
typedef double ld;

void debug_out(){cerr<<endl;}
template<typename Head, typename... Tail> 
void debug_out(Head H, Tail... T){
	cerr << H << ' ';
	debug_out(T...);
}

#define debug(...) cerr << "(" << #__VA_ARGS__ << "): ", debug_out(__VA_ARGS__)
#define F first
#define S second
#define all(x) x.begin(), x.end()
#define MP(x, y) make_pair(x, y)

const int maxn = 1e5 + 10;

int n, m, a, b, c, A, B, C, sz[maxn], r1 = -1, r2 = -1, ans[maxn];
vector<int> g[maxn];
vector<pii> edge;
bool vis[maxn];

void dfs(int v){
	vis[v] = true;
	for (auto u: g[v]){
		if (!vis[u]){
			edge.push_back({v, u});
			dfs(u);
		}
	}
}

void find(int v, int p = -1){
	sz[v] = 1;
	for (auto u: g[v]){
		if (u != p){
			find(u, v);
			sz[v] += sz[u];
		}
	}
	if (min(n-sz[v], sz[v]) >= a){
		r1 = v;
		r2 = p;
	}
}

int solve(int v, int p, int cnt, int val){
	if (cnt == 0) return 0;
	ans[v] = val;
	cnt--;
	for (auto u: g[v]){
		if (u != p){
			cnt = solve(u, v, cnt, val);
		}
	}
	return cnt;
}

vector<int> find_split(int _n, int _a, int _b, int _c, vector<int> p, vector<int> q) {
	
	n = _n;
	m = p.size();
	a = _a;
	b = _b;
	c = _c;
	A = 1, B = 2, C = 3;
	if (a > b){
		swap(a, b);
		swap(A, B);
	}
	if (b > c){
		swap(b, c);
		swap(B, C);
	}
	if (a > b){
		swap(a, b);
		swap(A, B);
	}
	for (int i = 0; i < m; i++){
		g[p[i]].push_back(q[i]);
		g[q[i]].push_back(p[i]);
	}
	dfs(0);
	for (int i = 0; i < n; i++) g[i].clear();
	for (auto [u, v]: edge){
		g[u].push_back(v);
		g[v].push_back(u);
	}
	find(0);
	vector<int> res(n, 0);
	if (r1 == -1) return res;
	res = vector<int>(n, C);
	if (sz[r1] >= b) swap(r1, r2);
	solve(r1, r2, a, A);
	solve(r2, r1, b, B);
	for (int i = 0; i < n; i++){
		if (ans[i] != 0) res[i] = ans[i];
	}
	return res;
}
#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...