L2-001 緊急救援 (25分)

zxjjjjjjjj發表於2020-11-26

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

輸入格式:
輸入第一行給出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

1 輸出最短路徑條數是一共有多少條最短路徑,不是最短路徑有多少條路(我當初是怎麼想的???)
2 注意條件為 初始地址==目的地址 的時候

#include<bits/stdc++.h>
using namespace std;
int grapg[510][510];//記錄距離
int lowcost[510];//記錄最短距離
int mst[510];//記錄前節點
int flag[510]={0};//判斷是否被記錄
int cnt[510]={0};//記錄最短路徑條數
int coun[510];//記錄城市救援隊數量
int people[510]={0};//記錄總的救援隊數量
int main(){
    int num,start,road,end;
	fill(lowcost,lowcost+510,10000);
    fill(mst,mst+510,-1);
	cin>>num>>road>>start>>end;
	for(int i=0;i<num;i++){
		for(int j=0;j<num;j++)grapg[i][j] = 10000;
	}
    for(int i=0;i<num;i++){
        cin>>coun[i];
    }
    for(int j=0;j<road;j++){
        int a,b,c;
        cin>>a>>b>>c;
        grapg[a][b]=c;
        grapg[b][a]=c;
    }
	lowcost[start]=0;//起點設定最短路徑為0
    people[start]=coun[start];//初始化起點的救援隊數量
	mst[start]=start;//初始化起點的前節點
	cnt[start]=1;//初始化起點最短路徑數量
    while(1){
        int mini=-1;
        for(int i=0;i<num;i++){//找到lowcost裡最小資料
            if((lowcost[i]<lowcost[mini]||mini==-1)&&!flag[i]){
                mini=i;
            }
		}
        flag[mini]=1;
        for(int i=0;i<num;i++){
            if(grapg[mini][i]+lowcost[mini]<=lowcost[i]){
                if(grapg[mini][i]+lowcost[mini]==lowcost[i]){
                    cnt[i]+=cnt[mini];//只要最短距離相同,最短路徑數量就要增加
                    if(people[mini]+coun[i]<=people[i]){//比較救援隊數量
                       continue;
                    }			
                }
				else cnt[i]=cnt[mini];
				people[i]=people[mini]+coun[i];//救援隊數量增加
                lowcost[i]=grapg[mini][i]+lowcost[mini];
                mst[i]=mini;//修改前繼節點
            }
        }
        if(mini==end)break;
    }
	
    int end1=end;
	cout<<cnt[end]<<" "<<people[end]<<endl;
    stack<int>st1;
    while (1)//用棧存路徑
    {   st1.push(end);
	    if(end==mst[end])break;
        end=mst[end];
    }
    while(!st1.empty()){//輸出路徑
        cout<<st1.top();
        st1.pop();
		if(!st1.empty())cout<<" ";
    }
    system("pause");
    
}

相關文章