YYH的球盒游戏

Posted 2020-09-21 Yzyet

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int h[5210],d[5210],used[5210],que[100010],last[5210];
int k=1,INF=0x7fffffff,ans=0,a[101],b[101][101],c[101];
inline int read(){
    int t=1,num=0;
    char c=getchar();
    while(c>\'9\'||c<\'0\'){if(c==\'-\')t=-1;c=getchar();}
    while(c>=\'0\'&&c<=\'9\'){num=num*10+c-\'0\';c=getchar();}
    return num*t;
}
struct edge{
    int to,cap,cost,next;
}g[120010];
void add(int f,int t,int c1,int c2){
    g[++k].next=h[f];h[f]=k;g[k].to=t;g[k].cap=c1;g[k].cost=c2;
    g[++k].next=h[t];h[t]=k;g[k].to=f;g[k].cap=0;g[k].cost=-c2;
}
bool spfa(int s,int t){
    memset(last,0,sizeof(last));
    memset(d,127/3,sizeof(d));INF=d[0];
    memset(used,0,sizeof(used));
    int tail,head;
    head=tail=50002;
    que[head]=s;used[s]=1;d[s]=0;
    while(head>=tail){
        int x=que[tail++];
        for(int i=h[x];i;i=g[i].next){
            if(g[i].cap&&d[x]+g[i].cost<d[g[i].to]){
                d[g[i].to]=d[x]+g[i].cost;
                last[g[i].to]=i;
                if(!used[g[i].to]){
                    if(d[g[i].to]<d[que[tail]])que[--tail]=g[i].to;
                    else que[++head]=g[i].to;
                    used[g[i].to]=1;
                }
            }
        }
        used[x]=0;
    }
    return d[t]!=INF;
}
void mcf(int t){
    int minn=INF;
    for(int i=last[t];i;i=last[g[i^1].to])minn=min(minn,g[i].cap);
    for(int i=last[t];i;i=last[g[i^1].to]){
        ans+=g[i].cost*minn;
        g[i].cap-=minn;
        g[i^1].cap+=minn;
    }
}
int main()
{
    int n,m,s,t;
    n=read();m=read();
    s=0,t=n+m+1;
    for(int i=1;i<=n;i++){a[i]=read();add(s,i,a[i],0);}
    for(int i=1;i<=m;i++){c[i]=read();add(n+i,t,c[i],0);}
    for(int i=1;i<=m;i++){
        int liu=-1;
        for(int j=1;j<=n-c[i];j++){
            liu+=2;
            add(n+i,t,1,liu);
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            b[i][j]=read();
            add(i,n+j,1,-b[i][j]);
        }
    }
    for(int i=1;i<=n;i++)add(i,t,INF,0);
    while(spfa(s,t))mcf(t);
    printf("%d\\n",-ans);
    return 0;
}

本文由Yzyet编写，网址为www.cnblogs.com/Yzyet。非Yzyet同意，禁止转载，侵权者必究。