P1119 灾后重建 最短路 floyd算法

题目：


思路：
一看这道题就是个多源最短路问题，考虑采用floyd算法
如果每次都用一次floyd，显然大数据会tle
所以先一遍把所有的情况都算出来，后面再去调用
如果day不是递增的数组，需要先排序
AC代码：

#include<cstdio>
#include <iostream>
#include<cstring>
#include<algorithm>
#include <queue>
#include <vector>
#include <cmath>
using namespace std;
const int maxn=211;
int n,m,que,day[maxn],f[maxn][maxn][maxn];
void floyd(int x,int y,int t)
{
    for(int k=1;k<=n;k++)
    {
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                if(day[k]<=t)
                {
                    f[k][i][j]=min(f[k-1][i][j],f[k-1][i][k]+f[k-1][k][j]);
                    //cout<<"k="<<k<<" "<<"i="<<i<<" "<<"j="<<j<<" "<<"dis[k][i][j]="<<dis[k][i][j]<<endl;
                }
            }
        }
    }
}
int main()
{
	cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>day[i];
    memset(f,0x3f,sizeof(f));
    for(int i=1;i<=m;i++)
    {
        int u,v,w;
        cin>>u>>v>>w;
        u++;
        v++;
        f[0][u][v]=f[0][v][u]=w;
    }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++) f[i][j][j]=0;
    floyd(n,n,day[n]);
    cin>>que;
    for(int i=1;i<=que;i++)
    {
        int x,y,t;
        cin>>x>>y>>t;
        x++;
        y++;
        if(day[x]>t||day[y]>t) cout<<-1<<endl;
        else 
        {
            int temp=0;
            for(int j=1;j<=n;j++) if(day[j]<=t) temp=j;
            if(f[temp][x][y]==0x3f3f3f3f) cout<<-1<<endl;
            else cout<<f[temp][x][y]<<endl;
        }
    }
	return 0;
}

复习版：
复习的过程中，还是发现了不少容易错误的点，首先就是由于每次都进行floyd会导致tle。所以其实可以一次把所有情况都算出来，但因为t有限制，所以f必须保留成三维数组不能精简成二维。同时，考虑f[k][i][j]的含义是，i-j的路径里，经过1-k的点，所以i,j的数组都得从1，1开始存，而f[0][i][j]则为i-j的初始路径长度，还得注意，当i=j时，f[k][i][j]=0

#include <bits/stdc++.h>
using namespace std;
const int maxn=211;
int n,m,a[maxn],f[maxn][maxn][maxn],q;
void floyd(int x,int y)
{
    
    for(int k=1;k<=n;k++)
    {
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                f[k][i][j]=min(f[k-1][i][j],f[k-1][i][k]+f[k-1][k][j]);
                //cout<<"i="<<i<<" "<<"j="<<j<<" "<<f[i][j]<<endl;
            }
        }
    }
    
}
int main()
{
	cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>a[i];
    memset(f,0x3f3f3f3f,sizeof(f));
    for(int i=1;i<=m;i++)
    {
        int x,y,z;
        cin>>x>>y>>z;
        x++;
        y++;
        f[0][x][y]=f[0][y][x]=z;
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++) f[i][j][j]=0;
    }
    floyd(n,n);
    cin>>q;
    for(int i=1;i<=q;i++)
    {
        int x,y,t;
        cin>>x>>y>>t;
        x++;
        y++;
        if(a[x]>t||a[y]>t) 
        {
            cout<<-1<<endl;
        }
        else
        {
            int temp=0;
            for(int j=1;j<=n;j++) if(a[j]<=t) temp=j;
            if(f[temp][x][y]==0x3f3f3f3f) cout<<-1<<endl;
            else cout<<f[temp][x][y]<<endl;
        }
    }
	return 0;
}