题目链接:https://pintia.cn/problem-sets/994805342720868352/problems/994805523835109376
题目大意:n个城市(从0开始编号),m条路,每个城市都有一定数量的救援人员,请你输出从c1到c2的最短路的条数(不是最短路长度),以及在最短路上所能召集的最多救援人员数。
思路:最短路算法Dijk即可,数据不强,没用spfa。用cnt[i]表示第i个城市的最短路的数量,ans[i]表示第i个城市所能召集的最多人数。
#include<bits/stdc++.h>
using namespace std;
const int maxn=500+5;
const int inf=0x3f3f3f3f;
int n,m,st,ed,a,b,c,num[maxn],cnt[maxn],vis[maxn],dis[maxn],ans[maxn],g[maxn][maxn];
int Dijk(int s) {
memset(vis,0,sizeof(vis));
memset(dis,inf,sizeof(dis));
cnt[s]=1;
dis[s]=0;
for(int i=0; i<n; i++) {
int _min=inf,pos=-1;
for(int j=0; j<n; j++) {
if(!vis[j]&&dis[j]<_min) {
pos=j;
_min=dis[j];
}
}
if(_min!=inf&&pos!=-1) vis[pos]=1;
else return 0;
for(int j=0; j<n; j++) {
if(!vis[j]&&g[pos][j]!=inf) {
if(dis[pos]+g[pos][j]<dis[j]) {
dis[j]=dis[pos]+g[pos][j];
ans[j]=ans[pos]+num[j];
cnt[j]=cnt[pos];
} else if(dis[pos]+g[pos][j]==dis[j]) {
cnt[j]=cnt[j]+cnt[pos];
if(ans[pos]+num[j]>ans[j]) {
ans[j]=ans[pos]+num[j];
}
}
}
}
}
return 1;
}
int main() {
cin>>n>>m>>st>>ed;
for(int i=0; i<n; i++) {
cin>>num[i];
ans[i]=num[i];
}
memset(g,inf,sizeof(g));
for(int i=1; i<=m; i++) {
cin>>a>>b>>c;
g[a][b]=g[b][a]=c;
}
if(Dijk(st))
cout<<cnt[ed]<<' '<<ans[ed]<<endl;
}