题解 P3894【[GDOI2014] 传送】

难倒不难,128MB 的空间限制快恶心死我了。

我们设 du0,u1 表示到节点 (u0,u1) 距离最近的叶子的距离,这个可以很容易换根 DP 求出。设 pu0 表示树 u0 中距离最近的两个叶子的距离。设 dis(u0,u1,v0,v1)u0=v0)表示树中两个节点 u1v1 之间的距离。

(一)若 s0e0,不妨设 s0<e0

显然,在第 s0 棵树上,从 (s0,s1) 花费 ds0,s1 的代价走到最近的叶子是最优的。第 e0 棵树同理。对于 [s0+1,e01] 范围内的树 i,我们会花费 pi 的代价选择最近的两个叶子经过。在两棵树之间传送时,还会产生 1 的代价。

因此,这种情况的答案为:

ans=ds0,s1+i=s0+1e01pi+de0,e1+(e0s0)

(二)若 s0=e0

容易想到树上两点间距离。但这是错的,还可能从 s0 走到叶子,传送到相邻的某棵树,经过最近的两个叶子,再传送回来走到 e0

因此,这种情况的答案为:

ans=min{dis(s0,s1,e0,e1),ds0,s1+ps01+de0,e1+2,ds0,s1+ps0+1+de0,e1+2}

你可能会有疑问:万一离 (s0,s1)(e0,e1) 最近的叶子是同一个,怎么办呢?容易发现,这种情况下从 (s0,s1)(e0,e1) 分别走到叶子的路径拼起来已经是 dis(s0,s1,e0,e1) 了,但还需要经过更多的边,这种情况自然更劣,不会作为 min 保留。

时间复杂度 O(mi+qlogmi)

本题特别卡空间,建议使用树剖 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;
}
posted @   rui_er  阅读(18)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 25岁的心里话
· 按钮权限的设计及实现
点击右上角即可分享
微信分享提示