QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#370910 | #791. Mousetrap | Camillus | 0 | 147ms | 66112kb | C++20 | 1.8kb | 2024-03-29 19:10:33 | 2024-03-29 19:10:37 |
Judging History
answer
/// @author Camillus <3
#include "bits/stdc++.h"
using namespace std;
static constexpr int max_n = 2'000'000;
vector<int> g[max_n];
int parent[max_n];
int root;
int dp[max_n];
void dfs(int u, int p, int sum) {
parent[u] = p;
auto it = find(g[u].begin(), g[u].end(), p);
if (it != g[u].end()) g[u].erase(it);
for (int v : g[u]) {
dfs(v, u, sum + g[v].size() - 2);
}
if (g[u].empty()) {
dp[u] = sum;
} else {
if (g[u].size() == 1) {
dp[u] = sum;
} else {
sort(g[u].begin(), g[u].end(), [&](int u, int v) -> bool {
return dp[u] > dp[v];
});
dp[u] = dp[g[u][1]] + 1;
}
}
}
signed main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> root >> m;
for (int i = 0; i < n - 1; i++) {
int u, v;
cin >> u >> v;
g[u].push_back(v);
g[v].push_back(u);
}
dfs(root, -1, 1);
int l = -1, r = 2 * n;
while (r - l > 1) {
int x = (l + r) / 2;
int bank = 1;
int used = 0;
int u = m;
int last = -1;
while (u != root) {
for (int v : g[u]) {
if (v == last) {
continue;
}
if (dp[v] + used + 1 > x) {
bank -= 1;
used += 1;
}
}
if (bank < 0) {
break;
}
last = u;
u = parent[u];
bank += 1;
}
if (bank < 0) {
l = x;
} else {
r = x;
}
}
cout << r << '\n';
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Subtask #1:
score: 0
Wrong Answer
Test #1:
score: 0
Wrong Answer
time: 2ms
memory: 3660kb
input:
10 2 10 2 1 3 2 4 2 5 3 6 3 7 4 8 4 9 5 10 5
output:
0
result:
wrong answer 1st lines differ - expected: '2', found: '0'
Subtask #2:
score: 0
Wrong Answer
Test #11:
score: 0
Wrong Answer
time: 147ms
memory: 66112kb
input:
1000000 1 2 2 1 3 2 4 2 5 3 6 3 7 4 8 4 9 5 10 5 11 6 12 6 13 7 14 7 15 8 16 8 17 9 18 9 19 10 20 10 21 11 22 11 23 12 24 12 25 13 26 13 27 14 28 14 29 15 30 15 31 16 32 16 33 17 34 17 35 18 36 18 37 19 38 19 39 20 40 20 41 21 42 21 43 22 44 22 45 23 46 23 47 24 48 24 49 25 50 25 51 26 52 26 53 27 5...
output:
37
result:
wrong answer 1st lines differ - expected: '36', found: '37'
Subtask #3:
score: 0
Skipped
Dependency #1:
0%
Subtask #4:
score: 0
Skipped
Dependency #1:
0%