loj6008「网络流 24 题」餐巾计划(费用流)

这个问题的主要约束条件是每天的餐巾够用,而餐巾的来源可能是最新购买,也可能是前几天送洗,今天刚刚洗好的餐巾。每天用完的餐巾可以选择送到快洗部或慢洗部,或者留到下一天再处理。
经过分析可以把每天要用的和用完的分离开处理,建模后就是二分图。二分图X集合中顶点Xi表示第i天用完的餐巾,其数量为ri,所以从S向Xi连接容量为ri的边作为限制。Y集合中每个点Yi则是第i天需要的餐巾,数量为ri,与T连接的边容量作为限制。每天用完的餐巾可以选择留到下一天(Xi→Xi+1),不需要花费,送到快洗部(Xi→Yi+m),费用为F,送到慢洗部(Xi−>Yi+n),费用为S。每天需要的餐巾除了刚刚洗好的餐巾,还可能是新购买的(S→Yi),费用为P。然后跑zkw费用流就好啦啦啦

#include 
using namespace std;
#define ll long long
#define N 2010
#define inf 0x3f3f3f3f
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,P,t1,c1,t2,c2,T=2001,mncost=0,pay=0,h[N],num=1;
bool vis[N];
struct edge{
    int to,next,w,c;
}data[7000<<1];
inline void add(int x,int y,int w,int c){
    data[++num].to=y;data[num].next=h[x];h[x]=num;data[num].w=w;data[num].c=c;
    data[++num].to=x;data[num].next=h[y];h[y]=num;data[num].w=0;data[num].c=-c;
}
inline int dinic(int x,int low){
    vis[x]=1;if(x==T){mncost+=low*pay;return low;}int tmp=low;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(vis[y]||data[i].c||!data[i].w) continue;
        int res=dinic(y,min(tmp,data[i].w));
        tmp-=res;data[i].w-=res;data[i^1].w+=res;
        if(!tmp) return low;
    }return low-tmp;
}
inline bool label(){
    int d=inf;
    for(int x=0;x<=n<<1;++x){
        if(!vis[x]) continue;
        for(int i=h[x];i;i=data[i].next){
            int y=data[i].to;if(vis[y]||!data[i].w) continue;
            d=min(d,data[i].c);
        }
    }if(d==inf) return 0;
    for(int x=0;x<=n<<1;++x){
        if(!vis[x]) continue;
        for(int i=h[x];i;i=data[i].next) data[i].c-=d,data[i^1].c+=d;
    }pay+=d;return 1;
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();P=read();t1=read();c1=read();t2=read();c2=read();
    for(int i=1;i<=n;++i){
        int x=read();add(0,i,x,0);add(i+n,T,x,0);
        add(0,i+n,inf,P);if(i+t1<=n) add(i,i+t1+n,inf,c1);
        if(i+t2<=n) add(i,i+t2+n,inf,c2);if(i+1<=n) add(i,i+1,inf,0);
    }do do memset(vis,0,sizeof(vis));while(dinic(0,inf));while(label());
    printf("%d\n",mncost);
    return 0;
}

你可能感兴趣的:(其他oj,-----网络流-------,费用流)