PAT 1003.Emergency

#include<iostream>
#include<vector>
#include<cstring>
#include<algorithm>
#include<map>
#include<queue>
using namespace std;
const int maxn = 510;//最大的节点数
const int INF = 1e9;//表示不可达
//n代表节点个数，m代表边数，st代表起点，ed代表终点，weight[]存储节点的权值
int n, m, st, ed, G[maxn][maxn], weight[maxn];
int num[maxn], w[maxn], d[maxn];//num[]到节点的最短路径个数，w[]存储从起始点到节点最大的资源，d存储到节点的距离，即花费
bool vis[maxn] = {false};//判断节点有没有被看过

void Dijkstra(int s){
     //初始化num，w，d，num 和 w要都赋成零，d赋值为无穷大
     memset(num, 0, sizeof(num));
     memset(w, 0, sizeof(w));
     fill(d, d + maxn, INF);
     //对起始点进行特殊赋值
     d[s] = 0;
     w[s] = weight[s];
     num[s] = 1;
     //开始循环，每次找到一个距离起始点最短的点，进行处理
     for(int i = 0; i < n; i ++){
          int u = -1, MIN = INF;
          //找到为遍历中距离最短的，注意第一次时起点还没有被遍历
          for(int j = 0; j < n; j ++){
               if(d[j] < MIN && vis[j] == false){
                    u = j;
                    MIN = d[j];
               }
          }
          //如果找到，标记已经遍历，否则退出
          if(u == -1){
               printf("%d %d", num[ed], w[ed]);
               return;
          }
          vis[u] = true;
          //更新最短距离
          for(int j = 0; j < n; j ++){
               if(G[u][j] != INF && vis[j] == false){
                    //如果是最短距离，更新距离，最短距离数直接覆盖，w直接覆盖
                    if(d[u] + G[u][j] < d[j]){
                         d[j] = d[u] + G[u][j];
                         num[j] = num[u];
                         w[j] = w[u] + weight[j];
               //如果存在多条最短路径，num相加，判断w是不是最优解
                    } else if(d[u] + G[u][j] == d[j]){
                         num[j] += num[u];
                         if(w[u] + weight[j] > weight[j])
                              w[j] = w[u] + weight[j];
                    }
               }
          }
     }
}
int main(){
     int s, e, k;
     scanf("%d%d%d%d", &n, &m, &st, &ed);
     for(int i = 0; i < n; i ++) scanf("%d", &weight[i]);
     fill(G[0], G[0] + maxn * maxn, INF);
     for(int i = 0; i < m; i ++){
          scanf("%d%d%d", &s, &e, &k);
          G[s][e] = k;
          G[e][s] = k;
     }
     Dijkstra(st);
     printf("%d %d", num[ed], w[ed]);
     system("pause");
     return 0;
}