根据团体程序设计天梯赛L2-001 紧急救援思考的dijkstra模板(dijkstra)

本文介绍了一个城市应急救援队伍负责人如何利用特殊地图,规划从出发城市到目的地城市的最短路径,并沿途召集最多的救援队。通过输入城市数量、快速道路信息及救援队分布,采用Dijkstra算法实现路径规划。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

作为一个城市的应急救援队伍的负责人,你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候,你的任务是带领你的救援队尽快赶往事发地,同时,一路上召集尽可能多的救援队。

输入格式:
输入第一行给出4个正整数N、M、S、D,其中N(2≤N≤500)是城市的个数,顺便假设城市的编号为0 ~ (N−1);M是快速道路的条数;S是出发地的城市编号;D是目的地的城市编号。
第二行给出N个正整数,其中第i个数是第i个城市的救援队的数目,数字间以空格分隔。随后的M行中,每行给出一条快速道路的信息,分别是:城市1、城市2、快速道路的长度,中间用空格分开,数字均为整数且不超过500。输入保证救援可行且最优解唯一。

输出格式:
第一行输出最短路径的条数和能够召集的最多的救援队数量。第二行输出从S到D的路径中经过的城市编号。数字间以空格分隔,输出结尾不能有多余空格。

输入样例:
4 5 0 3
20 30 40 10
0 1 1
1 3 2
0 3 3
0 2 2
2 3 2

输出样例:
2 60
0 1 3

dijkstra模板

注释部分为本题所需 看情况使用
#define INF 0x3f3f3f3f
#define rep(i, l, r) for(int i  = l; i < r; i++)
const int MAXN = 1e5 +7;
int N, M;  //N点M边 
int vis[MAXN];  //标记是否加入 
int dis[MAXN];  //记录i点和源点距离 
int num[MAXN];  //记录源点到i点最短路条数
int G[MAXN][MAXN] //记录边图
int P[MAXN]  //记录前驱 

/*
int ret[MAXN] //记录每点的物品数
int gather[MAXN]  //记录到i点的最短路上的最多物品数 
*/

void dijstra(int s){
 fill(vis, vis + MAXN, 0);
 fill(P, P + MAXN, -1);
 fill(num, num, 0);
 fill(dis, dis + MAXN, INF);
/*
 fill(gather, gather + MAXN, 0);
*/
 rep(i, 0, N) {
  int u = -1, min = INF;
  rep(j, 0, N)
   if(!vis[j] && dis[j] < min) {
    min = dis[j];
    u = j;
   }
  if(u == -1) return;
  vis[u] = 1;
  rep(v, 0, N) {
   if(!vis[v] && G[u][v] != INF) {
    if(dis[v] > dis[u] + G[u][v]){  //v由u点到达更短 
     P[v] = u;  //记录前驱 
     dis[v] = dis[u] + G[u][v];  //修改最短路长度 
     num[v] = num[u];  //到v的最短路经过u 条数也就和u一致 
     /*
      gather[v] = gather[u] + ret[v]; 
     */ 
    }
    else if(dis[v] == dis[u] + G[u][v]){
     num[v] += num[u];  //相等时有多条路径可达 条数累加 
     /*
     因为在 ‘城市救援 ’一题中还要求救援人数最多 
     if(gather[v] < gather[u] + ret[v]){
      P[v] = u;
      gather[v] = gather[u] + ret[v];
       
     }
     */
    }
   }
  }
 }
} 

AC代码

