最短路徑 pat 1003

#include <iostream>
//1003
using namespace std;
const int N = 510;
const int inf = 1e9;
int n,m,st,ed;
int vis[N]={0};
int G[N][N] = {0};
int w[N],weight[N];
int d[N];
int num[N];
void dij(int s){
       fill(d,d+N,inf);
       d[s] = 0;            //起點出發到起點的距離爲0
                            //不用vis[s] = 1; 因爲 d[s] = 0
                            //在後面的循環判斷之後 第一次 會令u = 起點s
       w[s] = weight[s];//路徑的初始重量 爲 起點的重量
       num[s] = 1;      //路徑的條數 初始爲 1!!
       for(int i = 0; i < n; i++){
             int u = -1,Min = inf;
             for(int j = 0; j <n;j++){
              if(vis[j]==0 && d[j]<Min){
                     Min = d[j];
                     u=j;
                }
             }
             if(u==-1) return;
             vis[u] = 1;
             for(int v = 0;  v<n;v++){
                     if(vis[v]==0 && G[u][v]!=inf  ){
                            if(G[u][v]+d[u] < d[v]){
                                   d[v] = d[u] + G[u][v];
                                   w[v] = w[u] + weight[v];
                                   num[v] = num[u]; //從 起點 到 v 的路徑 即爲 起點到u 的路徑數目
                            }else if(G[u][v]+d[u] == d[v] ){
                                   if(w[u]+ weight[v] > w[v]){
                                          w[v]= w[u]+ weight[v];
                                   }
                                   num[v]+=num[u];//條數與權重無關所以姚在外面
                                                 ////從起點 到 v 的路徑數 X  起點到u 的路徑數目
                                                 //乘即爲累加
                            }
                     }
             }
       }
}


int main()
{
       freopen("1003.txt","r",stdin);
       cin>>n>>m>>st>>ed;//st起點,ed 爲終點
       //輸入點權
       for(int i = 0; i < n;i++){
              cin>>weight[i];
       }
       fill(G[0],G[0]+N*N,inf);//初始化圖
       //輸入邊及邊權
       int a,b,c;
       for(int i = 0; i < m;i++){
              cin>>a>>b>>c;
              G[a][b] = G[b][a] = c;
       }
       dij(st);
       printf("%d %d\n",num[ed],w[ed]);


    //cout << "Hello world!" << endl;
    return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章