QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#168750 | #7103. Red Black Tree | Fu_su | WA | 3ms | 5492kb | C++17 | 5.0kb | 2023-09-08 21:08:41 | 2023-09-08 21:08:41 |
Judging History
answer
#include <set>
#include <cmath>
#include <queue>
#include <vector>
#include <iostream>
#include <algorithm>
#define int long long
std::vector<std::vector<std::pair<int, long long>>> E;
std::vector<char> vis;
signed main() {
std::ios::sync_with_stdio(false);
std::cin.tie(nullptr);
int T;
E.resize(1e5 + 5);
vis.resize(1e5 + 5);
for (std::cin >> T; T; --T) {
int n, m, q;
std::cin >> n >> m >> q;
std::vector<int> col(n + 1);
for (int r, i = 1; i <= m; ++i) {
std::cin >> r; col[r] = 1;
}
std::vector<std::vector<std::pair<int, int>>> e(n + 1);
for (int i = 1, u, v, w; i < n; ++i) {
std::cin >> u >> v >> w;
e[u].push_back({v, w});
e[v].push_back({u, w});
}
int vistime = 0;
std::vector<int> anc(n + 1), stk, dfn(n + 1), euler, lastpos(n + 1), height(n + 1);
std::vector<long long> sum(n + 1), ans(n + 1);
auto dfs = [&](auto &&self, const int u) -> void {
if (col[u] == 1) stk.push_back(u);
dfn[u] = ++vistime;
anc[u] = stk.back();
ans[u] = sum[u] - sum[anc[u]];
euler.push_back(u);
lastpos[u] = euler.size() - 1;
for (auto [v, w] : e[u]) if (anc[v] == 0) {
height[v] = height[u] + 1;
sum[v] = sum[u] + w;
self(self, v);
euler.push_back(u);
lastpos[u] = euler.size() - 1;
}
if (col[u] == 1) stk.pop_back();
};
dfs(dfs, 1);
auto minNode = [&](const int u, const int v) {
if (height[u] < height[v]) return u;
else return v;
};
std::vector ST(18, std::vector<int>(2 * (n + 3)));
for (int i = 0; i < euler.size(); ++i) ST[0][i] = euler[i];
for (int i = 0; i < 17; ++i) {
for (int l = 0; l < euler.size(); ++l) {
int r = l + (1 << (i + 1)) - 1;
if (r >= euler.size()) break;
int mid = l + (1 << (i));
ST[i + 1][l] = minNode(ST[i][l], ST[i][mid]);
}
}
auto LCA = [&](int u, int v) {
int x = lastpos[u], y = lastpos[v];
if (x > y) std::swap(x, y);
int len = y - x + 1;
return minNode(ST[log2(len)][x], ST[log2(len)][y - (1 << signed(log2(len))) + 1]);
};
auto dist = [&](int u, int v) {
if (height[u] < height[v]) std::swap(u, v);
return sum[u] - sum[v];
};
auto build_virtual_tree = [&](std::vector<int> &h, auto &e) {
std::vector<int> a;
std::sort(h.begin(), h.end(), [&](int a, int b) { return dfn[a] < dfn[b]; });
for (int i = 0; i < h.size(); ++i) {
a.push_back(h[i]);
if (i + 1 != h.size()) {
a.push_back(LCA(h[i], h[i + 1]));
}
}
std::sort(a.begin(), a.end(), [&](int a, int b) { return dfn[a] < dfn[b]; });
a.erase(std::unique(a.begin(), a.end()), a.end());
std::vector<int> res;
for (int i = 0; i < a.size() - 1; ++i) {
int lc = LCA(a[i], a[i + 1]);
e[lc].push_back({a[i + 1], dist(a[i + 1], lc)});
e[a[i + 1]].push_back({lc, dist(a[i + 1], lc)});
res.push_back(lc);
res.push_back({a[i + 1]});
}
std::sort(res.begin(), res.end());
res.erase(std::unique(res.begin(), res.end()), res.end());
return res;
};
std::vector<int> iskey(n + 1);
for (int k, x; q; --q) {
std::vector<int> h;
std::multiset<long long> s;
long long res = 0;
for (std::cin >> k; k; --k) {
std::cin >> x;
h.push_back(x);
iskey[x] = 1;
res = std::max(res, ans[x]);
}
int ances = 0;
long long maxdep = 0;
std::vector<int> keyNodes;
for (auto u : h) if (ans[u] == res) {
maxdep = sum[u];
keyNodes.push_back(u);
if (ances == 0) ances = anc[u];
else if (anc[u] != ances) ances = -1;
}
if (ances == -1) {
for (auto u : h) iskey[u] = 0;
std::cout << res << '\n';
continue;
}
for (auto u : h) if (anc[u] == ances) {
s.insert(ans[u]);
}
k = h.size();
for (int i = 0; i < k; ++i) h.push_back(anc[h[i]]);
auto nodes = build_virtual_tree(h, E);
std::sort(keyNodes.begin(), keyNodes.end(), [&](int a, int b) { return dfn[a] < dfn[b]; });
int lca = keyNodes[0];
for (int i = 1; i < keyNodes.size(); ++i) lca = LCA(keyNodes[i], lca);
auto dfs2 = [&](auto &&self, int u) -> void {
vis[u] = 1;
if (iskey[u] && anc[u] == ances) s.erase(s.find(ans[u]));
for (auto [v, w] : E[u]) if (height[v] > height[u] && vis[v] == 0) {
self(self, v);
}
};
while (lca != ances) {
dfs2(dfs2, lca);
res = std::min(res, std::max(maxdep - sum[lca], (s.size() ? *s.rbegin() : 0ll)));
for (auto [v, w] : E[lca]) if (height[v] < height[lca]) {
lca = v; break;
}
}
for (auto u : nodes) E[u].clear();
std::cout << res << '\n';
for (auto u : nodes) iskey[u] = vis[u] = 0;
}
}
}
詳細信息
Test #1:
score: 0
Wrong Answer
time: 3ms
memory: 5492kb
input:
2 12 2 4 1 9 1 2 1 2 3 4 3 4 3 3 5 2 2 6 2 6 7 1 6 8 2 2 9 5 9 10 2 9 11 3 1 12 10 3 3 7 8 4 4 5 7 8 4 7 8 10 11 3 4 5 12 3 2 3 1 2 1 2 1 1 3 1 1 1 2 1 2 3 1 2 3
output:
4 5 2 8 0 0 0
result:
wrong answer 3rd lines differ - expected: '3', found: '2'