QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#470940 | #6396. Puzzle: Kusabi | UESTC_OldEastWest# | WA | 29ms | 11968kb | C++20 | 3.0kb | 2024-07-10 16:59:02 | 2024-07-10 16:59:02 |
Judging History
answer
#pragma GCC optimize("Ofast")
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int M = 1e5 + 5;
const int mod = 998244353;
int n, k, a[M], opt[M], dep[M];
vector<int> g[M];
vector<pair<int, int>> ans;
int dfs(int x, int fa) {
dep[x] = dep[fa] + 1;
vector<int> chang, duan, tong;
if (opt[x] == 1) chang.push_back(x);
if (opt[x] == 2) duan.push_back(x);
if (opt[x] == 3) tong.push_back(x);
for (int y : g[x]) {
if (y == fa) continue;
int t = dfs(y, x);
if (t == -1) continue;
if (opt[t] == 1) chang.push_back(t);
if (opt[t] == 2) duan.push_back(t);
if (opt[t] == 3) tong.push_back(t);
}
sort(chang.begin(), chang.end(), [](const int&x, const int&y) {
return dep[x] < dep[y];
});
sort(duan.begin(), duan.end(), [](const int&x, const int&y) {
return dep[x] > dep[y];
});
sort(tong.begin(), tong.end(), [](const int&x, const int&y) {
return dep[x] > dep[y];
});
int node = -1;
int cnt = 0;
for (int i = 0; i < (int)tong.size(); i += 2) {
if (i == (int)tong.size() - 1) {
cnt ++;
node = tong[i];
i-=1;
break;
}
if (dep[tong[i]] == dep[tong[i + 1]]) {
ans.push_back({tong[i], tong[i + 1]});
} else {
cnt ++;
node = tong[i];
i -= 1;
}
}
if (cnt > 1) {
cout << "NO" << endl;
exit(0);
}
if (chang.size() > duan.size()) {
if (chang.size() != duan.size() + 1 || cnt) {
cout << "NO" << endl;
exit(0);
}
int i = 0, j = 0;
for (;j < duan.size(); j++, i++) {
ans.push_back({chang[i], duan[i]});
}
return chang.back();
} else if (chang.size() < duan.size()) {
if (duan.size() != chang.size() + 1 || cnt) {
cout << "NO" << endl;
exit(0);
}
int i = 0, j = 0;
for (;j < chang.size(); j++, i++) {
ans.push_back({chang[i], duan[i]});
}
return duan.back();
} else {
int i = 0, j = 0;
for (;j < chang.size(); j++, i++) {
ans.push_back({chang[i], duan[i]});
}
}
if (cnt) {
return node;
}
return -1;
}
signed main() {
ios::sync_with_stdio(0);
cin.tie(0);cout.tie(0);
cin >> n;
int cnt = 1;
for (int i = 1; i < n; i++) {
int x, y; string ts;
cin >> x >> y >> ts;
g[x].push_back(y);
g[y].push_back(x);
if (ts[0] == 'C') opt[x] = 1;
else if (ts[0] == 'D') opt[x] = 2;
else if (ts[0] == 'T') opt[x] = 3;
else cnt++;
}
if (n - cnt & 1) {
cout << "NO" << endl;
return 0;
}
dfs(1, 0);
cout << "YES" << endl;
for (auto [x, y] : ans) {
cout << x << " " << y << endl;
}
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 5628kb
input:
8 2 1 - 3 1 - 4 2 Tong 5 2 Tong 6 3 Duan 7 3 - 8 7 Chang
output:
YES 4 5 8 6
result:
ok Correct.
Test #2:
score: 0
Accepted
time: 1ms
memory: 5580kb
input:
10 2 1 Duan 3 2 Duan 4 2 - 5 4 Chang 6 2 Chang 7 1 Duan 8 6 Tong 9 6 Tong 10 3 Chang
output:
YES 10 3 8 9 6 2 5 7
result:
ok Correct.
Test #3:
score: 0
Accepted
time: 1ms
memory: 5884kb
input:
2 2 1 Tong
output:
NO
result:
ok Correct.
Test #4:
score: -100
Wrong Answer
time: 29ms
memory: 11968kb
input:
100000 2 1 Duan 3 1 Duan 4 3 - 5 4 Duan 6 3 - 7 4 Duan 8 4 - 9 8 - 10 7 Duan 11 9 - 12 7 Duan 13 7 Duan 14 8 Duan 15 13 - 16 10 Duan 17 11 Duan 18 12 - 19 1 Duan 20 5 Duan 21 4 Duan 22 14 Duan 23 16 - 24 22 Duan 25 16 Duan 26 13 - 27 13 - 28 17 - 29 5 Duan 30 22 - 31 23 - 32 9 Duan 33 5 - 34 30 Duan...
output:
YES 46666 56115 88119 38802 93143 10006 11412 76473 42604 15988 6569 30148 31531 2008 64525 46703 71289 70618 81204 47748 42353 46131 97634 20514 83516 82155 66867 62410 15712 9975 4978 3205 83026 5622 48783 10902 82167 30893 93809 65878 33377 66951 94549 66936 79128 64411 8453 22475 54702 36857 629...
result:
wrong answer Pair 11412-76473 symbol not satisfied.