hdu 6331 Walking Plan （至少经过k条路的最短路）

http://acm.hdu.edu.cn/showproblem.php?pid=6331

题意：至少经过k条路的最短路

思路：因为k最大是10000的如果直接，用矩阵跑floyd k次的话铁定TLE，然后我们可以用a数组来记录两个点经过100以内的边的最短路，b数组来记录整百的路径数的最短路，这样就极大优化了时间和空间，a数组就用图中所给的路径来跑矩阵floyd100次，b数组就用a[100]来跑矩阵floyd100次，但是这样b数组不一定是最短的，所以在用每两个点的最短距离更新一波，保证最短距离，最后枚举中间点来找到最短距离。

代码

#include<bits/stdc++.h>
using namespace std;
const int inf=0x3f3f3f3f;
int n,m,k,q;
int a[105][55][55],b[105][55][55],dp[55][55],f[55][55],g[55][55];
void mul(int a[][55],int b[][55],int c[][55])
{
    static int z[55][55];
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            z[i][j]=inf;
            for(int k=1;k<=n;k++)
            {
                z[i][j]=min(z[i][j],a[i][k]+b[k][j]);
            }
        }
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            c[i][j]=z[i][j];
        }
    }
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        memset(g,inf,sizeof(g));
        int u,v,w;
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            g[u][v]=min(g[u][v],w);
        }
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                if(i==j)
                {
                    a[0][i][j]=b[0][i][j]=0;
                }
                else a[0][i][j]=b[0][i][j]=inf;
            }
        }
        for(int i=1;i<105;i++)
        {
            mul(a[i-1],g,a[i]);
        }
        for(int i=1;i<105;i++)
        {
            mul(b[i-1],a[100],b[i]);
        }
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                if(i==j)dp[i][j]=0;
                else dp[i][j]=g[i][j];
            }
        }
        for(int k=1;k<=n;k++)
        {
            for(int i=1;i<=n;i++)
            {
                for(int j=1;j<=n;j++)
                {
                    dp[i][j]=min(dp[i][j],dp[i][k]+dp[k][j]);
                }
            }
        }
        for(int x=0;x<105;x++)
        {
            for(int i=1;i<=n;i++)
            {
                for(int j=1;j<=n;j++)
                    f[i][j]=inf;
            }
            for(int i=1;i<=n;i++)
            {
                for(int j=1;j<=n;j++)
                {
                    for(int k=1;k<=n;k++)
                    {
                        f[i][j]=min(f[i][j],b[x][i][k]+dp[k][j]);
                    }
                }
            }
            for(int i=1;i<=n;i++)
            {
                for(int j=1;j<=n;j++)
                    b[x][i][j]=f[i][j];
            }
        }
        scanf("%d",&q);
        while(q--)
        {
            scanf("%d%d%d",&u,&v,&w);
            int x=w%100,y=w/100;
            int ans=inf;
            for(int k=1;k<=n;k++)
               ans=min(ans,a[x][u][k]+b[y][k][v]);
            if(ans>=inf)
                printf("-1\n");
            else printf("%d\n",ans);
        }
    }
    return 0;
}