洛谷 P1396 营救

---

---

营救

题目链接：https://www.luogu.com.cn/problem/P1396

题目背景

“咚咚咚……”“查水表！”原来是查水表来了，现在哪里找这么热心上门的查表员啊！小明感动得热泪盈眶，开起了门……

题目描述

妈妈下班回家，街坊邻居说小明被一群陌生人强行押上了警车！妈妈丰富的经验告诉她小明被带到了 $t$ 区，而自己在 $s$ 区。

该市有 $m$ 条大道连接 $n$ 个区，一条大道将两个区相连接，每个大道有一个拥挤度。小明的妈妈虽然很着急，但是不愿意拥挤的人潮冲乱了她优雅的步伐。所以请你帮她规划一条从 $s$ 至 $t$ 的路线，使得经过道路的拥挤度最大值最小。

输入格式

第一行有四个用空格隔开的 $n$，$m$，$s$，$t$，其含义见【题目描述】。

接下来 $m$ 行，每行三个整数 $u,v,w$，表示有一条大道连接区 $u$ 和区 $v$，且拥挤度为 $w$。

两个区之间可能存在多条大道。

输出格式

输出一行一个整数，代表最大的拥挤度。

样例 #1

样例输入 #1

3 3 1 3
1 2 2
2 3 1
1 3 3

样例输出 #1

2

提示

数据规模与约定

• 对于 $30\%$ 的数据，保证 $n\le 10$。
• 对于 $60\%$ 的数据，保证 $n\le 100$。
• 对于 $100\%$ 的数据，保证 $1\le n\le 10^4$，$1\le m\le 2\times 10^4$，$w\le 10^4$，$1\le s,t\le n$。且从 $s$ 出发一定能到达 $t$ 区。

---

样例输入输出 1 解释

小明的妈妈要从 $1$ 号点去 $3$ 号点，最优路线为 $1$->$2$->$3$。

---

---

具体思路

最短路问题，但是是求路径上的尽可能小的最大权值。所以我们的公式需要稍微变一下，不是单纯的比较路径长度问题，而是将当前节点的最大权与路径权取最大后跟下一个节点取最小。

int mnum = max(dist[ver], w[i]);
if(mnum < dist[j]){
    dist[j] = mnum;
    heap.push({dist[j], j});
}

---

代码实现

#include <iostream>
#include <vector>
#include <cstring>
#include <algorithm>
#include <queue>
#define ll long long
#define INF 0x3f3f3f3f 

using namespace std;

typedef pair<int, int> pii; // 定义一个pair类型，用于存储距离和区号

const int N = 1E4 + 10, M = 4E4 + 10; // 定义常量，表示最大的区数和大道数
int n, m, s, k; // 定义变量，分别表示区的总数，大道的总数，起点区号和终点区号
int h[N], e[M], ne[M], w[M], idx; // 定义数组，分别表示邻接表的头指针，边的终点，下一条边的下标，边的权值，边的编号
int dist[N]; // 定义数组，表示从起点到每个区的最大拥挤度
bool st[N]; // 定义数组，表示每个区是否已经确定了最短距离

void add(int a, int b, int c){
    e[idx] = b; 	// 存储边的终点
    ne[idx] = h[a]; // 存储下一条边的下标
    w[idx] = c; 	// 存储边的权值
    h[a] = idx++; 	// 更新头指针，并增加边的编号
}

void dijkstra(){
    memset(dist, INF, sizeof dist); // 初始化距离数组为无穷大
    dist[s] = 0; // 起点到自己的距离为0
    
    priority_queue<pii, vector<pii>, greater<pii>> heap; // 定义一个优先队列，按照距离从小到大排序
    heap.push({0, s}); // 将起点入队
    
    while(heap.size()){ // 当队列不为空时循环
        auto t = heap.top(); // 取出队首元素
        heap.pop(); // 弹出队首元素
        int ver = t.second, dis = t.first; // 获取当前区号和距离
        
        if(st[ver]) continue; // 如果当前区已经确定了最短距离，跳过
        
        st[ver] = true; // 否则标记当前区为已确定
        
        for(int i = h[ver]; i != -1; i = ne[i]){ // 遍历当前区连接的所有大道
            int j = e[i]; // 获取另一个端点
            
            int mnum = max(dist[ver], w[i]); // 计算从起点到另一个端点经过的最大拥挤度
            if(mnum < dist[j]){ // 如果这个拥挤度小于之前记录的拥挤度
                dist[j] = mnum; // 更新拥挤度
                heap.push({dist[j], j}); // 将另一个端点入队
            }
        }
    }
}

int main()
{
    cin >> n >> m >> s >> k; // 输入区的总数，大道的总数，起点区号和终点区号
    memset(h, -1, sizeof h); // 初始化邻接表头指针为-1
    
    while(m--){ // 输入m条大道的信息
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c); // 输入两个端点和拥挤度
        add(a, b, c), add(b, a, c); // 添加两条有向边，构成无向边
    }
    
    dijkstra(); // 调用Dijkstra算法求解最短路
    
    cout << dist[k] << endl; // 输出结果
}

---

注意事项

• 本题我用 $SPFA$ 最后一个点会TLE，但是题解过了（甚至 $4ms$），不太懂怎么办到的…
• 堆优化 $Dijkstra$ 必须要用pii的堆结构，因为堆的比较是按照拥挤度来排的。如果只用int，我们每次插入堆的是编号，而不是拥挤度的值，所以在排序的时候排的是节点编号的大小而不是拥挤度的值。