#include "stations.h"
#include <bits/stdc++.h>
using namespace std;
const int N=1e3+10;
int tmpcase1,cnt=0,root;
vector<int> adj[N],lb;
/*case1
void labelcase1(int u,int p){
lb[u]=cnt++;
for(int i=0;i<adj[u].size();i++){
if(adj[u][i]!=p) labelcase1(adj[u][i],u);
}
}*/
void labelcase3(int u,int p){
lb[u]=cnt;
bool next=false;
for(int i=0;i<adj[u].size();i++){
if(adj[u][i]!=p){
next=true;
cnt++;
labelcase3(adj[u][i],u);
}
}
if(!next) cnt+=999;
}
vector<int> label(int n, int k, vector<int> u, vector<int> v) {
lb.resize(n);
cnt=0;
for(int i=0;i<n-1;i++){
adj[u[i]].push_back(v[i]);
adj[v[i]].push_back(u[i]);
}
/*case1
for(int i=0;i<n;i++){
if(adj[i].size()==1) tmpcase1=i;
}
labelcase1(tmpcase1,-1);*/
/*case2
for(int i=0;i<n;i++) lb[i]=i;*/
for(int i=0;i<n;i++){
if(adj[i].size()>2){
root=i;
break;
}
}
labelcase3(root,-1);
for(int i=0;i<n;i++) adj[i].clear();
/*for(int i=0;i<lb.size();i++) cout<<lb[i] <<" ";
cout<<"\n\n\n";*/
return lb;
}
int find_next_station(int s, int t, vector<int> c) {
/*case1
if(c.size()==1) return c[0];
if(s<t) return max(c[0],c[1]);
return min(c[0],c[1]);*/
/*case2
int cnts=0,cntt=0,tmps=s+1,tmpt=t+1;
s++,t++;
while(tmps!=0) cnts++,tmps/=2;
while(tmpt!=0) cntt++,tmpt/=2;
if(cnts>cntt) return (s/2)-1;
if(cntt>cnts){
while(true){
if(t==s*2) return s*2-1;
if(t==s*2+1) return s*2;
if(t==s || t<s) return s/2-1;
t/=2;
}
}
return s/2-1;*/
int ind;
if(c.size()==1) return c[0];
if(c.size()>2){
ind=upper_bound(c.begin(),c.end(),t)-c.begin()-1;
return c[ind];
}
if(abs(t-s)>=1000 || t<s) return c[0];
return c[1];
}
Compilation message
stations.cpp: In function 'void labelcase3(int, int)':
stations.cpp:17:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
17 | for(int i=0;i<adj[u].size();i++){
| ~^~~~~~~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
344 KB |
Invalid labels (values out of range). scenario=0, k=1000, vertex=1, label=1005 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
3 ms |
344 KB |
Invalid labels (values out of range). scenario=0, k=1000, vertex=3, label=242244 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
380 ms |
684 KB |
Wrong query response. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
650 ms |
688 KB |
Output is correct |
2 |
Incorrect |
487 ms |
936 KB |
Wrong query response. |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
371 ms |
684 KB |
Wrong query response. |
2 |
Halted |
0 ms |
0 KB |
- |