您的位置:首页 > 其它

GYM 101061 C.Ramzi(SPFA)

2017-03-25 09:27 309 查看
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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: