CSES - Datatähti Open 2021 - Results
Submission details
Task:Distances
Sender:pidddgy
Submission time:2021-01-31 21:00:10 +0200
Language:C++ (C++11)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
Test results
testverdicttimegroup
#10.01 s1, 2details
#20.03 s2details

Code

#include <bits/stdc++.h>
using namespace std;
#define cerr if(false) cerr
#define watch(x) cerr << (#x) << " is " << (x) << endl;
#define endl '\n'
#define ld long double
#define int long long
#define pii pair<int, int>
#define fi first
#define se second
#define sz(a) (int)(a).size()
#define y1 lsdjkfhshfdsighoihweogihewoghi
#define all(x) (x).begin(), (x).end()
const int maxn = 200;
int n;
int dis[maxn][maxn];
vector<int> G[maxn];
// graph split by parity
vector<int> Gpar[maxn];
void bfs(int st) {
queue<int> Q;
Q.push(st);
dis[st][st] = 0;
while(!Q.empty()) {
int S = Q.front(); Q.pop();
for(int to: G[S]) {
if(dis[st][to] > dis[st][S]+1) {
Q.push(to);
dis[st][to] = dis[st][S]+1;
}
}
}
}
bool vis[maxn];
void solve() {
for(int i = 0; i < maxn; i++) {
for(int j = 0; j < maxn; j++) {
dis[i][j] = 1e18;
}
G[i].clear();
Gpar[i].clear();
vis[i] = false;
}
cin >> n;
for(int i = 1; i <= n-1; i++) {
int u, v;
cin >> u >> v;
G[u].emplace_back(v);
G[v].emplace_back(u);
}
for(int i = 1; i <= n; i++) bfs(i);
for(int i = 1; i <= n; i++) {
for(int j = 1; j <= n; j++) {
if(dis[1][i]%2 == dis[1][j]%2 and dis[i][j] <= 3) {
Gpar[i].emplace_back(j);
Gpar[j].emplace_back(i);
}
}
}
// do all with even distance from 1
vector<int> v;
stack<int> Q;
Q.push(1);
vis[1] = true;
while(!Q.empty()) {
int S = Q.top(); Q.pop();
v.emplace_back(S);
for(int to: Gpar[S]) {
if(!vis[to]) {
vis[to] = true;
Q.push(to);
}
}
if(Q.empty()) {
for(int to: G[S]) {
if(!vis[to]) {
vis[to] = true;
Q.push(to);
}
}
}
}
for(int x: v) {
cout << x << " ";
}
cout << endl;
}
signed main() {
ios::sync_with_stdio(0);
cin.tie(0);
int t;
cin >> t;
while(t--) solve();
}
/*
*/
// Did you read the bounds?
// Did you make typos?
// Are there edge cases (N=1?)
// Are array sizes proper?
// Integer overflow?
// DS reset properly between test cases?
// Is using long longs causing TLE?
// Are you using floating points?

Test details

Test 1

Group: 1, 2

Verdict:

input
100
8
5 2
2 3
3 7
...

correct output
1 8 2 5 6 7 3 4 
1 7 2 8 3 6 4 5 
1 4 6 2 7 5 8 3 
1 8 3 2 4 7 6 5 
1 6 4 7 5 2 3 8 
...

user output
1 8 7 2 6 3 4 5 
1 7 2 6 4 5 3 8 
1 6 5 2 7 8 3 4 
1 8 7 4 2 6 5 3 
1 4 7 5 2 3 8 6 
...
Truncated

Test 2

Group: 2

Verdict:

input
100
100
37 59
81 37
44 81
...

correct output
1 99 82 81 59 5 71 55 17 24 13...

user output
1 99 82 93 47 81 98 95 50 31 8...
Truncated