P1119 灾后重建
题目背景
B地区在地震过后,所有村庄都造成了一定的损毁,而这场地震却没对公路造成什么影响。但是在村庄重建好之前,所有与未重建完成的村庄的公路均无法通车。换句话说,只有连接着两个重建完成的村庄的公路才能通车,只能到达重建完成的村庄。
题目描述
给出B地区的村庄数N,村庄编号从000到N−1,和所有MMM条公路的长度,公路是双向的。并给出第i个村庄重建完成的时间ti,你可以认为是同时开始重建并在第ti天重建完成,并且在当天即可通车。若ti为000则说明地震未对此地区造成损坏,一开始就可以通车。之后有Q个询问(x,y,t),对于每个询问你要回答在第t天,从村庄x到村庄y的最短路径长度为多少。如果无法找到从x村庄到y村庄的路径,经过若干个已重建完成的村庄,或者村庄x或村庄y在第t天仍未重建完成 ,则需要返回−1。
输入格式
第一行包含两个正整数N,M,表示了村庄的数目与公路的数量。
第二行包含NNN个非负整数t0,t1,…,tN−1,表示了每个村庄重建完成的时间,数据保证了t0≤t1≤…≤tN−1。
接下来M行,每行3个非负整数i,j,w为不超过10000的正整数,表示了有一条连接村庄i与村庄jjj的道路,长度为w,保证i≠ji≠j,且对于任意一对村庄只会存在一条道路。
接下来一行也就是M+3行包含一个正整数Q,表示Q个询问。
接下来Q行,每行333个非负整数x,y,tx,询问在第t天,从村庄x到村庄y的最短路径长度为多少,数据保证了t是不下降的。
输出格式
共Q行,对每一个询问(x,y,t)输出对应的答案,即在第ttt天,从村庄x到村庄y的最短路径长度为多少。如果在第t天无法找到从x村庄到y村庄的路径,经过若干个已重建完成的村庄,或者村庄x或村庄y在第t天仍未修复完成,则输出−1。
输入输出样例
输入
4 5
1 2 3 4
0 2 1
2 3 1
3 1 2
2 1 4
0 3 5
4
2 0 2
0 1 2
0 1 3
0 1 4
输出
-1
-1
5
4
这里要注意询问所给出的时间是递增的,每个村庄所用时间也是递增的,这就可以给本题进行有个小小的优化,让原来Floyed算法O(QN^3)优化到O(N^3+Q),不然会超时,具体操作就是对中转点k进行优化,直接看代码里的注释
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1005;
const int inf = 0x3f3f3f3f;
int t[maxn];
int f[maxn][maxn];
int n,m,q;
int x,y,tt;
int main()
{
fill(f[0],f[0]+maxn*maxn,inf);
fill(t,t+maxn,inf);
scanf("%d %d",&n,&m);
for(int i = 0 ; i < n ; i++)
scanf("%d",&t[i]);
for(int i = 0 ; i < m ; i++)
{
int u,v,w;
scanf("%d %d %d",&u,&v,&w);
f[u][v] = f[v][u] = w;
}
scanf("%d",&q);
int k = 0;
while(q--)
{
scanf("%d %d %d",&x,&y,&tt);
while(t[k] <= tt)//由于询问是递增的,所以后面的时间要想满足条件,必须要在前面的条件满足的基础上进行计算,这就让算法的复杂度大大降低了
{
for(int i = 0 ; i < n ; i++)
{
for(int j = 0 ; j < n ; j++)
{
if(f[i][j] >= f[i][k] + f[k][j]) //k是中转点
f[i][j] = f[i][k] + f[k][j];
}
}
k++;
}
if(f[x][y] == inf || t[x] > tt || t[y] > tt) //判断是否满足条件
printf("-1\n");
else
printf("%d\n",f[x][y]);
}
return 0;
}