#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
vector<int> find_split(int n, int a, int b, int c, vector<int> p, vector<int> q){
vector<vector<int>> G(n);
int m = p.size();
vector<int> ans(n,3);
for (int i = 0; i < m; ++i){
G[p[i]].push_back(q[i]);
G[q[i]].push_back(p[i]);
}
int start = 0;
for (int i = 0; i < n; ++i){
if(G[i].size()==1){
start = i;
break;
}
}
int cur = start;
ans[start]=1;
int cnt = 1;
while(cnt<a){
for(auto &i : G[cur]){
if(ans[i]!=1){
ans[i]=1;
cur = i;
++cnt;
break;
}
}
}
cnt = 0;
while(cnt<b){
for(auto &i : G[cur]){
if(ans[i]==3){
ans[i]=2;
cur = i;
++cnt;
}
}
}
return ans;
}
# | 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... |