最短路径救援队

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

【输入形式】输入第一行给出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

代码如下(C编写):

#include <stdio.h>
#include <stdlib.h>
#include <limits.h>

#define MAXN 505
#define INF INT_MAX

// 邻接表结构
typedef struct Edge {
    int to, w;
    struct Edge *next;
} Edge;

Edge* head[MAXN];
int N, M, S, D;
int team[MAXN];

// Dijkstra 相关数组
int dist[MAXN];        // 最短距离
long long cnt[MAXN];   // 最短路径条数
int maxTeam[MAXN];     // 对应最短路径能召集的最多队伍数
int pre[MAXN];         // 用来回溯最终路径
int vis[MAXN];         // 访问标记

// 添加一条无向边
void addEdge(int u, int v, int w) {
    Edge* e = (Edge*)malloc(sizeof(Edge));
    e->to = v; e->w = w;
    e->next = head[u]; head[u] = e;
}

int main() {
    scanf("%d %d %d %d", &N, &M, &S, &D);
    for (int i = 0; i < N; i++) {
        scanf("%d", &team[i]);
        head[i] = NULL;
    }
    for (int i = 0; i < M; i++) {
        int u, v, w;
        scanf("%d %d %d", &u, &v, &w);
        addEdge(u, v, w);
        addEdge(v, u, w);
    }
    // 初始化
    for (int i = 0; i < N; i++) {
        dist[i] = INF;
        cnt[i] = 0;
        maxTeam[i] = 0;
        pre[i] = -1;
        vis[i] = 0;
    }
    dist[S] = 0;
    cnt[S] = 1;
    maxTeam[S] = team[S];
    
    // 经典 Dijkstra 扩展
    for (int t = 0; t < N; t++) {
        int u = -1, mind = INF;
        // 找到未访问的最小 dist
        for (int i = 0; i < N; i++) {
            if (!vis[i] && dist[i] < mind) {
                mind = dist[i];
                u = i;
            }
        }
        if (u == -1) break;
        vis[u] = 1;
        // 松弛
        for (Edge* e = head[u]; e; e = e->next) {
            int v = e->to, w = e->w;
            if (vis[v]) continue;
            int nd = dist[u] + w;
            if (nd < dist[v]) {
                dist[v] = nd;
                cnt[v] = cnt[u];
                maxTeam[v] = maxTeam[u] + team[v];
                pre[v] = u;
            } else if (nd == dist[v]) {
                // 路径条数累加
                cnt[v] += cnt[u];
                // 如果这条新路能召集更多队伍,则更新前驱和队伍数
                if (maxTeam[u] + team[v] > maxTeam[v]) {
                    maxTeam[v] = maxTeam[u] + team[v];
                    pre[v] = u;
                }
            }
        }
    }
    
    // 输出:最短路径条数 + 最大救援队数
    printf("%lld %d\n", cnt[D], maxTeam[D]);
    // 回溯并输出路径
    int stack[MAXN], top = 0;
    for (int v = D; v != -1; v = pre[v]) {
        stack[top++] = v;
    }
    // 逆序打印
    for (int i = top - 1; i >= 0; i--) {
        printf("%d%s", stack[i], i == 0 ? "\n" : " ");
    }
    
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值