标签：int pay maxv 网络 POJ2516 add 强化 include define

　　k种货物分开求解最小费用最大流，主要减少了寻找最短路的时间。

#include<queue>
#include<cstdio>
#include<cstring>
using namespace std;
#define maxe 256000   //pay
#define maxv 5120    //pay
#define maxn 55    //pay
#define sc scanf
#define pt printf
#define rep(i,a,b)  for(int i=(a);i<(b);++i)
const int inf = 0x3f3f3f3f; 
int fee,cg,sp,need;
int N,M,K ,s,t;
typedef struct ed{
    int v,nxt,cap,dis;
}ed;
ed e[maxe];
int nxt[maxe],tot,head[maxv],vis[maxv],d[maxv],bk[maxv];
int mp[maxn][maxn][maxn];
int mi(int a,int b) {return a<b?a:b;}
void add(int u,int v,int cap,int dis)
{
    e[tot].v=v;         e[tot].nxt=head[u];
    e[tot].dis=dis;     e[tot].cap=cap;
    head[u]=tot++;

    e[tot].v=u;         e[tot].nxt=head[v];
    e[tot].dis=-dis;    e[tot].cap=0;
    head[v]=tot++;
} 
int spfa()
{
    queue<int> q;
    int u = s,v;
    q.push(u);
    while(!q.empty())
    {
        u = q.front(); q.pop();
        vis[u] = 0;
        for(int i = head[u];~i;i=e[i].nxt)
        {
            if(e[i].cap<1) continue;
            v = e[i].v;
            if(d[u]+e[i].dis<d[v])
            {
                bk[v]=i;
                d[v] = d[u]+e[i].dis;
                if(vis[v]==0)
                {
                    q.push(v);
                    vis[v]=1;
                }
            }
        }
    }
    return d[t]!=inf;
}

void init()
{
    tot=0;
    memset(head,-1,sizeof(head));   //pay 
}
int cust_need[maxn][maxn];
int halt_prov[maxn][maxn];
//第一次改进：如果仓库这种东西的存量为0或者客户需求为0，他们之间不建立边 —— 居然RE了
int main()
{
    freopen("in.txt","r",stdin);
    d[0]=0; bk[0]=-1;
    while(1)
    {
        need = fee = sp = 0;
        sc("%d%d%d",&N,&M,&K); //pt("%d %d %d\n",N,M,K);
        if(N==0&&M==0&&K==0) break;
        s=0,t=1+N+M;
        int i,j,k;
        memset(mp,-1,sizeof(mp));
        for(i=0;i<N;++i)  for(j=1;j<=K;++j) sc("%d",&cust_need[i][j]),need+=cust_need[i][j];
        for(i=0;i<M;++i)  for(j=1;j<=K;++j) sc("%d",&halt_prov[i][j]);
        for(k=1;k<=K;++k)  for(i=0;i<N;++i) for(j=0;j<M;++j)
        {
            sc("%d",&mp[k][i][j]);
            if(halt_prov[j][k]==0||cust_need[i][k]==0) continue;
            add(j*K+k,K*M+i*K+k,mi(halt_prov[j][k],cust_need[i][k]),mp[k][i][j]);
        }  
        for(k=1;k<=K;++k)
        {
            init();
            for(i=0;i<N;++i) if(cust_need[i][k]>0) add(1+M+i,t,cust_need[i][k],0);
            for(j=0;j<M;++j) if(halt_prov[j][k]>0) add(0,1+j,halt_prov[j][k],0);
            for(i=0;i<N;++i) for(j=0;j<M;++j)
            {
                if(halt_prov[j][k]==0||cust_need[i][k]==0) continue;
                add(1+j,1+M+i,mi(halt_prov[j][k],cust_need[i][k]),mp[k][i][j]);
            }
            while(1)
            {
                for(i=1;i<=t;++i) d[i]=inf,vis[i]=0; 
                if(spfa()==0) break;
                int min_flow = inf;
                cg=bk[t];
                while(cg!=-1)
                {
                    min_flow = mi(min_flow,e[cg].cap);
                    cg=bk[e[cg^1].v];
                }
                sp+=min_flow;
                cg=bk[t];
                while(cg!=-1)
                {
                    e[cg].cap-=min_flow;
                    e[cg^1].cap+=min_flow;
                    fee+=e[cg].dis*min_flow;
                    cg=bk[e[cg^1].v];
                }
            }
        }
        if(sp==need)   pt("%d\n",fee); //  write(fee), pt("\n");
        else pt("-1\n");    
    }
    
    return 0;
}

标签：int,pay,maxv,网络,POJ2516,add,强化,include,define
来源： https://www.cnblogs.com/lighten-up-belief/p/11345934.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。