6-08. 城市间紧急救援（25）--Dijkstra

6-08. 城市间紧急救援（25）题目地址
时间限制
150 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作为一个城市的应急救援队伍的负责人，你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候，你的任务是带领你的救援队尽快赶往事发地，同时，一路上召集尽可能多的救援队。

输入格式说明：

输入说明：输入第1行给出4个正整数N、M、S、D，其中N（2<=N<=500）是城市的个数，顺便假设城市的编号为0~(N-1)；M是快速道路的条数；S是出发地的城市编号；D是目的地的城市编号。第2行给出N个正整数，其中第i个数是第i个城市的救援队的数目，数字间以空格分隔。随后的M行中，每行给出一条快速道路的信息，分别是：城市1、城市2、快速道路的长度，中间用空格分开，数字均为整数且不超过500。输入保证救援可行且最优解唯一。

输出格式说明：

第1行输出最短路径的条数和能够召集的最多的救援队数量。第2行输出从S到D的路径中经过的城市编号。数字间以空格分隔，输出结尾不能有多余空格。

样例输入与输出：

序号 输入 输出
1
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
2
8 10 0 1
2 1 5 1 1 1 1 3
0 4 1
0 3 2
0 7 1
4 5 1
3 2 1
7 3 1
3 6 1
2 1 1
1 6 1
2 5 1
5 12
0 7 3 2 1
3
2 1 0 1
2 1
1 0 2
1 3
0 1

• dijkstra 经典应用变式，无论加上什么条件 解法都差不多
• 注意变量初始化，退出条件等

#include <iostream>  
#include <stdio.h>
#include <stdlib.h>
#include <vector>
#include <string>
#include <string.h>
#include <algorithm>
#include <math.h>
#include <queue>

using namespace std;

#define N 501
#define INF 99999999

int n, m, s, d;
int teams[N];
int mp[N][N];

bool vis[N];
int pre[N];
int amount[N];
int pathCount[N];
int dis[N];

void Dijkstra()
{
    int i;
    for (i = 0; i < n; i++)
    {
        pathCount[i] = 0;
        vis[i] = false;
        amount[i] = teams[i];
        pre[i] = -1;
        dis[i] = mp[s][i];
    }
    pathCount[s] = 1;
    dis[s] = 0;
    vis[s] = true;
    int newP = s;

    while (newP != d)
    {
        for (i = 0; i < n; i++)
        {
            if (!vis[i])
            {
                //
                if (dis[newP] + mp[newP][i] < dis[i])
                {
                    dis[i] = dis[newP] + mp[newP][i];
                    pathCount[i] = pathCount[newP];
                    amount[i] = amount[newP] + teams[i];
                    pre[i] = newP;
                }
                else if (dis[newP] + mp[newP][i] == dis[i])
                {
                    pathCount[i] += pathCount[newP];
                    if (amount[newP] + teams[i] > amount[i])
                    {
                        amount[i] = amount[newP] + teams[i];
                        pre[i] = newP;
                    }
                }
            }// if
        }// for

        int minn = INF;
        for (i = 0; i < n; i++)
        {
            if (!vis[i] && dis[i] < minn)
            {
                minn = dis[i];
                newP = i;
            }
        }
        vis[newP] = true;
    }// while
}

int main()
{
    //freopen("in", "r", stdin);
    while (scanf("%d%d%d%d", &n,&m,&s,&d) != EOF)
    {
        int i,j;
        for (i = 0; i < n; i++)
        {
            scanf("%d", &teams[i]);
        }
        for (i = 0; i < n; i++)
        {
            for (j = 0; j < n; j++)
            {
                mp[i][j] = INF;
            }
            mp[i][i] = 0;
        }
        int tmpi, tmpj,tmpdist;
        for (i = 0; i < m; i++)
        {
            scanf("%d%d%d", &tmpi, &tmpj, &tmpdist);
            mp[tmpi][tmpj] = tmpdist;
            mp[tmpj][tmpi] = tmpdist;
        }

        Dijkstra();

        printf("%d %d\n", pathCount[d], amount[d]);

        vector<int> v;
        int lenv = 0;
        v.clear();
        int vn = d;
        while (vn != -1)
        {
            v.push_back(vn);
            lenv++;
            vn = pre[vn];
        }
        printf("%d", v[lenv-1]);
        for (i = lenv-2; i >= 0; i--)
        {
            printf(" %d", v[i]);
        }
        printf("\n");
    }
    return 0;
}