图论专项shortest_paths：UVa 10269

来源：互联网发布：淘宝好的军品店编辑：程序博客网时间：2024/05/21 00:00
3维SPFA，注意状态转移的细节即可。
#include <iostream>#include <cstring>#include <cstdio>#include <queue>#include <algorithm>using namespace std;const int maxn=110;const int maxe=10010;const int inf=1<<29;int head[maxn],nxt[maxe],cost[maxe],pnt[maxe];int a,b,m,e,l,k;void addedge(int u,int v,int c){    pnt[e]=v;cost[e]=c;nxt[e]=head[u];head[u]=e++;}struct node{    int u,l,c;    node(int u=0,int l=0,int c=0):u(u),l(l),c(c){}};int vis[maxn][510][12];int d[maxn][510][12];void bellmanford(){    queue<node> q;    for(int i=1;i<=a+b;i++)        for(int j=0;j<=l;j++)            for(int x=0;x<=k;x++)            {                vis[i][j][x]=0;                d[i][j][x]=inf;            }    q.push(node(a+b,0,k));    vis[a+b][0][k]=1;    d[a+b][0][k]=0;    while(!q.empty())    {        node tmp=q.front();q.pop();        int u=tmp.u,len=tmp.l,cnt=tmp.c;        //cout<<u<<" "<<len<<" "<<cnt<<" "<<d[u][len][cnt]<<endl;        vis[u][len][cnt]=0;        for(int i=head[u];i!=-1;i=nxt[i])        {            int v=pnt[i],c=cost[i];            if(len==0)            {                if(d[v][len][cnt]>d[u][len][cnt]+c)                {                    d[v][len][cnt]=d[u][len][cnt]+c;                    if(!vis[v][len][cnt])                    {                        vis[v][len][cnt]=1;                        q.push(node(v,len,cnt));                    }                }            }            else            {                if(d[v][0][cnt]>d[u][len][cnt]+c)                {                    d[v][0][cnt]=d[u][len][cnt]+c;                    if(!vis[v][0][cnt])                    {                        vis[v][0][cnt]=1;                        q.push(node(v,0,cnt));                    }                }            }            if(cnt>0 && l>=c)            {                int res;                if(v>a) res=0;                else res=l-c;                if(d[v][res][cnt-1]>d[u][len][cnt])                {                    d[v][res][cnt-1]=d[u][len][cnt];                    if(!vis[v][res][cnt-1])                    {                        vis[v][res][cnt-1]=1;                        q.push(node(v,res,cnt-1));                    }                }            }            if(len>=c)            {                int res;                if(v>a) res=0;                else res=len-c;                if(d[v][res][cnt]>d[u][len][cnt])                {                    d[v][res][cnt]=d[u][len][cnt];                    if(!vis[v][res][cnt])                    {                        vis[v][res][cnt]=1;                        q.push(node(v,res,cnt));                    }                }            }        }    }}int main(){    //freopen("in.txt","r",stdin);    int T;    scanf("%d",&T);    while(T--)    {        memset(head,-1,sizeof(head));        e=0;        scanf("%d%d%d%d%d",&a,&b,&m,&l,&k);        while(m--)        {            int u,v,c;            scanf("%d%d%d",&u,&v,&c);            addedge(u,v,c);            addedge(v,u,c);        }        bellmanford();        int ans=inf;        for(int i=0;i<=l;i++)            for(int j=0;j<=k;j++)                ans=min(ans,d[1][i][j]);        printf("%d\n",ans);    }    return 0;}