Submission #1290990

#TimeUsernameProblemLanguageResultExecution timeMemory
1290990gustavo_dSplit the Attractions (IOI19_split)C++20
11 / 100
68 ms16376 KiB
#include "split.h"
#include <bits/stdc++.h>
using namespace std;

const int MAXN = 1e5;

vector<int> adj[MAXN];
vector<int> res;
int n;
bool vis[MAXN];
vector<int> pre;

void dfs(int v) {
	pre.push_back(v);
	vis[v] = true;
	for (int viz : adj[v]) {
		if (vis[viz]) continue;
		dfs(viz);
	}
}

vector<int> find_split(int N, int a, int b, int c, vector<int> p, vector<int> q) {
	n = N;

	for (int i=0; i<(int)p.size(); i++) {
		adj[p[i]].push_back(q[i]);
		adj[q[i]].push_back(p[i]);
	}

	pre.reserve(n);
	res = vector<int> (n);

	dfs(0);

	for (int i=0; i<b; i++) res[pre[i]] = 2;
	res[pre[b]] = 1;
	for (int i=b+1; i<n; i++) res[pre[i]] = 3;

	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...