#include<bits/stdc++.h> 
#define reg register
#define ll long long
#define ull unsigned long long
#define INF 0x3f3f3f3f
#define lINF (1ll << 60)
#define eps 1e-8
#define pi acos(-1.0)
#define e exp(1.0)
#define ios ios::sync_with_stdio(false)
#define rep(i,l,r) for(int i=l;i<r;i++)
#define per(i,r,l) for(int i=r;i>=l;i--)
using namespace std;
const int MAXN = 505;
const int mod = 109;
template<class T>inline void read(T &res){
  char c;T flag = 1;
  while((c = getchar()) < '0' || c > '9')if(c == '-')flag = -1;res = c - '0';
  while((c = getchar()) >= '0' && c <= '9')res = res * 10 + c - '0'; res *= flag;
}
template<class T>inline void write(T x){
  if(x < 0) putchar('-'),x = -x;
  if(x > 9) write(x/10);
  putchar(x % 10 + '0');
}
ll MyPow(ll a, ll b){ ll sum = 1;while (b) {if (b & 1) {sum = (sum * a) % mod;b--;}b /= 2;a = a * a % mod;}return sum;}
ll gcd(ll a,ll b){ return b==0?a:gcd(b,a%b); }
ll isprime(ll n){if(n <= 1) return 0;if(n == 2) return 1;rep(i, 2, sqrt(n) + 1) if(n % i == 0) return 0;return 1;}
int num[MAXN], G[MAXN][MAXN], vis[MAXN], dis[MAXN], gather[MAXN];
int res[MAXN], P[MAXN], node[MAXN];
int N, M, S, D;
int x, y, z;
void dijkstra (int s) {
 fill(P, P + MAXN, -1);
 fill(num, num + MAXN, 0);
 fill(vis, vis + MAXN, 0);
 fill(dis, dis + MAXN, INF);
 fill(gather, gather + MAXN, 0);
 dis[s] = 0;
 num[s] = 1;
 P[s] = 0;
 gather[s] = res[s];
 rep(i, 0, N) {
  int u = -1, min = INF;
  rep(j, 0, N) {
   if(!vis[j] && dis[j] < min) {
    min = dis[j];
    u = j;
   }
  }
  if(u == -1) return ;
  vis[u] = 1;
  rep(v, 0, N) {
   if(!vis[v] && G[u][v] != INF) {
    if(dis[v] > dis[u] + G[u][v]) {
     P[v] = u;
     num[v] = num[u];
     dis[v] = dis[u] + G[u][v];
     gather[v] = gather[u] + res[v];
    }
    else if(dis[v] == dis[u] + G[u][v]) {
     num[v] += num[u];
     if(gather[v] < gather[u] + res[v]) {
      P[v] = u;
      gather[v]  = gather[u] + res[v]; 
     }
    }
   }
  }
 }
}
int main() {
 cin >> N >> M >> S >> D;
 fill(G[0], G[0] + MAXN * MAXN, INF);
 rep(i, 0, N) cin >> res[i];
 rep(i, 0, M) {
  cin >> x >> y >> z;
  G[x][y] = G[y][x] = z;
 }
 dijkstra(S);
 cout << num[D] << " " << gather[D] << endl;
 int o = 0;
 node[o++] = D;
 while(P[D] != 0) {
  node[o++] = P[D];
  D = P[D];
 }
 node[o] = S;
 per(i, o, 0){
  if(i) cout << node[i] << " ";
  else cout << node[i] << endl;
 }
  return 0;
}
### 关于 PAT 天梯赛 L2-001 紧急救援 的分析 题目描述通常涉及在一个加权图中找到从起点到终点的最短路径,同时满足某些条件(如最大权重边最小)。此类问题可以通过 **Dijkstra算法** 和 **优先队列优化** 来解决。 #### 图论基础回顾 在图论中,单源最短路径问题是经典的算法问题之一。对于带正权值的有向无环图 (Directed Acyclic Graph, DAG),或者一般的加权图,Dijkstra 是一种高效的解决方案[^1]。该算法通过维护一个优先队列来动态更新节点的距离,并逐步扩展已知最优路径至目标节点。 针对本题的具体需求——不仅需要求出最短路径长度,还需要记录路径上的最大权重以及可能的不同路径数量,以下是详细的解答思路: --- #### 数据结构设计 为了高效处理输入数据并支持后续查询操作,需定义如下辅助类或变量: 1. 使用邻接表存储图的信息。 2. 定义三个数组分别保存当前顶点的状态: - `dist[v]` 表示从起始点到达 v 的最短距离; - `max_edge[v]` 记录从起点到 v 路径上遇到的最大边权值; - `path_count[v]` 统计有多少条不同的路径能够达到相同的 `(dist[v], max_edge[v])` 值组合。 初始化时设所有顶点不可达 (`inf`) 并设置起点状态为零。 --- #### 主要逻辑流程 采用 Dijkstra 算法的核心思想,结合优先队列实现多维度比较功能。具体步骤如下所示: 1. 初始化优先队列 PQ,按照三元组 `(distance, maximum edge weight, node)` 排序,其中 distance 应作为主要关键字升序排列,maximum edge weight 则次之降序排列。 2. 将初始结点加入队列,其对应的距离和最大边权均置为 0;路径数记作 1。 3. 循环提取队首元素 u 进行松弛操作:遍历与 u 相连的所有邻居 w,尝试更新它们的相关属性。如果发现更优解,则同步调整 dist[w], max_edge[w] 及 path_count[w] 的取值;当新旧两方案具有相同效果时累加路径数目即可。 4. 结束循环直至访问完全图中的每一个可触及位置为止。 最终输出结果应包括总耗时、途中经历过的最高危险等级数值以及符合条件的有效路线总数模去某个固定常量后的余数形式呈现出来。 --- #### 实现代码样例 下面是基于 Python 编写的完整程序版本: ```python import heapq def dijkstra(n, start, end, graph): INF = float('inf') # Initialize arrays to store distances, max edges and counts. dist = [INF]*n max_edge = [-1]*n count = [0]*n pq = [(0, 0, start)] # Priority queue with tuple of (current_distance, current_max_edge, vertex). dist[start] = 0 max_edge[start] = 0 count[start] = 1 while pq: d_u, m_e_u, u = heapq.heappop(pq) if u == end: break for v, cost in graph[u]: alt_dist = d_u + cost # Compute the new potential 'max edge' value along this route. candidate_me_v = max(m_e_u, cost) if alt_dist < dist[v]: dist[v] = alt_dist max_edge[v] = candidate_me_v count[v] = count[u] # Push updated state into priority queue. heapq.heappush(pq, (alt_dist, candidate_me_v, v)) elif alt_dist == dist[v] and candidate_me_v < max_edge[v]: max_edge[v] = candidate_me_v count[v] += count[u] return dist[end], max_edge[end], count[end]%int(1e9+7) if __name__ == "__main__": n_vertices, n_edges, src, dst = map(int, input().split()) g = [[]for _ in range(n_vertices)] for i in range(n_edges): a,b,cost=map(int,input().strip().split()) g[a].append((b,cost)) res=dijkstra(n_vertices,src,dst,g) print(*res) ``` 上述脚本实现了完整的读入解析过程并通过标准输入获取必要的参数配置信息之后调用了核心函数完成任务执行最后按指定格式打印答案内容. --- #### 时间复杂度分析 由于采用了二叉堆作为底层支撑的数据结构,在每次迭代过程中最多只需要 O(log E) 的时间成本来进行插入/删除动作,因此整体运行效率大致维持在线性对数级别范围内即 O(E log V)---
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值