洛谷 P10206 建设工程 2 题解

SolemnLament

于 2024-10-02 23:25:20 发布

阅读量354

点赞数 9

分类专栏：洛谷题目 JOI 文章标签：算法 c++

本文链接：https://blog.csdn.net/weixin_47736562/article/details/142686844

版权

洛谷题目同时被 2 个专栏收录

8 篇文章 0 订阅

订阅专栏

JOI

1 篇文章 0 订阅

订阅专栏

前言

模拟赛的 T3，寄了，所以写篇题解（为什么 gg 掉的全都是 T3？ ${\mathbf {\color{orange}SolIII}}$ 你要好好反思一下）。

题目分析

简化题意

给定一个无向图，你可以任选两点 $u$ 和 $v$ 并在它们之间连一条长度为 $L$ 的无向边。问有多少种可能，使得连边后从 $s$ 到 $t$ 的最短路长度小于等于 $K$ ？

思路

先看 Subtask1 的思路，因为 $K$ 只为 $2$ ，而每条边的长度都为 $1$ ，所以从 $s$ 到 $t$ 的最短路只有可能经过经过至多两条边，即 $\leftrightarrow u$ 和 $\leftrightarrow t$ （ $u$ 为中间点）。所以考虑从 $s$ 和 $t$ 两点分别做两次最短路，求出这两点各自到所有点的最短路后再枚举点 $u$ 就可以了。

有了 Subtask1 的思路，Subtask2 和 Subtask3 的想法也就顺理成章了。同样地，我们分别从 $s$ 和 $t$ 两点做最短路后，枚举 $u$ 和 $v$ 两点。令从 $s$ 到 $u$ 的最短路为 $dis1_u$ ，从 $v$ 到 $t$ 的最短路为 $dis2_v$ ，若在 $u$ 和 $v$ 两点之间连一条长度为 $L$ 的无向边后这条路径（ $\to u \to v \to t$ ）的长度小于等于 $K$ ，则必有 $dis1_u + L + dis2_v \le K$ 。所以对于点对 $(u, v)$ ，只要它满足上述不等式，就为一个可行的方案。时间复杂度 $O(n ^ 2)$ ，足以满足这两个 Subtask 的数据范围。

对于 $\le N \le 2 \times 10 ^ 5$ ，我们考虑优化。

运用初中解不等式的知识，我们发现

$dis1_u + L + dis2_v \le K$

可以变形为

$dis2_v \le K - dis1_u - L$

所以对于某个点 $u$ ，若想算出满足 $dis1_u + L + dis2_v \le K$ 的点对 $(u, v)$ 的个数，我们可以将存储 $dis2_v$ 的数组排序并二分最大的满足小于等于 $K - dis1_u - L$ 的 $dis2_v$ ，再把 $v$ 的数量累加，就可以算出答案。总体时间复杂度 $\log n)$ 。

参考代码

#include<bits/stdc++.h>
using namespace std;
#define LL long long
const int NR = 2e5;
const int MR = 3e3;
const LL inf = 1e18; // 不开 long long 见祖宗
vector<pair<int, LL> > mp[NR + 10];
LL dis1[NR + 10];
LL dis2[NR + 10];
bool vis[NR + 10];
int n, m, s, t;
LL l, k;
// 最短路板子，不做解释
struct Node{
    int u;
    LL dis;
};
bool operator < (const Node &x, const Node &y){
    return x.dis > y.dis;
}
priority_queue<Node> q;
void dij(int st, LL dis[]){
    while(!q.empty()) q.pop();
    for(int i = 1;i <= n;i++){
        dis[i] = inf;
        vis[i] = false;
    }
    dis[st] = 0;
    q.emplace(Node{st, 0});
    while(!q.empty()){
        auto top = q.top();
        q.pop();
        int u = top.u;
        vis[u] = true;
        if(top.dis > dis[u]) continue;
        for(auto i : mp[u]){
            int v = i.first;
            LL w = i.second;
            if(!vis[v] && dis[u] + w < dis[v]){
                dis[v] = dis[u] + w;
                q.emplace(Node{v, dis[v]});
            }
        }
    }
    return ;
}

int main(){
    scanf("%d%d%d%d%lld%lld", &n, &m, &s, &t, &l, &k); // 读入，注意要开 long long
    for(int i = 1;i <= m;i++){
        int u, v;
        LL w;
        scanf("%d%d%lld", &u, &v, &w);
        mp[u].emplace_back(make_pair(v, w));
        mp[v].emplace_back(make_pair(u, w));
    }
    dij(s, dis1); // 从 s 和 t 两点分别跑一边最短路
    dij(t, dis2);
    if(dis1[t] <= k){ // 一个特判，如果不修铁路时已经满足了国王的要求，那么在随便哪两个点之间修铁路都可以取悦国王
        printf("%lld\n", (1ll * n * (n - 1)) / 2ll);
        return 0;
    }
    sort(dis2 + 1, dis2 + 1 + n); // 排序，方便二分
    LL ans = 0;
    for(int u = 1;u <= n;u++){
        if(dis1[u] >= inf) continue;                                             // s 走不到这里，直接跳过
        int v = upper_bound(dis2 + 1, dis2 + 1 + n, k - l - dis1[u]) - dis2 - 1; // 二分求满足要求的 v 的个数
        ans += 1ll * v;                                                          // 累加答案
    }
    printf("%lld\n", ans); // 输出
    return 0;
}