QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#627405 | #9230. Routing K-Codes | ucup-team1264 | WA | 95ms | 28672kb | C++20 | 4.1kb | 2024-10-10 15:49:36 | 2024-10-10 15:49:39 |
Judging History
answer
// https://www.youtube.com/watch?v=CrymicX875M
// Angel of mercy
// How did you move me
// Why am I on my feet again
#ifndef ONLINE_JUDGE
#include "templates/debug.hpp"
#else
#define debug(...)
#endif
#include <bits/stdc++.h>
using namespace std;
using i64 = int64_t;
using u64 = uint64_t;
// #define int i64
void solve() {
int n, m; cin >> n >> m;
vector<vector<int>> adj(n);
for (int e = 0; e < m; e++) {
int u, v; cin >> u >> v;
u--, v--;
adj[u].push_back(v);
adj[v].push_back(u);
}
if (m != n - 1) {
cout << "NIE\n";
return;
}
for (int u = 0; u < n; u++) {
if (adj[u].size() > 3) {
cout << "NIE\n";
return;
}
}
vector<vector<int>> sons(n);
constexpr int W = 32;
// sum when root val is 1
struct Node {
i64 sum{}, weight{};
i64 dep{};
bool valid = true;
Node grow() {
Node res = *this;
res.sum += res.weight;
res.weight *= 2;
res.dep++;
if (res.dep >= W) res.valid = false;
return res;
}
void update(i64 &ans) {
if (valid) ans = min(ans, sum);
}
};
auto from_son = [](Node a) {
Node res = a.grow();
res.sum++, res.weight++;
if (res.dep >= W) res.valid = false;
return res;
};
auto merge = [](Node a, Node b) {
Node res; res.valid = a.valid && b.valid;
res.sum += min(a.weight, b.weight);
a = a.grow(), b = b.grow();
res.sum += a.sum + b.sum;
res.weight = a.weight + b.weight;
res.sum++, res.weight++;
res.dep = max(a.dep, b.dep);
if (res.dep >= W) res.valid = false;
return res;
};
vector<Node> dp(n);
function<void(int, int)> pre_calc = [&](int u, int p) {
vector<Node> children;
for (int v : adj[u]) {
if (v == p) continue;
pre_calc(v, u);
children.push_back(dp[v]);
sons[u].push_back(v);
}
if (children.empty()) {
dp[u] = {1, 1, 1, true};
} else if (children.size() == 1) {
dp[u] = children[0].grow();
dp[u].sum++, dp[u].weight++;
} else if (children.size() == 2) {
dp[u] = merge(children[0], children[1]);
}
};
pre_calc(0, -1);
i64 ans = 1e18;
if (sons[0].size() == 1) {
dp[sons[0][0]].update(ans);
} else if (sons[0].size() == 2) {
merge(dp[sons[0][0]], dp[sons[0][1]]).update(ans);
}
function<void(int, int, Node)> reroot = [&](int u, int p, Node from_fa) {
if (p != -1) {
if (sons[u].empty()) {
from_fa.update(ans);
} else if (sons[u].size() == 1) {
merge(from_fa, dp[sons[u][0]]).update(ans);
}
}
int m = sons[u].size();
if (p == -1) {
if (m == 1) {
reroot(sons[u][0], u, {1, 1});
} else if (m == 2) {
reroot(sons[u][0], u, from_son(dp[sons[u][1]]));
reroot(sons[u][1], u, from_son(dp[sons[u][0]]));
} else if (m == 3) {
reroot(sons[u][0], u, merge(dp[sons[u][1]], dp[sons[u][2]]));
reroot(sons[u][1], u, merge(dp[sons[u][0]], dp[sons[u][2]]));
reroot(sons[u][2], u, merge(dp[sons[u][0]], dp[sons[u][1]]));
}
} else {
if (m == 1) {
reroot(sons[u][0], u, from_son(from_fa));
} else if (m == 2) {
reroot(sons[u][0], u, merge(from_fa, dp[sons[u][1]]));
reroot(sons[u][1], u, merge(from_fa, dp[sons[u][0]]));
}
}
};
reroot(0, -1, Node{0, 0});
if (ans == 1e18) cout << "NIE\n";
else cout << ans << "\n";
}
#undef int
// Make bold hypotheses and verify carefully
int main() {
cin.tie(nullptr);
ios::sync_with_stdio(false);
int t = 1;
// cin >> t;
while (t--) {
solve();
};
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 3552kb
input:
4 3 1 2 1 3 1 4
output:
6
result:
ok single line: '6'
Test #2:
score: 0
Accepted
time: 0ms
memory: 3564kb
input:
4 6 1 2 2 3 3 4 4 1 1 3 2 4
output:
NIE
result:
ok single line: 'NIE'
Test #3:
score: 0
Accepted
time: 0ms
memory: 3608kb
input:
10 10 9 3 5 10 1 4 10 8 8 3 7 3 9 6 8 6 7 2 4 5
output:
NIE
result:
ok single line: 'NIE'
Test #4:
score: 0
Accepted
time: 90ms
memory: 28624kb
input:
200000 199999 172774 188052 195063 155577 38023 148303 30605 155047 170238 19344 46835 58255 154268 109062 197059 116041 136424 12058 42062 149807 109545 115380 132322 51106 16706 162612 62234 31319 195735 80435 173898 84051 19876 102910 186924 9136 123094 117097 145054 173049 137364 152731 82662 18...
output:
NIE
result:
ok single line: 'NIE'
Test #5:
score: 0
Accepted
time: 94ms
memory: 28580kb
input:
200000 199999 168192 30733 164413 46673 175210 2329 69389 67102 33991 152553 91061 55265 166724 117726 90148 157176 34831 12213 60756 105488 121891 155192 82202 155666 102083 156661 38968 75200 190004 107321 72436 92732 64314 65004 39210 91106 70455 173568 179742 29844 126232 19552 133509 110602 131...
output:
20980030044349
result:
ok single line: '20980030044349'
Test #6:
score: 0
Accepted
time: 0ms
memory: 3616kb
input:
4 3 1 2 1 3 1 4
output:
6
result:
ok single line: '6'
Test #7:
score: 0
Accepted
time: 92ms
memory: 28396kb
input:
199021 199020 189105 111001 147300 187047 67395 102319 25317 152109 56495 115535 11656 19974 119178 6528 84571 159100 198873 156684 21161 163040 91720 165273 168305 140152 104884 119802 131 177991 35930 74934 27528 278 177640 162738 118439 69472 20365 85043 184995 54207 64542 188847 97881 167717 188...
output:
NIE
result:
ok single line: 'NIE'
Test #8:
score: 0
Accepted
time: 90ms
memory: 28128kb
input:
200000 199999 145608 31176 94180 155303 112924 85699 196865 176102 34049 30841 84924 191665 164317 97582 10102 125492 114493 20622 127660 194591 183851 21461 167689 53839 59189 126425 135853 79950 173910 4196 8134 182272 42157 63799 5109 182005 197391 154014 93467 130380 1508 66644 129681 199910 677...
output:
25146839515965
result:
ok single line: '25146839515965'
Test #9:
score: 0
Accepted
time: 95ms
memory: 28672kb
input:
200000 199999 160386 189041 24452 73297 75992 87415 87012 116413 18645 2219 151007 100916 87623 77520 51217 45179 51633 67938 183428 99891 74684 129965 186795 70345 28743 22633 107782 28087 117185 78477 46846 176763 18968 80952 118201 35872 123906 140127 65784 66684 24802 134847 42591 150517 27123 1...
output:
9894117829832
result:
ok single line: '9894117829832'
Test #10:
score: -100
Wrong Answer
time: 84ms
memory: 28592kb
input:
200000 199999 42958 26294 73743 94861 161036 525 22581 6649 152064 106483 126795 178801 147721 107972 43433 197974 75281 163319 170596 167054 180443 168322 1443 163261 197722 164837 144573 16585 6005 143774 195029 188032 112864 105465 108330 154314 138435 103667 66734 146178 15416 123293 22322 10216...
output:
NIE
result:
wrong answer 1st lines differ - expected: '28756428378750', found: 'NIE'