CCF CSP认证201712-4 行车路线

201712-4 行车路线

题目

思路

昨天可能受第三题的影响，心态有点爆炸，再加上剩的时间不多了，没有想出好的方法。最开始的想法是把小道能连通的点统统计算一下其间最小小道距离，再换算成大道，计算最短路。运行一下发现不对，这样还是会把连续的小道分开算平方。

于是想到了先用Floyd算出走小道的全源最短路，再用它更新大道的最短路，这样只要保证更新的时候每个点只能出现一次（用一次小道更新），就能得到正确结果。可是更新的情况太多了，不太好实现，时间复杂度也高，于是我没写下去了。时间要不够了，我改了改代码，解决不存在小道和小道不相交的40%用例，就提交了。（然而ccf评测机罢工了…）

今天上午再想了想，很快就有方案了。
dijkstra最短路算法，更新两次d[]数组，分别用大道和小道更新，设置一个pre[]数组，pre[i]表示使d[i]最小的路径上i前面的连续小道长度。详见代码。

AC代码如下

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
const long long inf=1e18;
int n,m;
long long d[505],dis[3][505][505],pre[505];
bool vis[505]={false};

long long dijkstra(){
 	fill(d,d+n+1,inf);
 	memset(pre,0,sizeof(pre));
 	memset(vis,0,sizeof(vis));
 	d[1]=0;
 	for(int i=1;i<=n;i++){
  		int u=-1;
  		long long Min=inf;
  		for(int j=1;j<=n;j++){//找集合外最近点
   			if(vis[j]==false&&d[j]<Min){
    				Min=d[j];
    				u=j;
   			}
  		}
  		if(u==n) return d[n];//到达n的最小花费算出之后退出
  		vis[u]=true;
  		for(int j=1;j<=n;j++){
   			if(vis[j]==false){
    				if(d[j]>d[u]+dis[0][u][j]){//如果大道能更新小花费
     					d[j]=d[u]+dis[0][u][j];
     					pre[j]=0;//pre[u]清0
    				}
    				long long l=dis[1][u][j]+pre[u];
    				if(l<inf&&d[j]>d[u]-pre[u]*pre[u]+l*l){//如果小道能更新最小花费
     					d[j]=d[u]+l*l-pre[u]*pre[u];//l平方-pre平方等于增加的花费
     					pre[j]=l;//记录连续小道距离
	    			}   			
	    		}
  		}
 	}
}

int main(){
 	scanf("%d%d",&n,&m);
 	int t,a,b,c;
 	fill(dis[0][0],dis[0][0]+3*505*505,inf);
 	for(int i=0;i<m;i++){
  		scanf("%d%d%d%d",&t,&a,&b,&c);
  		if(dis[t][a][b]>c) dis[t][a][b]=dis[t][b][a]=c;//需要考虑有重边的情况
 	}
 	printf("%lld",dijkstra());
 	return 0;
}