hdu4396 spfa有限制的最短路

利用二维的spfa

#include <iostream>
#include <cstdio>
using namespace std;
const int oo=1e9;
const int mm=5555;
const int mn=200020;
int ver[mn],next[mn],cost[mn];
int head[mm],qx[mn],qy[mn],dis[mm][55];
bool vis[mm][55];
int edge;
void addedge(int u,int v,int c)
{
    ver[edge]=v,cost[edge]=c,next[edge]=head[u],head[u]=edge++;
    ver[edge]=u,cost[edge]=c,next[edge]=head[v],head[v]=edge++;
}
int main()
{
    int n,m,i,j,l,r,s,t,k,c,z,u,v;
    while(~scanf("%d%d",&n,&m))
    {
        for(edge=i=0;i<=n;i++)   head[i]=-1;
        while(m--)
        {
            scanf("%d%d%d",&u,&v,&c);
            addedge(u,v,c);
        }
        scanf("%d%d%d",&s,&t,&k);
        k=k/10+((k%10)>0);
        for(i=1;i<=n;i++)
            for(j=0;j<=k;j++)
                dis[i][j]=1e9,vis[i][j]=0;
        dis[s][0]=0;r=1;
        qx[0]=s,qy[0]=0;
        for(l=0;l<r;++l)
            for(i=head[u=qx[l]],j=qy[l],vis[u][j]=0;i>=0;i=next[i])
                if(dis[v=ver[i]][z=j+(j<k)]>dis[u][j]+cost[i])
                {
                    dis[v][z]=dis[u][j]+cost[i];
                    if(vis[v][z])continue;
                    vis[v][z]=1;
                    qx[r]=v,qy[r]=z,++r;
                }
        printf("%d\n",(dis[t][k]<1e9)?dis[t][k]:-1);
    }
    return 0;
}