图论专项shortest_paths:UVa 10269

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;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值