1003 Emergency - 最短路的条数

思路：

又学了一遍dijk，具体的代码意思如下：

void dijk(){
    d[S]=0;//起点到起点距离为0
    for(int i=0;i<n;i++){//循环n次每次加一个点，每次加入的点都是已更新完成的点
        int v=-1;
        for(int j=0;j<n;j++){
            //每次都遍历所有的点，从没加入图中的点中选一个到原点距离最小的点，加入图中
            if(!vis[j]&&(v==-1||d[j]<d[v])){
                v=j;
            }
        }
        vis[v]=1;
        for(int j=0;j<n;j++){//进行松弛操作
            d[j]=min(d[j],d[v]+cost[v][j]);
        }
    }
}

用dijk走一遍最短路，可以用tot数组保存起点到每个点最短路的条数，tot[S]=1，用num数组表示到点i时候的人数，num[S]=a[S]

然后在松弛操作的时候:

（1）若d[j]==d[v]+cost[v][j]，则tot[j]+=tot[v]；num[j]=max(num[j],num[v]+a[j])

（2）若d[j]<d[v]+cost[v][j]，则tot[j]=tot[v]；num[j]=num[v]+a[j]

代码如下：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<queue>
#include<cmath>
#include<set>
#define ll long long
using namespace std;
const int N=505,INF=0x3f3f3f3f;
int a[N],mape[N][N];
int num[N],d[N],vis[N],n,m,c1,c2,tot[N];

void init(){
    memset(d,INF,sizeof(d));
    memset(vis,0,sizeof(vis));
    memset(num,0,sizeof(num));
    memset(tot,0,sizeof(tot));
}

void dijk(){
    d[c1]=0;
    num[c1]=a[c1];
    tot[c1]=1;
    for(int i=0;i<n;i++){
        int v=-1;
        for(int j=0;j<n;j++){
            if(!vis[j]&&(v==-1||d[j]<d[v])){
               v=j;
            }
        }
        vis[v]=1;
        for(int j=0;j<n;j++){
            if(j==v)continue;
            if(d[j]==d[v]+mape[v][j]){
                tot[j]+=tot[v];
                num[j]=max(num[j],num[v]+a[j]);
            }
            if(d[j]>d[v]+mape[v][j]){
                d[j]=d[v]+mape[v][j];
                tot[j]=tot[v];
                num[j]=num[v]+a[j];
            }
        }
    }
}

int main(){
    scanf("%d%d%d%d",&n,&m,&c1,&c2);
    init();
    for(int i=0;i<n;i++){
        scanf("%d",&a[i]);
    }
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            if(i==j)mape[i][j]=0;
            else mape[i][j]=INF;
        }
    }
    int s,e,c;
    while(m--){
        scanf("%d%d%d",&s,&e,&c);
        mape[s][e]=c;
        mape[e][s]=c;
    }
    dijk();
    printf("%d %d\n",tot[c2],num[c2]);
}