This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include "split.h"
//#include "grader.cpp"
#include <bits/stdc++.h>
using namespace std;
#define lol long long
#define pii pair<int,int>
#define OK puts("OK");
#define NO puts("NO");
#define YES puts("YES");
#define fr first
#define sc second
#define ret return
#define scanl(a) scanf("%lld",&a);
#define scanll(a,b) scanf("%lld %lld",&a, &b);
#define scanlll(a,b,c) scanf("%lld %lld %lld",&a,&b,&c);
#define scan1(a) scanf("%d",&a);
#define scan2(a,b) scanf("%d %d",&a, &b);
#define scan3(a,b,c) scanf("%d %d %d",&a,&b,&c);
#define all(s) s.begin(),s.end()
#define allr(s) s.rbegin()Ñ,s.rend()
#define pb push_back
#define sz(v) (int)v.size()
#define endi puts("");
#define eps 1e-12
vector <int> v;
bool vis[100001];
int inex[100001];
vector <int> g[100001];
void dfs(int x,int p){
vis[x]=1;
for (int to : g[x]){
if (to != p && vis[to] == 0)dfs(to,x);
}
v.pb(x);
}
vector<int> find_split(int n, int a, int b, int c, vector<int> p, vector<int> q) {
vector<int> res;
int i,j,s=0;
for (i=0;i<p.size();++i){
g[p[i]].pb(q[i]);
g[q[i]].pb(p[i]);
}
for (i=0;i<n;++i){
if (g[i].size() == 1){
s = i;
}
}
dfs(s,-1);
for (i=0;i<v.size();++i){
if (i < a)inex[v[i]] = 1;
else if (i < a+b)inex[v[i]] = 2;
else inex[v[i]]=3;
}
for (i=0;i<n;++i){
res.pb(inex[i]);
}
return res;
}
Compilation message (stderr)
split.cpp: In function 'std::vector<int> find_split(int, int, int, int, std::vector<int>, std::vector<int>)':
split.cpp:39:12: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
39 | for (i=0;i<p.size();++i){
| ~^~~~~~~~~
split.cpp:49:12: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
49 | for (i=0;i<v.size();++i){
| ~^~~~~~~~~
split.cpp:38:8: warning: unused variable 'j' [-Wunused-variable]
38 | int i,j,s=0;
| ^
# | 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... |