#include "split.h"
#include<bits/stdc++.h>
using namespace std;
vector<vector<int>> g;
vector<int> fo;
void dfs(int cur, int par){
fo.push_back(cur);
for (int next : g[cur]){
if (next != par) dfs(next, cur);
}
}
vector<int> find_split(int n, int a, int b, int c, vector<int> p, vector<int> q){
g.resize(n);
for (int i=0; i<p.size(); i++){
g[p[i]].push_back(q[i]);
g[q[i]].push_back(p[i]);
}
for (int i=0; i<n; i++){
if (g[i].size() == 1){
dfs(i, i);
break;
}
}
vector<int> res(n);
for (int i=0; i<n; i++){
if (i < a) res[fo[i]] = 1;
else if (i < a+b) res[fo[i]] = 2;
else res[fo[i]] = 3;
}
return res;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |