QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#425984 | #6102. Query on a Tree | HKOI0# | WA | 1ms | 3664kb | C++20 | 3.9kb | 2024-05-30 19:58:02 | 2024-05-30 19:58:04 |
Judging History
answer
#include <bits/stdc++.h>
#define int long long
#define sz(v) (int)v.size()
#define all(v) v.begin(), v.end()
using namespace std;
using ll = long long;
using pii = pair<int, int>;
using vi = vector<int>;
struct FunnyStruct {
int n;
vector<unordered_map<int, int>> g;
vector<ll> mx, mx2;
vector<int> opt;
vector<ll> dp;
ll curAns;
int bestU, bestV;
void dfs(int u, int p) {
vector<pair<ll, int>> paths;
for (auto& [v, w] : g[u]) {
dp[u] = max(dp[u], dp[v]);
paths.emplace_back(mx[v] + w, v);
}
sort(paths.rbegin(), paths.rend());
if ((int)paths.size() >= 1) {
mx[u] = paths[0].first;
opt[u] = paths[0].second;
dp[u] = max(dp[u], paths[0].first);
}
if ((int)paths.size() >= 2) {
mx2[u] = paths[1].first;
dp[u] = max(dp[u], paths[0].first + paths[1].first);
}
}
void dfs2(int u, int p) {
for (auto& [v, w] : g[u]) {
if (v == p) continue;
const ll val = dp[v] + mx[u];
cerr << "DELETE " << u + 1 << ' ' << v + 1 << ' ' << val << '\n';
if (val > curAns) {
curAns = val;
bestU = u;
bestV = v;
}
ll oldMxU = mx[u];
mx[u] = (v == opt[u] ? mx2[u] : mx[u]);
ll oldMxV = mx[v];
mx[v] = max(mx[v], mx[u] + w);
dfs2(v, u);
mx[v] = oldMxV;
mx[u] = oldMxU;
}
}
FunnyStruct(int n, const vector<unordered_map<int, int>>& e) :
n(n), g(e), mx(n), mx2(n), opt(n), dp(n) {
}
pair<int, int> solve(int u) {
dfs(u, -1);
curAns = -1;
bestU = bestV = INT_MIN;
dfs2(u, -1);
assert(bestU != -1 && bestV != -1);
return {bestU, bestV};
}
pair<int, ll> furthest(int u) {
vector<ll> dist((int)g.size(), LLONG_MAX / 4);
dist[u] = 0;
queue<int> q;
q.push(u);
while (!q.empty()) {
int v = q.front();
q.pop();
for (auto& [v2, w] : g[v]) {
if (dist[v2] > dist[v] + w) {
dist[v2] = dist[v] + w;
q.push(v2);
}
}
}
int maxDistVertex = INT_MIN;
ll maxDist = -1;
for (int v = 0; v < (int)g.size(); v++) {
if (dist[v] == LLONG_MAX / 4) continue;
if (dist[v] > maxDist) {
maxDist = dist[v];
maxDistVertex = v;
}
}
return {maxDistVertex, maxDist};
}
};
void solve() {
int n,k;
cin>>n>>k;
vector<unordered_map<int,int>> e(n);
for(int i=0;i<n-1;i++) {
int u,v,w;
cin>>u>>v>>w;
u--;
v--;
e[u].emplace(v,w);
e[v].emplace(u,w);
}
{
FunnyStruct g(n,e);
auto [u,dis1]=g.furthest(0);
auto [v,dis2]=g.furthest(u);
// cout<<dis2;
}
for(int i=1;i<=k;i++) {
FunnyStruct g(n,e);
auto [u0,v0]=g.solve(0);
int w=e[u0][v0];
e[u0].erase(v0);
e[v0].erase(u0);
cout<<u0<<' '<<v0<<' '<<w<<'\n';
FunnyStruct ng(n,e);
auto [u1,dis11]=ng.furthest(u0);
auto [u2,dis12]=ng.furthest(u1);
auto [v1,dis21]=ng.furthest(v0);
auto [v2,dis22]=ng.furthest(v1);
cout<<u1 << ' ' << v1 << ' ' << dis12<<' '<<dis22<<' '<<'\n';
// cout<<' '<<dis12+w+dis22;
e[u1].emplace(v1,w);
e[v1].emplace(u1,w);
}
cout << "END" << endl;
}
signed main() {
#ifndef LOCAL
ios::sync_with_stdio(false);
cin.tie(nullptr);
#endif
int T = 1;
// cin >> T;
while (T--) solve();
}
详细
Test #1:
score: 0
Wrong Answer
time: 1ms
memory: 3664kb
input:
7 1 2 1 3 1 5 2 7 4 6 4 7 6 1 1 2 1 2 4 1 2 3 4 5 1 2 4 6 7 6 1 2 3 4 5 6 7 1 2 3 4 5 6 7
output:
0 0 2 1 1 5 5 END
result:
wrong answer 2nd numbers differ - expected: '1', found: '0'