PTA--旅游规划--SPFA

  有了一张自驾旅游路线图，你会知道城市间的高速公路长度、以及该公路要收取的过路费。现在需要你写一个程序，帮助前来咨询的游客找一条出发地和目的地之间的最短路径。如果有若干条路径都是最短的，那么需要输出最便宜的一条路径。

输入格式:

输入说明：输入数据的第1行给出4个正整数NN、MM、SS、DD，其中NN（2\le N\le 5002≤N≤500）是城市的个数，顺便假设城市的编号为0~(N-1N−1)；MM是高速公路的条数；SS是出发地的城市编号；DD是目的地的城市编号。随后的MM行中，每行给出一条高速公路的信息，分别是：城市1、城市2、高速公路长度、收费额，中间用空格分开，数字均为整数且不超过500。输入保证解的存在。

输出格式:

在一行里输出路径的长度和收费总额，数字间以空格分隔，输出结尾不能有多余空格。

输入样例:

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

输出样例:

3 40

------------------------------

分析：多了一个判断收费，但是想法跟路径的权一样；我称为最短路径的多权算法

//SPFA：队列优化的Bellman-Ford算法；
#include<iostream>
#include<cstdlib>
#include<queue>
using namespace std;
int main(){
    int flag = 0; 
    int infinity = 99999999;
    int n,m,s,d;
    cin>>n>>m>>s>>d;
    int *u;
    int *v;
    int *len;
    int *cost;
    u =  (int *)malloc(m*sizeof(int));
    v =  (int *)malloc(m*sizeof(int));
    len =  (int *)malloc(m*sizeof(int));
    cost =  (int *)malloc(m*sizeof(int));
    int *first;
    int *next;
    first = (int *)malloc(n*sizeof(int));
    next = (int * )malloc(m*sizeof(int));
    for(int i = 0; i < n; i++) first[i] = -1;
    
    for(int i =0;i < m; i++){
        cin>>u[i]>>v[i]>>len[i]>>cost[i];
        next[i] = first[u[i]];
        first[u[i]] = i;
        if(u[i] == s && v[i] == d) flag = 1;
    }
    if(!flag){
        s = s^d;
        d = s^d;
        s = s^d;
    }
    int *dis;
    dis = (int *)malloc(n*sizeof(int));
    int *count;
    count = (int *)malloc(n*sizeof(int));
    for(int i = 0; i < n; i++){
        dis[i] = infinity;
        count[i] = infinity;
    }
    dis[s] = 0;
    count[s] = 0;
    queue<int> que;
    que.push(s);
    int *book;
    book = (int *)malloc(n*sizeof(int));
    for(int i = 0; i < n; i++) book[i] = 0;
    book[s] = 1;
    while(!que.empty()){
        int k = first[que.front()];
        while(k != -1){
            if(dis[u[k]] + len[k] < dis[v[k]]){
                dis[v[k]] = dis[u[k]] + len[k];
                count[v[k]] = count[u[k]] + cost[k];
                if(book[v[k]] == 0){
                    que.push(v[k]);
                    book[v[k]] = 1;
                }
            }
            else if((dis[u[k]] + len[k] == dis[v[k]]) && (count[u[k]] + cost[k] < count[v[k]])){
                count[v[k]] = count[u[k]] + cost[k];
                if(book[v[k]] == 0){
                    que.push(v[k]);
                    book[v[k]] = 1;
                }
            }
            k = next[k];
        }
        book[que.front()] = 0;
        que.pop();
    }
    cout<<dis[d]<<' '<<count[d]<<endl;
    return 0;
}