7-35 城市间紧急救援 (25分)-dijkstra最短路径

#include <iostream>
#define infinity 65535
using namespace std;
int cnt[1000];//S到某点最短路径的数目
int low[1000];//S到某点最短路径长度
int high[1000] = { 0 };//储存最短路径中前一结点救援队集结数的最大值
int weight[1000];//某结点集结队数目
int g[1000][1000];
int front[1000];//最短路径中前驱结点
int final[1000] = { 0 };//某结点是否处理完毕，处理完置1
int N, M, S, D;
void print();    
int main()
{
    cin >> N >> M >> S >> D;
        for(int i=0;i<N;i++)cnt[i]=1;
        /*cnt需要初始化为1，因为当没有其他最短路径的时候，cnt的值没有机会被修改，
        也就是默认值，那么就应该为1,不应初始化为0*/
    for (int i = 0; i < N; i++)
    {
        for (int j = 0; j < N; j++)
        {
            g[i][j] = infinity;
        }
        low[i] = infinity;
    }
    for (int i = 0; i < N; i++)
        cin >> weight[i];
    for (int i = 0; i < M; i++)
    {
        int a, b, c;
        cin >> a >> b >> c;
        g[a][b] = g[b][a] = c;
    }
    for (int i = 0; i < N; i++)
    {
        low[i] = g[0][i];
    }
    for (int i = 0; i < N; i++)
    {
        if (g[S][i] != infinity)
        {
            low[i] = g[S][i];
            front[i] = S;
            high[i] = weight[S];
        }
    }
    front[S] = -1;
    final[S] = 1;
    low[S] = 0;
    int k;
//以下是dijkstra算法的实现
    for (int i = 0; i < N; i++)
    {
        int min = infinity;
        for (int j = 0; j < N; j++)
        {
            if (low[j] < min && !final[j])
            {
                min = low[j];
                k = j;
            }
        }
        high[k] = high[front[k]] + weight[front[k]];
        final[k] = 1;
        for (int w = 0; w < N; w++)
        {
            if (!final[w] && min + g[k][w] < low[w])
            {
                cnt[w]=cnt[k];
                low[w] = min + g[k][w];
                front[w] = k;
                high[w] = high[k] + weight[k];
            }
            else if (!final[w] && min + g[k][w] == low[w])/*else不能漏掉，因为low[w]在上一步改变了，有可能上下两个条件都符合，就会出现错误*/
            {
                cnt[w]+=cnt[k];//即使权值没有比原来大，也需要增加最短路径数
                if(high[k] + weight[k] > high[w])
                {
                    front[w] = k;
                    high[w] = high[k] + weight[k];
                }
            }
        }
    }
    print();
    return 0;
}
void print()
{    
    cout << cnt[D]<<' ';
    cout << high[D]+weight[D] << endl;
    int s[1000];
    int r=-1;
    while (D!=-1)//需要利用堆栈输出
    {
        s[++r] = D;
        D = front[D];
    }
    while (r != -1)
    {//注意处理最后面的空格
        if (r == 0)cout << s[r--];
        else cout << s[r--] << ' ';
    }
}