【洛谷1119】灾后重建 最短路

灾后重建

题目背景

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我也很吃惊……

看一眼t好像有玄机，是递增的。所以对于每次提问都将能够已经修好的点用floyd来更新距离。由于n很小所以能过。

 

在线spfa60分版

 

#include<iostream>
#include<queue>
#include<cstdio>
#include<cstring>
#define INF 0x3f3f3f3f
#define N 10100
using namespace std;

int to[N],c[N],head[N],next[N],dist[N],fix[N],t[N],visit[N];
int n,m,qu,tot=0;

inline int read()
{
    int ret=0;char ch=getchar();
    while(ch<'0' || ch>'9') ch=getchar();
    while(ch>='0' && ch<='9') ret=ret*10+ch-'0',ch=getchar();
    return ret;
}

void add_edge(int x,int y,int z)
{
    to[++tot]=y;
    c[tot]=z;
    next[tot]=head[x];
    head[x]=tot;
}
queue<int> q;
void spfa(int o)
{
    for (int i=0;i<=n-1;i++) dist[i]=INF,visit[i]=0;
    dist[o]=0;visit[o]=1;q.push(o);
    while(!q.empty())
    {
        int x=q.front();q.pop();visit[x]=0;
        for (int i=head[x];i;i=next[i])
        {
            int y=to[i];
            if (fix[y] && dist[y] > dist[x]+c[i])
            {
                dist[y]=dist[x]+c[i];
                if (!visit[y]) q.push(y),visit[y]=1;
            }
        }
    }
}

int main()
{
    memset(fix,0,sizeof(fix));
    n=read();m=read();
    for (int i=0;i<n;i++) t[i]=read();
    for (int i=0;i<m;i++)
    {
        int x=read(),y=read(),z=read();
        add_edge(x,y,z);
        add_edge(y,x,z);
    }
    qu=read();
    for (int i=0;i<qu;i++)
    {
        int a=read(),b=read(),d=read();
        for (int i=0;i<n;i++)
            if (t[i] <= d) fix[i]=1;
        if (!fix[a] || !fix[b]) {puts("-1");continue;}
        spfa(a);
        if (dist[b] != INF) printf("%d\n",dist[b]);
        else puts("-1\n");
    }
    return 0;
}

 

 

 

Floyd正解100分版

 

#include<iostream>
#include<cstdio>
#include<cstring>
#define M 50050
#define N 250
#define INF 0x3f3f3f3f
using namespace std;

int t[M];
int dist[N][N];
int n,m,tot,cnt,qu;

int main()
{
    scanf("%d%d",&n,&m);
    memset(t,INF,sizeof(t));cnt=0;
    memset(dist,INF,sizeof(dist));
    for (int i=0;i<n;i++) dist[i][i]=0;
    for (int i=0;i<n;i++) scanf("%d",&t[i]);
    for (int i=0;i<m;i++)
    {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        dist[x][y]=dist[y][x]=z;
    }
    scanf("%d",&qu);
    while(qu--)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        while(t[cnt] <= c)
        {
            for (int i=0;i<n;i++)
                for (int j=0;j<n;j++)
                    dist[i][j]=min(dist[i][j],dist[i][cnt]+dist[cnt][j]);
            cnt++;
        }
        if (dist[a][b] == INF || t[a]>c || t[b]>c) puts("-1");
        else printf("%d\n",dist[a][b]);
    }
    return 0;
}