GYM 101061 C.Ramzi（SPFA）

Description
一张n个点的图，有两种边，第一种是步行道，第二种是车行道（在车行道坐车故步行距离为0），现在要从起点s到终点e，问步行的最短距离，如果有多种方案使得步行距离最短则要求车行最短
Input
第一行一整数T表示用例组数，每组用例首先输入点数n和边数m，之后m行每行输入四个整数u,v,w,k表示u和v之间有一条边权为w的边，k=1表示该条边是步行道，k=2表示该条边是车行道(1<=n<=100,1<=m<=n*(n-1),0<=w<=1e4)
Output
输出最短步行距离和在步行距离最短的情况下的最短总距离，如果不可达则输出-1
Sample Input
3
3 3
1 2 5 1
1 3 5 2
3 2 4 1
1 2
2 2
1 2 5 2
1 2 3 1
1 2
3 1
1 2 5 1
1 3
Sample Output
4 9
0 5
-1
Solution
最短路变形，以dis[i][0]和dis[i][1]分别表示起点到i点的步行最短距离和起点到i点的包含车行和步行的最短距离，那么在spfa每次松弛时首先以dis[i][0]来松弛，如果dis[i][0]不能被松弛则以dis[i][1]来松弛，最后dis[e][0]和dis[e][1]即为答案
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 111
#define maxm 22222 
struct edge
{
    int to,next,cost,flag;
}g[maxm];
int head[maxm],tol;
int dis[maxn][2];//所有点到起点的最短距离
int T,n,m;
void init()//初始化 
{
    memset(head,-1,sizeof(head));
    tol=0;
}
void add(int u,int v,int c,int flag)//单向边,从u到v,权值为c 
{
    g[tol].cost=c,g[tol].to=v,g[tol].next=head[u],g[tol].flag=flag,head[u]=tol++;
}
void spfa(int s)//单源最短路,s是起点
{
    bool vis[maxn];
    memset(vis,false,sizeof(vis));
    queue<int>que;
    for(int i=1;i<=n;i++)dis[i][0]=dis[i][1]=INF;
    dis[s][0]=dis[s][1]=0;
    vis[s]=true;
    que.push(s);
    while(!que.empty())
    {
        int u=que.front();
        que.pop();
        vis[u]=false;
        for(int i=head[u];i!=-1;i=g[i].next)
        {
            int v=g[i].to,c=g[i].cost,flag=g[i].flag;
            int c1=0,c2=c;
            if(flag==1)c1=c;
            if(dis[v][0]>dis[u][0]+c1)
            {
                dis[v][0]=dis[u][0]+c1;
                dis[v][1]=dis[u][1]+c2;
                if(!vis[v])
                {
                    vis[v]=true;
                    que.push(v);
                }
            }
            else if(dis[v][0]==dis[u][0]+c1&&dis[v][1]>dis[u][1]+c2)
            {
                dis[v][1]=dis[u][1]+c2;
                if(!vis[v])
                {
                    vis[v]=true;
                    que.push(v);
                }
            }
        }
    }
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        init();
        for(int i=1;i<=m;i++)
        {
            int u,v,w,flag;
            scanf("%d%d%d%d",&u,&v,&w,&flag);
            add(u,v,w,flag),add(v,u,w,flag);
        }
        int s,e;
        scanf("%d%d",&s,&e);
        spfa(s);
        if(dis[e][0]==INF)printf("-1\n");
        else printf("%d %d\n",dis[e][0],dis[e][1]);
    }
    return 0;
}