洛谷P1119 灾后重建

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 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。

#include<iostream>
#include<cstdio>
#include<deque>
#define maxn 201
using namespace std;
int n,m,t[maxn],num,head[maxn],dis[maxn],Num,Head[maxn];
bool vis[maxn];
struct node{
    int to,pre,v;
}e[maxn*(maxn-1)],E[maxn*(maxn-1)];
void Insert(int from,int to,int v){
    e[++num].to=to;
    e[num].v=v;
    e[num].pre=head[from];
    head[from]=num;
}
void Insert2(int from,int to,int v){
    E[++Num].to=to;
    E[Num].v=v;
    E[Num].pre=Head[from];
    Head[from]=Num;
}
int find(int x){//找t数组中<=x的数的最后一个位置 
    int l=1,r=n,res=0;
    while(l<=r){
        int mid=(l+r)>>1;
        if(t[mid]<=x)res=mid,l=mid+1;
        else r=mid-1;
    }
    return res;
}
int spfa(int s,int t){
    for(int i=1;i<=n;i++)dis[i]=0x3f3f3f3f;
    deque<int>q;
    q.push_back(s);vis[s]=1;dis[s]=0;
    while(!q.empty()){
        int now=q.front();q.pop_front();vis[now]=0;
        for(int i=head[now];i;i=e[i].pre){
            int to=e[i].to;
            if(dis[to]>dis[now]+e[i].v){
                dis[to]=dis[now]+e[i].v;
                if(!vis[to]){
                    vis[to]=1;
                    if(!q.empty()&&dis[to]<dis[q.front()])q.push_front(to);
                    else q.push_back(to);
                }
            }
        }
    }
    if(dis[t]==0x3f3f3f3f)return -1;
    return dis[t];
}
int main(){
    freopen("Cola.txt","r",stdin);
    scanf("%d%d",&n,&m);
    int x,y,z;
    for(int i=1;i<=n;i++)scanf("%d",&t[i]);
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&x,&y,&z);
        x++;y++;
        Insert2(x,y,z);Insert2(y,x,z);
    }
    int q,p=0;
    for(int i=1;i<=n;i++){
        if(t[i]>0)break;
        for(int j=Head[i];j;j=E[j].pre){
            int to=E[j].to;
            if(t[to]==0)Insert(i,to,E[j].v),Insert(to,i,E[j].v);
        }
    }
    scanf("%d",&q);
    for(int i=1;i<=q;i++){
        scanf("%d%d%d",&x,&y,&z);
        int l=find(p),r=find(z);p=z;
        l++;
        for(int j=l;j<=r;j++){
            for(int k=Head[j];k;k=E[k].pre){
                int to=E[k].to;
                if(t[to]<=z)Insert(j,to,E[k].v),Insert(to,j,E[k].v);
            }
        }
//        cout<<num<<endl;
        printf("%d
",spfa(x+1,y+1));
    }
}

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn 201
#define INF 0x3fffffff
using namespace std;
int n,m,q,dis[maxn][maxn],t[maxn],cnt;
int main(){
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++)scanf("%d",&t[i]);
    for(int i=0;i<n;i++)for(int j=0;j<n;j++)dis[i][j]=INF;
    int x,y,z;
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&x,&y,&z);
        dis[x][y]=dis[y][x]=z;
    }
    scanf("%d",&q);
    for(int query=1;query<=q;query++){
        scanf("%d%d%d",&x,&y,&z);
        while(t[cnt]<=z&&cnt<n){//在线floyed 
            for(int i=0;i<n;i++)
                for(int j=0;j<n;j++)
                    dis[i][j]=min(dis[i][j],dis[i][cnt]+dis[cnt][j]);
            cnt++;
        }
        if(dis[x][y]==INF||t[x]>z||t[y]>z){
            puts("-1");
            continue;
        }
        printf("%d
",dis[x][y]);
    }
    return 0;
}