PAT1003 Emergency

题目描述

题意说明

输入： 

（1）第一行输入城市数量N、道路数量M、救援起点、救援终点。

（2）第二行输入这N个城市分别具有的救援物资数量，可以看成是城市点自带的权重。（救援队每经过一个城市，可以携带上这个城市的救援物资）

（3）下面M行，分别记录M条道路的起点城市、终点城市和路程。

输出：

（1）最短路径共有几条

（2）在最短路径中的，最大救援物资数量的值

代码

#include<bits/stdc++.h>
using namespace std;
#define INF 99999 

int main()
{
    int n,m,c1,c2;
    cin>>n>>m>>c1>>c2;
    vector< vector<int> > dis(n,vector<int>(n,INF)); //初始化路都不通
    vector<int> v(n,0); //救援队数量(即路的权重) 
    for(int i=0; i<n; i++)
    {
        cin>>v[i];
    }
    for(int i=0; i<m; i++)
    {
        int a,b,c;
        cin>>a>>b>>c;
        dis[a][b] = dis[b][a] = c;
    }
    
    vector<int> index(n,1); //标记已遍历的点
    vector<int> w(n,0); //用来更新权重
    vector<int> num(n,0); //用来保存最优路径数量
    vector<int> D(n,INF);
    D[c1] = 0;
    w[c1] = v[c1];
    num[c1] = 1;
    
    for(int i=0; i<n; i++)
    {
        //先找出发点去往下一个最近的点
        int p=-1;
        int minD=INF;
        for(int j=0; j<n; j++)
        {
            if(index[j] && minD>D[j])
            {
                p = j;
                minD = D[j];
            }
        }
        if(p == -1) break; //遍历完毕
        index[p] = false; //已经遍历过了
        //那就去遍历点P能去往的点
        for(int j=0; j<n; j++)
        {
            //更新start->j的距离
            if(index[j] && D[j]>D[p]+dis[p][j])
            {
                D[j] = D[p]+dis[p][j];
                num[j] = num[p];
                w[j] = w[p]+v[j];
            }
            else if(index[j] && D[j]==D[p]+dis[p][j]) //出现相同路径 
            {
                num[j] += num[p]; //叠加最优路径数量 
                w[j] = w[j]>(w[p]+v[j]) ? w[j] : (w[p]+v[j]); //选择权重更大的新路 
            }
        } 
    }
    cout<<num[c2]<<" "<<w[c2]<<endl;
    
    return 0;
}