洛谷P1119灾后重建

题目背景

B地区在地震过后，所有村庄都造成了一定的损毁，而这场地震却没对公路造成什么影响。但是在村庄重建好之前，所有与未重建完成的村庄的公路均无法通车。换句话说，只有连接着两个重建完成的村庄的公路才能通车，只能到达重建完成的村庄。

题目描述

给出B地区的村庄数N，村庄编号从0到N-1，和所有M条公路的长度，公路是双向的。并给出第i个村庄重建完成的时间t[i]，你可以认为是同时开始重建并在第t[i]天重建完成，并且在当天即可通车。若t[i]为0则说明地震未对此地区造成损坏，一开始就可以通车。之后有Q个询问(x, y, t)，对于每个询问你要回答在第t天，从村庄x到村庄y的最短路径长度为多少。如果无法找到从x村庄到y村庄的路径，经过若干个已重建完成的村庄，或者村庄x或村庄y在第t天仍未重建完成 ，则需要返回-1。

输入输出格式

输入格式：

输入文件rebuild.in的第一行包含两个正整数N，M，表示了村庄的数目与公路的数量。

第二行包含N个非负整数t[0], t[1], …, t[N – 1]，表示了每个村庄重建完成的时间，数据保证了t[0] ≤ t[1] ≤ … ≤ t[N – 1]。

接下来M行，每行3个非负整数i, j, w，w为不超过10000的正整数，表示了有一条连接村庄i与村庄j的道路，长度为w，保证i≠j，且对于任意一对村庄只会存在一条道路。

接下来一行也就是M+3行包含一个正整数Q，表示Q个询问。

接下来Q行，每行3个非负整数x, y, t，询问在第t天，从村庄x到村庄y的最短路径长度为多少，数据保证了t是不下降的。

输出格式：

输出文件rebuild.out包含Q行，对每一个询问(x, y, t)输出对应的答案，即在第t天，从村庄x到村庄y的最短路径长度为多少。如果在第t天无法找到从x村庄到y村庄的路径，经过若干个已重建完成的村庄，或者村庄x或村庄y在第t天仍未修复完成，则输出-1。

输入输出样例

输入样例#1：

4 51 2 3 40 2 12 3 13 1 22 1 40 3 542 0 20 1 20 1 30 1 4

输出样例#1：

-1-154

说明

对于30%的数据，有N≤50；

对于30%的数据，有t[i] = 0，其中有20%的数据有t[i] = 0且N＞50；

对于50%的数据，有Q≤100；

对于100%的数据，有N≤200，M≤N*(N-1)/2，Q≤50000，所有输入数据涉及整数均不超过100000。

我看见数据规模不是很大，果断用了 floyd ，加优化，速度还是可以的。

附代码：

#include<iostream>#include<algorithm>#include<cstdio>#define MAX 9999999#define MAXN 210#define MAXM 50010using namespace std;int n,m,s,times[MAXN],a[MAXN][MAXN];struct node{       int x,y,t;}que[MAXM];inline int read(){       int date=0,w=1;char c=0;       while(c!='-'&&(c<'0'||c>'9'))c=getchar();       if(c=='-'){w=-1;c=getchar();}       while(c>='0'&&c<='9'){date=date*10+c-'0';c=getchar();}       return date*w;}void floyd(){     int now=1,ans;     for(int k=0;k<n;k++){             while(now<=s&&que[now].t<times[k]){             ans=a[que[now].x][que[now].y];             if(times[que[now].x]>=times[k]||times[que[now].y]>=times[k])             ans=-1;             printf("%d\n",(ans==a[n][n])?-1:(ans==MAX?-1:ans));             now++;             }             for(int i=0;i<n;i++)             for(int j=0;j<n;j++)             a[i][j]=min(a[i][j],a[i][k]+a[k][j]);             }     while(now<=s){                   ans=a[que[now].x][que[now].y];                   printf("%d\n",ans==a[n][n]?-1:(ans==MAX?-1:ans));                   now++;                   }}int main(){    int u,v,w;    n=read();m=read();    for(int i=0;i<n;i++)    times[i]=read();    for(int i=0;i<=n;i++)    for(int j=0;j<=n;j++)    a[i][j]=(i==j)?0:MAX;    for(int i=1;i<=m;i++){            u=read();v=read();w=read();            a[u][v]=a[v][u]=w;            }    s=read();    for(int i=1;i<=s;i++){            que[i].x=read();            que[i].y=read();            que[i].t=read();            }    times[n]=times[n-1]+1;    floyd();    return 0;}