题解 P3894【[GDOI2014] 传送】
难倒不难,128MB 的空间限制快恶心死我了。
我们设 表示到节点 距离最近的叶子的距离,这个可以很容易换根 DP 求出。设 表示树 中距离最近的两个叶子的距离。设 ()表示树中两个节点 和 之间的距离。
(一)若 ,不妨设 。
显然,在第 棵树上,从 花费 的代价走到最近的叶子是最优的。第 棵树同理。对于 范围内的树 ,我们会花费 的代价选择最近的两个叶子经过。在两棵树之间传送时,还会产生 的代价。
因此,这种情况的答案为:
(二)若 。
容易想到树上两点间距离。但这是错的,还可能从 走到叶子,传送到相邻的某棵树,经过最近的两个叶子,再传送回来走到 。
因此,这种情况的答案为:
你可能会有疑问:万一离 和 最近的叶子是同一个,怎么办呢?容易发现,这种情况下从 和 分别走到叶子的路径拼起来已经是 了,但还需要经过更多的边,这种情况自然更劣,不会作为 保留。
时间复杂度 。
本题特别卡空间,建议使用树剖 LCA 而不是倍增 LCA,数组能不开 long long 就不要开。
// Problem: P3894 [GDOI2014] 传送
// Contest: Luogu
// URL: https://www.luogu.com.cn/problem/P3894
// Memory Limit: 125 MB
// Time Limit: 1000 ms
//
// Powered by CP Editor (https://cpeditor.org)
//By: OIer rui_er
#include <bits/stdc++.h>
#define rep(x, y, z) for(int x = (y); x <= (z); ++x)
#define per(x, y, z) for(int x = (y); x >= (z); --x)
#define debug(format...) fprintf(stderr, format)
#define fileIO(s) do {freopen(s".in", "r", stdin); freopen(s".out", "w", stdout);} while(false)
#define endl '\n'
using namespace std;
typedef long long ll;
mt19937 rnd(std::chrono::duration_cast<std::chrono::nanoseconds>(std::chrono::system_clock::now().time_since_epoch()).count());
int randint(int L, int R) {
uniform_int_distribution<int> dist(L, R);
return dist(rnd);
}
template<typename T> void chkmin(T& x, T y) {if(x > y) x = y;}
template<typename T> void chkmax(T& x, T y) {if(x < y) x = y;}
const int N = 3.5e5 + 5, M = 1e6 + 5, inf = 1e9 + 10;
const ll Inf = 2e12;
int n, m[N], q, L[N], R[N], dis[M], fa[M], sz[M], son[M], top[M], val[M], dp1[M], dp2[M];
ll near[N];
vector<tuple<int, int>> e[M];
void dfs1(int u, int f, int id) {
fa[u] = f;
sz[u] = 1;
for(auto [v, w] : e[u]) {
if(v != f) {
dis[v] = dis[u] + 1;
val[v] = val[u] + w;
dfs1(v, u, id);
chkmin(near[id], 1LL * dp1[u] + dp1[v] + w);
chkmin(dp1[u], dp1[v] + w);
sz[u] += sz[v];
if(sz[v] > sz[son[u]]) son[u] = v;
}
}
if(!son[u]) dp1[u] = 0;
}
void dfs2(int u, int f, int tp) {
top[u] = tp;
if(!son[u]) return;
for(auto [v, w] : e[u]) {
if(v != f) {
dp2[v] = min(dp1[v], dp2[u] + w);
dfs2(v, u, v == son[u] ? tp : v);
}
}
}
int LCA(int u, int v) {
while(top[u] != top[v]) {
if(dis[top[u]] < dis[top[v]]) swap(u, v);
u = fa[top[u]];
}
if(dis[u] < dis[v]) swap(u, v);
return v;
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
cin >> n;
rep(i, 1, n) {
cin >> m[i];
L[i] = R[i - 1] + 1;
R[i] = L[i] + m[i] - 1;
rep(j, 1, m[i] - 1) {
int u, v, w;
cin >> u >> v >> w;
e[L[i] + u].emplace_back(L[i] + v, w);
e[L[i] + v].emplace_back(L[i] + u, w);
}
near[i] = Inf;
rep(j, L[i], R[i]) dp1[j] = inf;
dis[L[i]] = 1; val[L[i]] = 0;
dfs1(L[i], 0, i);
if(near[i] >= inf) near[i] = Inf;
near[i] += near[i - 1];
dp2[L[i]] = dp1[L[i]];
dfs2(L[i], 0, L[i]);
}
for(cin >> q; q; --q) {
int x, u, y, v;
cin >> x >> u >> y >> v;
++x; ++y;
if(x > y) swap(x, y), swap(u, v);
u = L[x] + u;
v = L[y] + v;
if(x == y) {
int p = LCA(u, v);
ll ans = val[u] + val[v] - 2 * val[p];
if(x > 1 && near[x - 1] - near[x - 2] < Inf) chkmin(ans, dp2[u] + dp2[v] + (near[x - 1] - near[x - 2]) + 2);
if(x < n && near[x + 1] - near[x] < Inf) chkmin(ans, dp2[u] + dp2[v] + (near[x + 1] - near[x]) + 2);
cout << ans << endl;
}
else {
ll ans = dp2[u] + (near[y - 1] - near[x]) + dp2[v] + (y - x);
cout << (ans >= Inf ? -1 : ans) << endl;
}
}
return 0;
}
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 25岁的心里话
· 按钮权限的设计及实现