#include <bits/stdc++.h>
#include "sphinx.h"
using namespace std;
int pr2[251];
int find2(int x){
if(x==pr2[x])return x;
return pr2[x] = find2(pr2[x]);
}
void mergegroup2(int a,int b){
a = find2(a);
b = find2(b);
if(a==b)return ;
pr2[a] = b;
}
int pr[251];
vector<int> adjj[251],nadj[251];
int find(int x){
if(x==pr[x])return x;
return pr[x] = find(pr[x]);
}
bool mergegroup(int a,int b){
a = find(a);
b = find(b);
if(a==b)return 0;
pr[a] = b;
return 1;
}
vector<int> even,odd;
int VIS[251];
void dfs(int i,int dep){
VIS[i] =1 ;
if(dep)even.push_back(i);
else odd.push_back(i);
for(auto j:nadj[i]){
if(!VIS[j])dfs(j,dep^1);
}
}
vector<int> find_colours(int n,vector<int> X,vector<int> Y){
for(int i = 0;i<X.size();i++){
adjj[X[i]].push_back(Y[i]);
adjj[Y[i]].push_back(X[i]);
}
int d = 0;
for(int i = 0;i<n;i++){
pr2[i] = i;
}
for(int i = n-1;i>=0;i--){
int ans = 0;
while(ans!=-1){
vector<int> v;
int vis[n] = {0};
for(auto j:adjj[i]){
if(j>i&&vis[find2(j)]==0&&find2(j)!=find2(i)){
v.push_back(j);
vis[find2(j)] = 1;
}
}
int sz = v.size();
int l = 0 , r = sz-1;
ans = -1;
while(l<=r){
int mid = (l+r)/2;
vector<int> e(n,n);
e[i] = -1;
for(int j = 0;j<=mid;j++){
e[v[j]] = -1;
}
int cnt = 0;
for(int j = 0;j<n;j++){
if(e[j]==-1){
pr[j] = j;cnt++;
}else {
pr[j] = j;cnt++;
}
}
for(int j = 0;j<X.size();j++){
if(e[X[j]]==-1&&e[Y[j]]==-1){
if(X[j]==i||Y[j]==i)continue;
if(find2(X[j])==find2(Y[j]))cnt-=mergegroup(X[j],Y[j]);
}else if(e[X[j]]==n&&e[Y[j]]==n){
cnt-=mergegroup(X[j],Y[j]);
}
}
if(cnt==perform_experiment(e)){
l = mid+1;
}else {
ans = mid;
r = mid-1;
}
}
if(ans!=-1){
mergegroup2(i,v[ans]);
}
}
}
vector<int> V[n];
for(int i = 0;i<n;i++){
V[find2(i)].push_back(i);
}
for(int i = 0;i<X.size();i++){
if(find2(X[i])!=find2(Y[i])){
nadj[find2(X[i])].push_back(find2(Y[i]));
nadj[find2(Y[i])].push_back(find2(X[i]));
}
}
dfs(find2(0),0);
if(even.size()+odd.size()==1){
for(int i = 0;i<n;i++){
vector<int> e(n,i);
e[0] = -1;
if(perform_experiment(e)==1){
vector<int> na(n,i);
return na;
}
}
}
int in[n] = {0};
for(int i = 0;i<n;i++){
{
vector<int> lol;
for(auto j:even){
if(in[j]==0){
lol.push_back(j);
}
}
vector<int> ee(n,i);
for(auto j:lol){
for(auto w:V[j]){
ee[w] = -1;
}
}
int qw = lol.size()+1-perform_experiment(ee);
int ans = 0;
while(qw--){
int sz = lol.size();
int l = ans , r = sz-1;
int org = ans;
ans = -1;
while(l<=r){
int mid = (l+r)/2;
vector<int> e(n,i);
int cnt = 0;
for(int j = 0;j<n;j++){
if(e[j]==-1){
pr[j] = j;cnt++;
}else {
pr[j] = j;cnt++;
}
}
for(int j = org;j<=mid;j++){
for(auto w:V[lol[j]]){
e[w] = -1;
}
}
for(int j = 0;j<X.size();j++){
if(e[X[j]]==-1&&e[Y[j]]==-1){
if(find2(X[j])==find2(Y[j]))cnt-=mergegroup(X[j],Y[j]);
}else if(e[X[j]]==i&&e[Y[j]]==i){
cnt-=mergegroup(X[j],Y[j]);
}
}
if(cnt==perform_experiment(e)){
l = mid+1;
}else {
ans = mid;
r = mid-1;
}
}
if(ans==-1){
break;
}
in[lol[ans]] = i+1;
ans++;
}
}
swap(even,odd);
{
vector<int> lol;
for(auto j:even){
if(in[j]==0){
lol.push_back(j);
}
}
vector<int> ee(n,i);
for(auto j:lol){
for(auto w:V[j]){
ee[w] = -1;
}
}
int qw = lol.size()+1-perform_experiment(ee);
int ans = 0;
while(qw--){
int sz = lol.size();
int l = ans , r = sz-1;
int org = ans;
ans = -1;
while(l<=r){
int mid = (l+r)/2;
vector<int> e(n,i);
int cnt = 0;
for(int j = 0;j<n;j++){
if(e[j]==-1){
pr[j] = j;cnt++;
}else {
pr[j] = j;cnt++;
}
}
for(int j = org;j<=mid;j++){
for(auto w:V[lol[j]]){
e[w] = -1;
}
}
for(int j = 0;j<X.size();j++){
if(e[X[j]]==-1&&e[Y[j]]==-1){
if(find2(X[j])==find2(Y[j]))cnt-=mergegroup(X[j],Y[j]);
}else if(e[X[j]]==i&&e[Y[j]]==i){
cnt-=mergegroup(X[j],Y[j]);
}
}
if(cnt==perform_experiment(e)){
l = mid+1;
}else {
ans = mid;
r = mid-1;
}
}
if(ans==-1){
break;
}
in[lol[ans]] = i+1;
ans++;
}
}
}
vector<int> col(n,0);
for(int i = 0;i<n;i++){
col[i] = in[find2(i)]-1;
}
return col;
}
Compilation message
sphinx.cpp: In function 'std::vector<int> find_colours(int, std::vector<int>, std::vector<int>)':
sphinx.cpp:40:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
40 | for(int i = 0;i<X.size();i++){
| ~^~~~~~~~~
sphinx.cpp:77:28: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
77 | for(int j = 0;j<X.size();j++){
| ~^~~~~~~~~
sphinx.cpp:101:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
101 | for(int i = 0;i<X.size();i++){
| ~^~~~~~~~~
sphinx.cpp:156:36: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
156 | for(int j = 0;j<X.size();j++){
| ~^~~~~~~~~
sphinx.cpp:214:36: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
214 | for(int j = 0;j<X.size();j++){
| ~^~~~~~~~~
sphinx.cpp:44:9: warning: unused variable 'd' [-Wunused-variable]
44 | int d = 0;
| ^
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
1 ms |
336 KB |
#experiments: 16 |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
1 ms |
336 KB |
#experiments: 2 |
| 2 |
Correct |
1 ms |
336 KB |
#experiments: 7 |
| 3 |
Correct |
1 ms |
336 KB |
#experiments: 7 |
| 4 |
Correct |
1 ms |
336 KB |
#experiments: 3 |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
1 ms |
336 KB |
#experiments: 16 |
| 2 |
Correct |
1 ms |
336 KB |
#experiments: 2 |
| 3 |
Correct |
1 ms |
336 KB |
#experiments: 7 |
| 4 |
Correct |
1 ms |
336 KB |
#experiments: 7 |
| 5 |
Correct |
1 ms |
336 KB |
#experiments: 3 |
| 6 |
Correct |
1 ms |
336 KB |
#experiments: 74 |
| 7 |
Correct |
2 ms |
336 KB |
#experiments: 154 |
| 8 |
Incorrect |
3 ms |
336 KB |
Invalid value of G[4]: -1 |
| 9 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
1 ms |
336 KB |
#experiments: 2 |
| 2 |
Correct |
1 ms |
336 KB |
#experiments: 7 |
| 3 |
Correct |
1 ms |
336 KB |
#experiments: 7 |
| 4 |
Correct |
1 ms |
336 KB |
#experiments: 3 |
| 5 |
Correct |
1 ms |
336 KB |
#experiments: 74 |
| 6 |
Correct |
2 ms |
336 KB |
#experiments: 154 |
| 7 |
Incorrect |
3 ms |
336 KB |
Invalid value of G[4]: -1 |
| 8 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
1 ms |
336 KB |
#experiments: 2 |
| 2 |
Correct |
1 ms |
336 KB |
#experiments: 7 |
| 3 |
Correct |
1 ms |
336 KB |
#experiments: 7 |
| 4 |
Correct |
1 ms |
336 KB |
#experiments: 3 |
| 5 |
Correct |
2 ms |
336 KB |
#experiments: 64 |
| 6 |
Correct |
8 ms |
592 KB |
#experiments: 358 |
| 7 |
Correct |
6 ms |
496 KB |
#experiments: 371 |
| 8 |
Correct |
6 ms |
336 KB |
#experiments: 417 |
| 9 |
Correct |
7 ms |
336 KB |
#experiments: 464 |
| 10 |
Correct |
7 ms |
336 KB |
#experiments: 478 |
| 11 |
Correct |
9 ms |
496 KB |
#experiments: 489 |
| 12 |
Correct |
10 ms |
604 KB |
#experiments: 513 |
| 13 |
Correct |
116 ms |
1352 KB |
#experiments: 1105 |
| 14 |
Correct |
236 ms |
1716 KB |
#experiments: 1933 |
| 15 |
Correct |
313 ms |
1456 KB |
#experiments: 2442 |
| 16 |
Incorrect |
390 ms |
1460 KB |
#experiments reached 2751 |
| 17 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
1 ms |
336 KB |
#experiments: 16 |
| 2 |
Correct |
1 ms |
336 KB |
#experiments: 2 |
| 3 |
Correct |
1 ms |
336 KB |
#experiments: 7 |
| 4 |
Correct |
1 ms |
336 KB |
#experiments: 7 |
| 5 |
Correct |
1 ms |
336 KB |
#experiments: 3 |
| 6 |
Correct |
1 ms |
336 KB |
#experiments: 74 |
| 7 |
Correct |
2 ms |
336 KB |
#experiments: 154 |
| 8 |
Incorrect |
3 ms |
336 KB |
Invalid value of G[4]: -1 |
| 9 |
Halted |
0 ms |
0 KB |
- |
