【SSL1624】小萨的烦恼【floyed最短路变式】

Description

小萨将给出一张地图，它可以看做一个有N（N<=100）个结点的图。这张地图有N个公共汽车站，小萨只可以在这N个汽车站上公共汽车。有些车站之间存在一条双向通路，无论是公共汽车还是小萨，都只能走这些通路。若两个车站之间的距离为d，步行所需要的时间为2*d秒，坐公共汽车所需要的时间为d秒。小萨他们只有T秒的时间，他希望你求出他和他的MM出去约会后能否及时赶回来，若不能，则输出“You are day dreaming!”，否则输出他们所需要花费的最少时间。
注意：一张公共汽车车票可以使用两次（只可以搭乘同一辆公共汽车），即可以认为小萨去约会地点的时间和回来的时间相等。

题意：

有一个n*n的地图，要从点1到点s，有一次时间减半的机会，问最短时间。

Input

第一行有三个整数N、T、S
接下来是一个N*N的邻接矩阵。两两车站之间的距离不超过10^9。A[I,j]为0表示I和J车站不连通。

Output

若小萨他们不能在限定时间内赶回学校，那么输出“You are day dreaming!”（不包括引号）
否则，输出一个整数，为他们所需要花费的最少时间。

Sample Input

4 5 4
0 1 1 1
1 0 1 1
1 1 0 1
1 1 1 0

Sample Output

2

Hint

【数据范围】
20%的数据：N<=10
100%的数据：N<=70 T<=10^9

分析

这题也要用两个邻接矩阵（一个走路一个坐车）。先跑一遍最短路，然后把每一条边折半，最后再枚举最合适的地方，连接起点和最合适的地方取一个最小值，而这个最小值就是答案了。

上代码

#include
#include
#include
using namespace std;
typedef long long LL;
LL n,t,s,g[101][101],ans,a[101][101];
int main()
{
	ios::sync_with_stdio(false);
	cin>>n>>t>>s;
	for(register LL i=1;i<=n;i++)
	{
		for(register LL j=1;j<=n;j++)
		{
			cin>>g[i][j];
			if(!g[i][j]) g[i][j]=0x7fffffff;
			if(i==j) g[i][j]=0;
			a[i][j]=g[i][j]*2; //走路两倍时间
		}
	}
	for(register LL k=1;k<=n;k++)
	  for(register LL i=1;i<=n;i++)
	    for(register LL j=1;j<=n;j++)
	      a[i][j]=min(a[i][j],a[i][k]+a[k][j]);//坐车
	ans=0x7fffffff;
	for(register LL i=1;i<=n;i++)
	  for(register LL j=1;j<=n;j++)
     	ans=min(ans,g[i][j]+a[1][i]+a[j][s]);  //走路
	if(2*ans>t/*往返如果超时*/) 
	{
		std::cout<<"You are day dreaming!" ;
	}
	else std::cout<<ans*2;
	return 0; 
}