#include <bits/stdc++.h>
#include "grader.h"
using namespace std;
#define pb push_back
#define f first
#define s second
int vis[513],pos[513],ch[513];
int dfs(int n,vector<int>v[513],int idx=1){
vis[idx]=1;
int x=1;
for(int i=0;i<v[idx].size();i++){
if(!vis[v[idx][i]])
x+=dfs(n,v,v[idx][i]);
}
ch[idx]=x;
return x;
}
vector<int> dfs2(int n,vector<int>v[513],int idx=1){
vis[idx]=1;
vector<int> ans;
ans.pb(idx);
for(int i=0;i<v[idx].size();i++){
if(!vis[v[idx][i]]){
vector<int> z=dfs2(n,v,v[idx][i]);
for(int j=0;j<z.size();j++){
ans.pb(z[j]);
}
}
}
return ans;
}
int findEgg (int n, vector<pair<int,int> > edg){
memset(vis,0,sizeof vis);
memset(pos,0,sizeof pos);
vector<int> v[513];
for(int i=0;i<edg.size();i++){
v[edg[i].f].pb(edg[i].s);
v[edg[i].s].pb(edg[i].f);
pos[edg[i].f]=1;
pos[edg[i].s]=1;
}
dfs(n,v,1);
int x=1;
bool bol=1;
vector<int> cur;
while(1){
int tk[513]={0};
vector<int> h;
for(int i=1;i<=n;i++){
if(pos[i])
h.pb(i);
}
cout<<h.size()<<"\n";
for(int i=0;i<h.size();i++){
cout<<h[i]<<" ";
}
cout<<"cur: "<<cur.size()<<"\n";
for(int i=0;i<cur.size();i++){
cout<<cur[i]<<" ";
}
cout<<"\n";
cout<<"bol: "<<bol<<"\n";
if(cur.size()>=h.size()){
if(query(cur)){
cout<<"cur > h\n";
x=cur[0];
for(int i=0;i<cur.size();i++){
pos[cur[i]]=1;
}
for(int i=0;i<h.size();i++){
pos[h[i]]=0;
}
for(int i=0;i<v[cur[cur.size()-1]].size();i++){
if(v[cur[cur.size()-1]][i]==h[0]){
ch[cur[cur.size()-1]]-=ch[h[0]];
v[cur[cur.size()-1]][i]=v[cur[cur.size()-1]][v[cur[cur.size()-1]].size()-1];
cout<<"deleted: "<<v[cur[cur.size()-1]][v[cur[cur.size()-1]].size()-1]<<"\n";
v[cur[cur.size()-1]].pop_back();
cout<<"deleted: "<<v[cur[cur.size()-1]][v[cur[cur.size()-1]].size()-1]<<"\n";
}
}
}
cur.clear();
continue;
}
cout<<"\n\n";
if(h.size()==1){
if(bol==0&&query(h)==0)
for(int i=1;i<=n;i++){
for(int j=0;j<v[i].size();j++){
if(v[i][j]==x){
return i;
}
}
}
break;
}
int a=ch[x],b=1;
vector<int> va,vb;
vb.pb(x);
int zz=0;
for(int i=0;i<v[x].size();i++){
if(b+ch[v[x][i]]<a&&pos[v[x][i]]==1){
zz++;
memset(vis,0,sizeof vis);
vis[x]=1;
vector<int> z=dfs2(n,v,v[x][i]);
for(int j=0;j<z.size();j++){
vb.pb(z[j]);
tk[z[j]]=1;
}
b+=ch[v[x][i]];
a-=ch[v[x][i]];
}
}
if(zz==0){
cout<<"element "<<x<<" added to cur\n";
cur.pb(x);
pos[x]=0;
if(pos[v[x][0]])
x=v[x][0];
else
x=v[x][1];
continue;
}
for(int i=1;i<=n;i++){
if(pos[i]==1&&!tk[i]){
va.pb(i);
}
}
cout<<"va: "<<va.size()<<"\n";
for(int i=0;i<va.size();i++)
cout<<va[i]<<" ";
cout<<"\n";
cout<<"vb: "<<vb.size()<<"\n";
for(int i=0;i<vb.size();i++)
cout<<vb[i]<<" ";
cout<<"\n\n";
if(query(vb)){
bol=0;
memset(pos,0,sizeof pos);
for(int i=0;i<vb.size();i++){
pos[vb[i]]=1;
}
cout<<"zz: "<<zz<<"\n";
if(zz==1){
for(int i=0;i<v[x].size();i++){
if(pos[v[x][i]]){
pos[x]=0;
x=v[x][i];
break;
}
}
}
}
else if((bol==0&&query(va))||bol==1){
bol=1;
memset(pos,0,sizeof pos);
for(int i=0;i<va.size();i++){
pos[va[i]]=1;
}
cout<<"zz: "<<zz<<"\n";
if(x!=1)
zz++;
if(zz==v[x].size()-1){
for(int i=0;i<v[x].size();i++){
if(pos[v[x][i]]){
pos[x]=0;
x=v[x][i];
break;
}
}
}
}
else{
if(cur.size()){
for(int i=0;i<cur.size();i++){
h.pb(cur[i]);
pos[cur[i]]=1;
}
continue;
}
for(int i=1;i<=n;i++){
for(int j=0;j<v[i].size();j++){
if(v[i][j]==x){
return i;
}
}
}
}
}
return x;
}
Compilation message
eastereggs.cpp: In function 'int dfs(int, std::vector<int>*, int)':
eastereggs.cpp:13:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
13 | for(int i=0;i<v[idx].size();i++){
| ~^~~~~~~~~~~~~~
eastereggs.cpp: In function 'std::vector<int> dfs2(int, std::vector<int>*, int)':
eastereggs.cpp:24:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
24 | for(int i=0;i<v[idx].size();i++){
| ~^~~~~~~~~~~~~~
eastereggs.cpp:27:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
27 | for(int j=0;j<z.size();j++){
| ~^~~~~~~~~
eastereggs.cpp: In function 'int findEgg(int, std::vector<std::pair<int, int> >)':
eastereggs.cpp:38:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
38 | for(int i=0;i<edg.size();i++){
| ~^~~~~~~~~~~
eastereggs.cpp:56:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
56 | for(int i=0;i<h.size();i++){
| ~^~~~~~~~~
eastereggs.cpp:60:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
60 | for(int i=0;i<cur.size();i++){
| ~^~~~~~~~~~~
eastereggs.cpp:69:30: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
69 | for(int i=0;i<cur.size();i++){
| ~^~~~~~~~~~~
eastereggs.cpp:72:30: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
72 | for(int i=0;i<h.size();i++){
| ~^~~~~~~~~
eastereggs.cpp:75:30: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
75 | for(int i=0;i<v[cur[cur.size()-1]].size();i++){
| ~^~~~~~~~~~~~~~~~~~~~~~~~~~~~
eastereggs.cpp:92:34: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
92 | for(int j=0;j<v[i].size();j++){
| ~^~~~~~~~~~~~
eastereggs.cpp:104:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
104 | for(int i=0;i<v[x].size();i++){
| ~^~~~~~~~~~~~
eastereggs.cpp:110:30: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
110 | for(int j=0;j<z.size();j++){
| ~^~~~~~~~~
eastereggs.cpp:134:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
134 | for(int i=0;i<va.size();i++)
| ~^~~~~~~~~~
eastereggs.cpp:138:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
138 | for(int i=0;i<vb.size();i++)
| ~^~~~~~~~~~
eastereggs.cpp:144:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
144 | for(int i=0;i<vb.size();i++){
| ~^~~~~~~~~~
eastereggs.cpp:149:30: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
149 | for(int i=0;i<v[x].size();i++){
| ~^~~~~~~~~~~~
eastereggs.cpp:161:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
161 | for(int i=0;i<va.size();i++){
| ~^~~~~~~~~~
eastereggs.cpp:167:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
167 | if(zz==v[x].size()-1){
| ~~^~~~~~~~~~~~~~~
eastereggs.cpp:168:30: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
168 | for(int i=0;i<v[x].size();i++){
| ~^~~~~~~~~~~~
eastereggs.cpp:179:30: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
179 | for(int i=0;i<cur.size();i++){
| ~^~~~~~~~~~~
eastereggs.cpp:186:30: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
186 | for(int j=0;j<v[i].size();j++){
| ~^~~~~~~~~~~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Runtime error |
1 ms |
456 KB |
Execution killed with signal 6 |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Runtime error |
1 ms |
456 KB |
Execution killed with signal 6 |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Runtime error |
2 ms |
584 KB |
Execution killed with signal 6 |
| 2 |
Halted |
0 ms |
0 KB |
- |
