#include "stations.h"
#include <vector>
#include <bits/stdc++.h>
using namespace std;
const int N = 2e3;
int cnt = 0;
int dfs(int i,int pre,vector<int> &ret,vector<int> adj[]){
int mx = cnt;
ret[i] = cnt++;
for(int j=0;j<adj[i].size();j++){
int v = adj[i][j];
if(v==pre)continue;
mx = max(mx,dfs(v,i,ret,adj));
}
ret[i] += mx*1000;
return mx;
}
std::vector<int> label(int n, int k, std::vector<int> u, std::vector<int> v) {
cnt = 0;
vector<int>adj[N];
for(int i=0;i<n-1;i++){
adj[u[i]].push_back(v[i]);
adj[v[i]].push_back(u[i]);
}
vector<int> ret = vector<int>(n,0);
dfs(0,0,ret,adj);
return ret;
}
int get_val(int x,int l,int r){
int ret = 0;
for(int i=l;i<=r;i++)
ret |= (1<<(i-l))*(((1<<i)&x)>0);
return ret;
}
bool cmp(int x,int y){
return get_val(x,0,9)<get_val(y,0,9);
}
int find_next_station(int s, int t, std::vector<int> c) {
int s1 = get_val(s,0,9); int t1 = get_val(t,0,9);
sort(c.begin(),c.end(),cmp);
if(t1<s1)
return c[0];
vector<int>vec;
for(int i=1;i<c.size();i++)
vec.push_back(get_val(c[i],0,9));
vec.push_back(c.back()/1000+1);
for(int i=0;i<(int)vec.size()-1;i++)
if(t1>=vec[i]&&t1<vec[i+1])
return c[i+1];
return c[0];
}
Compilation message
stations.cpp: In function 'int dfs(int, int, std::vector<int>&, std::vector<int>*)':
stations.cpp:15:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
15 | for(int j=0;j<adj[i].size();j++){
| ~^~~~~~~~~~~~~~
stations.cpp: In function 'int find_next_station(int, int, std::vector<int>)':
stations.cpp:60:15: 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=1;i<c.size();i++)
| ~^~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
3 ms |
412 KB |
Invalid labels (values out of range). scenario=0, k=1000, vertex=0, label=9000 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
5 ms |
372 KB |
Invalid labels (values out of range). scenario=0, k=1000, vertex=0, label=995000 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
534 ms |
520 KB |
Wrong query response. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
950 ms |
520 KB |
Output is correct |
2 |
Incorrect |
801 ms |
528 KB |
Wrong query response. |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
561 ms |
708 KB |
Wrong query response. |
2 |
Halted |
0 ms |
0 KB |
- |