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
分析:
我tm的忘了n2的dijkstra咋写.......
抄了一套板子然后就明白了,更新k的时候条件变多一个,==的情况判断救援队伍多少,然后决定是否更新,+路径回溯.
#include<bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f3f
const int maxn = 503;
int road[maxn][maxn];
int dis[maxn];
int num[maxn];//救援队数目
int all_num[maxn];
bool vis[maxn];
int last[maxn];
int cnt[maxn];
int pre[maxn];
int n,m,s,d;
void init()
{
for(int i=0; i<=n; i++)
{
for(int j=0; j<=n; j++)
road[i][j] = INF;
dis[i] = INF;
vis[i] = false;
last[i] = i;
cnt[i] = 0;
pre[i] = -1;
all_num[i] = 0;
}
}
void dijkstra()
{
dis[s] = 0;
cnt[s] = 1;
for(int i=0; i<n; i++)
{
int minn = INF;
int k = -1;
for(int j=0; j<n; j++)
{
if(vis[j] == 0 && dis[j]<minn)
{
minn = dis[j];
k = j;
}
}
if(k == -1)
break;
vis[k] = true;
for(int j=0;j<n;j++)
{
if(!vis[j] && dis[k]+road[k][j] < dis[j])
{
dis[j] = dis[k]+road[k][j];
pre[j] = k;
cnt[j] = cnt[k];
all_num[j] = all_num[k]+num[j];
}
else if(!vis[j] && dis[k]+road[k][j] == dis[j])
{
cnt[j] += cnt[k];
if(all_num[j] < num[j]+all_num[k])
{
pre[j] = k;
all_num[j] = num[j] + all_num[k];
}
}
}
}
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
//freopen("in.txt","r",stdin);
cin>>n>>m>>s>>d;
init();
for(int i=0; i<n; i++)
{
cin>>num[i];
}
int a,b,c;
for(int i=0; i<m; i++)
{
cin>>a>>b>>c;
road[a][b] = c;
road[b][a] = c;
}
dijkstra();
cout<<cnt[d]<<" "<<all_num[d]+num[s]<<endl;
int x=0,t=d;
int ss[maxn];
while(pre[t]!=-1){//所经历的城市的从后往前的顺序
ss[x++]=pre[t];
t=pre[t];
}
cout<<s;//出发地
for(int i=x-2;i>=0;i--)
cout<<" "<<ss[i];
cout<<" "<<d;//目的地
}