PAT (Advanced Level) Practice A1003 Emergency （25 分）(甲级)(C++)（无向图、Dijkstra算法）

原题链接：A1003 Emergency

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<cstring>
#include <ctime>
#include <cmath>
#include <vector>
#include <set>
#include <map>
#include <stack>
#include <queue>
using namespace std;

const int MAX = 510, INF = 1<<30;
int N, M, now, des, C1, C2, L;
int G[MAX][MAX] = {{0}};//用无向图来表示城市中有多少救援队，且是否有公路
int Teams[MAX] = {0}, t[MAX] = {0}, d[MAX], n[MAX] = {0};//t表示到达每个城市可以聚集的最多的救援队
bool visited[MAX] = {0};//d表示到达每个城市的最小距离，n表示到达每个城市的最短路径条数

void Dijkstra(int u)//Dijkstra算法
{
    fill(d, d+MAX, INF);
    d[u] = 0, t[u] = Teams[u], n[u] = 1;
    for(int i=0; i<N; i++)
    {
        int MIN = INF, v = -1;
        for(int j=0; j<N; j++)//找出未访问的最短路径
        {
            if(!visited[j] && d[j] < MIN)
            {
                MIN = d[j], v = j;
            }
        }
        if(v == -1) return ;
        visited[v] = 1;//标记为已访问
        for(int j=0; j<N; j++)//更新
        {
            if(!visited[j] && G[v][j])
            {
                if(d[v] + G[v][j] < d[j])//距离近则直接走近的
                {
                    d[j] = d[v] + G[v][j];
                    t[j] = t[v] + Teams[j];
                    n[j] = n[v];
                }
                else if(d[v] + G[v][j] == d[j])//距离相等则路径条数增加，并更新救援队数量
                {
                    if(t[v] + Teams[j] > t[j])
                    {
                        t[j] = t[v] + Teams[j];
                    }
                    n[j] += n[v];
                }
            }
        }
    }
}

int main()
{
    scanf("%d %d %d %d", &N, &M, &now, &des);
    for(int i=0; i<N; i++) scanf("%d", &Teams[i]);
    for(int i=0; i<M; i++)
    {
        scanf("%d %d %d", &C1, &C2, &L);
        G[C1][C2] = G[C2][C1] = L;
    }
    Dijkstra(now);
    printf("%d %d", n[des], t[des]);
    return 0;
}