网络流强化-POJ2516

Posted 2022-07-04 lighten-up-belief

　　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;

POJ 2516