#include <bits/stdc++.h>
using namespace std;
const int N = 1005;
int sz[N], pref[N];
vector<int> val, g[N];
void pre_dfs(int v, int p = -1){
sz[v] = 1;
for (int u : g[v]){
if (u == p) continue;
pre_dfs(u, v);
sz[v] += sz[u];
}
}
void dfs(int v, int p = -1){
pref[v] = 0;
for (int u : g[v]){
if (u == p) continue;
pref[v] += sz[u];
val[u] = val[v] - sz[v] + pref[v];
dfs(u, v);
}
}
vector<int> label(int n, int k, vector<int> u, vector<int> v){
for (int i = 0; i <= n; i ++)
pref[i] = sz[i] = 0, g[i].clear();
val.resize(n, 0);
for (int i = 0; i < n - 1; i ++){
g[u[i]].push_back(v[i]);
g[v[i]].push_back(u[i]);
}
pre_dfs(0);
val[0] = sz[0];
dfs(0);
val.push_back(0);
return val;
}
int find_next_station(int s, int t, vector<int> c){
// cout << s << " ?? " << t << endl;
for (int x : c){
// cout << "cmp " << x << endl;
if (t <= x){
// cout << "goto " << x << endl;
return x;
}
}
// cout << "goto " << c.back() << endl;
return c.back();
}
// int main(){
// int t;
// cin >> t;
// while (t--){
// int n, k;
// cin >> n >> k;
// vector<int> u(n - 1), v(n - 1);
// for (int i = 0; i < n - 1; i ++)
// cin >> u[i] >> v[i];
// label(n, k, u, v);
// for (int i = 0; i < n; i ++){
// cout << i << " :: " << val[i] << endl;
// }
// for (int i = 0; i < n; i ++){
// for (int j = 0; j < n; j ++){
// if (i == j) continue;
// vector<int> adj;
// for (int x : g[i])
// adj.push_back(val[x]);
// sort(adj.begin(), adj.end());
// int res = find_next_station(val[i], val[j], adj);
// cout << val[i] << " -- " << val[j] << " == " << res << endl;
// }
// }
// }
// }
/*
10 10
0 1
1 2
2 3
3 4
4 5
5 6
6 7
7 8
8 9
*/
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
0 ms |
344 KB |
Invalid length of array as the response of 'label'. scenario=0, n=10, len=11 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
344 KB |
Invalid length of array as the response of 'label'. scenario=0, n=996, len=997 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
344 KB |
Invalid length of array as the response of 'label'. scenario=0, n=2, len=3 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
344 KB |
Invalid length of array as the response of 'label'. scenario=0, n=2, len=3 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
0 ms |
344 KB |
Invalid length of array as the response of 'label'. scenario=0, n=3, len=4 |
2 |
Halted |
0 ms |
0 KB |
- |