7-3 城市间紧急救援 (25 分)

7-3 城市间紧急救援 (25 分)

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

输入格式:

输入第一行给出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

分析:多条最短路径问题,只需要在dijstra的基础上改一下即可,保存多个前驱,最好dfs求最优解

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <map>
#include <set>
#include <vector>
#include <cstring>
#include <queue>
#include <cmath>
#define FRER() freopen("in.txt","r",stdin);
#define FREW() freopen("out.txt","w",stdout);
#define mem(a,b) memset(a,b,sizeof(a));
#define go int T;scanf("%d",&T);for(int cas=1;cas<=T;cas++)
#define mod 1000000007
using namespace std;
typedef pair<int,int> pii;
typedef long long ll;
const int maxn = 1000000 + 7;
int head[maxn],nxt[maxn],to[maxn],cost[maxn];
vector<int>prv[maxn];
vector<int>path;
vector<int>_path;
int d[maxn];
int nEdge,n,m,s,t,cnt,sum,res;
int w[maxn];
int vis[maxn];
void AddEdge(int u,int v,int c){
    nxt[nEdge] = head[u];
    to[nEdge] = v;
    cost[nEdge] = c;
    head[u] = nEdge++;
}
void Dijstra(){
    for(int i=0;i<n;i++) d[i] = 0x3f3f3f3f;
    memset(vis, 0, sizeof vis);
    priority_queue<pii,vector<pii>,greater<pii> >q;
    d[s] = 0;
    q.push(pii(0,s));
    while(!q.empty()){
        pii p = q.top();q.pop();
        int u = p.second , c = p.first;
        if(u==t) continue;
        if(vis[u]) continue;
        vis[u] = 1;
        for(int e=head[u];~e;e=nxt[e]){
            int v = to[e];
            if(c+cost[e]<d[v]){
                d[v] = c + cost[e];
                q.push(pii(d[v],v));
                prv[v].clear();
                prv[v].push_back(u);
            }
            else if(c+cost[e]==d[v]){
                prv[v].push_back(u);
            }
        }
    }
}
void dfs(int u){
    if(u==s){
        cnt++;
        if(res<sum){
            res = sum;
            path = _path;
        }
        return;
    }
    for(int i=0;i<prv[u].size();i++){
        int v = prv[u][i];
        sum+=w[v];
        _path.push_back(v);
        dfs(v);
        _path.erase(_path.end()-1);
        sum-=w[v];
    }
}
void Init(){
    res = -0x3f3f3f3f;
    nEdge = cnt = 0;
    sum = w[t];
    memset(head, -1, sizeof head);
    for(int i=0;i<=n;i++) prv[i].clear();
}
int main(){
    //FRER();
    scanf("%d%d%d%d",&n,&m,&s,&t);
    for(int i=0;i<n;i++) scanf("%d",&w[i]);
    Init();
    for(int i=0;i<m;i++){
        int u,v,c;
        scanf("%d%d%d",&u,&v,&c);
        AddEdge(u, v, c);
        AddEdge(v, u, c);
    }
    Dijstra();
    dfs(t);
    cout<<cnt<<" "<<res<<endl;
    for(int i=(int)path.size()-1;i>=0;i--)
        printf("%d ",path[i]);
    
    printf("%d\n",t);
}

猜你喜欢

转载自blog.csdn.net/Insist_77/article/details/84111333