灾后重建

题目背景

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 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
-1
5
4

说明

对于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

一开始先想的是SPFA；

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=300;
const int maxm=2e5;
inline int max_(int x,int y){return x>y?x:y;}
int n,m,q;
int a,b,c;
int t[maxn];
int h[maxn],hs=1,rs,p;
int e_s[maxm],e_t[maxm],e_w[maxm],e_n[maxm];
struct reserve{int s,t,w,nt;}re[maxm];
bool comp(const reserve&x,const reserve&y){return x.nt<y.nt;}
void add(int k){
    ++hs,e_s[hs]=re[k].s,e_t[hs]=re[k].t,e_w[hs]=re[k].w,e_n[hs]=h[re[k].s],h[re[k].s]=hs;
    ++hs,e_s[hs]=re[k].t,e_t[hs]=re[k].s,e_w[hs]=re[k].w,e_n[hs]=h[re[k].t],h[re[k].t]=hs;
}
int d[maxn];
int qe[maxm],head,tail;
void SPFA(int x,int y){
    memset(d,0x7f,sizeof(d));
    head=tail=0;
    d[x]=0,qe[head++]=x;
    while(head>tail){
        a=qe[tail++];
        for(int i=h[a];i;i=e_n[i])
        if(0ll+d[a]+e_w[i]<d[e_t[i]]){
            d[e_t[i]]=d[a]+e_w[i];
            qe[head++]=e_t[i];
        }
    }
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++){
        scanf("%d",&t[i]);
    }
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&a,&b,&c);
        re[rs++]=(reserve){a,b,c,max_(t[a],t[b])};
    }
    sort(re,re+rs,comp);
    scanf("%d",&q);
    while(q--){
        scanf("%d%d%d",&a,&b,&c);
        for(p;re[p].nt<=c&&p<rs;p++) add(p);
        SPFA(a,b);
        if(d[b]==d[n+1]) puts("-1");
        else printf("%d
",d[b]);
    }
    return 0;
}

然后看了一下题解，突然意识到，这可能是Floyd的翻身仗。

代码实现

#include<cstdio>
#include<cstring>
const int maxn=2e2+10;
const int maxq=5e4+10;
int n,m,q,now;
int a,b,c;
int t[maxn];
int d[maxn][maxn];
int q_u[maxq],q_v[maxq],q_t[maxq];
int main(){
    memset(d,0x7f,sizeof(d));
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++) scanf("%d",&t[i]);
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&a,&b,&c);
        d[a][b]=d[b][a]=c;
    }
    scanf("%d",&q);
    for(int i=0;i<q;i++) scanf("%d%d%d",&q_u[i],&q_v[i],&q_t[i]);
    t[n]=t[n-1]+1;
    for(int k=0;k<n;k++){
        while(now<q&&q_t[now]<t[k]){
            if(t[q_u[now]]>=t[k]||t[q_v[now]]>=t[k]||d[q_u[now]][q_v[now]]==d[n][n]) puts("-1");
            else printf("%d
",d[q_u[now]][q_v[now]]); 
            now++;
        }
        for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
        d[i][j]=0ll+d[i][k]+d[k][j]<d[i][j]?d[i][k]+d[k][j]:d[i][j];
    }
    while(now<q){
        if(d[q_u[now]][q_v[now]]==d[n][n]) puts("-1");
        else printf("%d
",d[q_u[now]][q_v[now]]); 
        now++;
    }
    return 0;
}