1. 程式人生 > >PTA 資料結構與演算法題目集(中文) 7-35 城市間緊急救援(25 分) 迪傑斯特拉演算法

PTA 資料結構與演算法題目集(中文) 7-35 城市間緊急救援(25 分) 迪傑斯特拉演算法

7-35 城市間緊急救援(25 分)

作為一個城市的應急救援隊伍的負責人,你有一張特殊的全國地圖。在地圖上顯示有多個分散的城市和一些連線城市的快速道路。每個城市的救援隊數量和每一條連線兩個城市的快速道路長度都標在地圖上。當其他城市有緊急求助電話給你的時候,你的任務是帶領你的救援隊儘快趕往事發地,同時,一路上召集儘可能多的救援隊。

輸入格式:

輸入第一行給出4個正整數N、M、S、D,其中N(2≤N≤500)是城市的個數,順便假設城市的編號為0 ~ (N−1);M是快速道路的條數;S是出發地的城市編號;D是目的地的城市編號。

第二行給出N個正整數,其中第i個數是第i個城市的救援隊的數目,數字間以空格分隔。隨後的M行中,每行給出一條快速道路的資訊,分別是:城市1、城市2、快速道路的長度,中間用空格分開,數字均為整數且不超過500。輸入保證救援可行且最優解唯一。

輸出格式:

第一行輸出最短路徑的條數和能夠召集的最多的救援隊數量。第二行輸出從S到D的路徑中經過的城市編號。數字間以空格分隔,輸出結尾不能有多餘空格。

輸入樣例:

4 5 0 3
20 30 40 10
0 1 1
1 3 2
0 3 3
0 2 2
2 3 2

輸出樣例:

2 60
0 1 3
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <vector>
#include <stack>
using namespace std;
const int maxn=505;
const int INF=0x3f3f3f3f;
int n,m,s,d;
int sp[maxn]; //救援人數
int dj[maxn]; //距離起點最短距離
int vis[maxn];//是否列入集合
int road[maxn];//最短路徑條數
int path[maxn];//字首節點
int total[maxn];//總救援人數
struct city
{
    int e;
    int len;
};
vector <city> ve[maxn];
//初始化
void init ()
{
    memset (vis,0,sizeof(vis));
    memset (total,0,sizeof(total));
    memset (road,0,sizeof(road)); 
    memset (path,-1,sizeof(path));
    for (int i=0;i<n;i++)
        dj[i]=INF;
    dj[s]=0;
    total[s]=sp[s];
    road[s]=1;
}
void djst ()
{
    while (1)
    {
       int maxx=INF;
       int u=-1;
       for (int i=0;i<n;i++)
            if(dj[i]<maxx&&!vis[i])
            {
               maxx=dj[i];
               u=i;
            }
       if(u==-1)
          break;
       vis[u]=1;
       for (int i=0;i<ve[u].size();i++)
       {
           int v=ve[u][i].e;
           if(!vis[v])
           {
               if(dj[v]>dj[u]+ve[u][i].len)
               {
                  dj[v]=dj[u]+ve[u][i].len;
                  path[v]=u;
                  road[v]=road[u];
                  total[v]=total[u]+sp[v];

               }
               else if(dj[v]==dj[u]+ve[u][i].len)
               {
                   //這裡要寫到if語句外面
                    road[v]+=road[u];
                    if(total[u]+sp[v]>total[v])
                    {
                       path[v]=u;
                       total[v]=total[u]+sp[v];
                    }
               }
           }
       }
    }
}
//輸出函式
void output()
{
    printf("%d %d\n",road[d],total[d]);
    int temp=d;
    stack<int>ss;
    while (path[temp]!=-1)
    {
        ss.push(temp);
        temp=path[temp];
    }
    ss.push(s);
    while (!ss.empty())
    {
        printf("%d%c",ss.top(),ss.size()==1? '\n':' ');
        ss.pop();
    }
}
int main()
{
    scanf("%d%d%d%d",&n,&m,&s,&d);
    for (int i=0;i<n;i++)
        scanf("%d",&sp[i]);
    init();
    for (int i=0;i<m;i++)
    {
        int x,y,len;
        scanf("%d%d%d",&x,&y,&len);
        city temp1,temp2;
        temp1.e=y; temp1.len=len;
        temp2.e=x; temp2.len=len;
        ve[x].push_back(temp1);
        ve[y].push_back(temp2);
    }
    djst();
    output();
    return 0;
}