PAT1030

http://pat.zju.edu.cn/contests/pat-a-practise/1030

题目大意：求最短路径，并计算最短路径边权重之和，若最短路径不唯一，取边权重之和最小者。类似之前PAT的1018，不过比那个简单点，

用的是map<int, vector<pair<int, vector<int>>>> info; 来存储相应点的信息，每条最短路径的cost 以及 最短路径所经过的结点（不包括自身）。

用的方法和之前1018基本一样。一个Dijkstra就够！

#include<iostream>
#include<vector>
#include<utility>
#include<map>
#include<algorithm>
using namespace std;
#define INF 0x1fffffff

bool comp(pair<int, vector<int>> p1, pair<int, vector<int>> p2)
{
    if(p1.first < p2.first)
        return true;
    else
        return false;
}

void Dijkstra_version(vector<vector<int>> &tmap, vector<vector<int>> &cost, 
                                map<int, vector<pair<int, vector<int>>>> &info, vector<int> &dist, int S, int D)
{
    dist[S] = 0;
    vector<int> used(dist.size(), 0);
    /*初始化起点*/
    vector<int> v;
    pair<int, vector<int>> p(0,v);
    info[S].push_back(p);
    while(S != D)
    {
        int kk(-1), min(INF);
        for(int i=0; i<dist.size(); ++i)
            if(used[i] == 0 && dist[i] < min)
            {
                min = dist[i];
                kk = i;
            }
        used[kk] = 1;
        for(int i=0; i<dist.size(); ++i)
            if(used[i] == 0)
                if(tmap[kk][i] + dist[kk] < dist[i])
                {
                    dist[i] = tmap[kk][i] + dist[kk];
                    info[i] = info[kk];
                    for(int j=0; j<info[i].size(); ++j)
                    {
                        /*更新cost*/
                        info[i][j].first += cost[kk][i];
                        /*将结点假如路径中*/
                        info[i][j].second.push_back(kk);
                    }
                }
                else if(tmap[kk][i] + dist[kk] == dist[i])
                {
                    for(int j=0; j<info[kk].size(); ++j)
                    {
                        info[i].push_back(info[kk][j]);
                        info[i][info[i].size()-1].first += cost[kk][i];
                        info[i][info[i].size()-1].second.push_back(kk);
                    }
                }
        S = kk;
    }            
}

int main()
{
    int N,M,S,D;
    while(cin>>N>>M>>S>>D)
    {
        vector<int> colum(N, INF);
        vector<vector<int>> tmap(N, colum);
        vector<vector<int>> cost(N, colum);
        for(int i=0; i<M; ++i)
        {
            int j,k,d,c; cin>>j>>k>>d>>c;
            tmap[j][k]=tmap[k][j]=d;
            cost[j][k]=cost[k][j]=c;
        }
        /*记录信息*/
        map<int, vector<pair<int, vector<int>>>> info;
        vector<int> dist(N, INF);
        Dijkstra_version(tmap, cost, info, dist, S, D);
        /*对所有最短路径按cost从小到大排序*/
        sort(info[D].begin(), info[D].end(), comp);
        for(int i=0; i<info[D][0].second.size(); ++i)
            cout<<info[D][0].second[i]<<" ";
        cout<<D<<" "<<dist[D]<<" "<<info[D][0].first<<endl;
    }
}