BZOJ 2424: [HAOI2010]订货|费用流

真是良心题,一开始想拆点,后来发现完全不用拆点
对于第x个月,要连以下边:
1. S->x 流量无穷大,费用为dx
2. x->T 流量为Ux,费用为0
3. x->x+1 流量为仓库容量(S被我用来当源点了),费用为m
code:

#include<set>
#include<map>
#include<ctime>
#include<queue>
#include<cmath>
#include<cstdio>
#include<vector>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define T 111
#define MX 1e9
using namespace std;
int sc()
{
    int i=0,f=1; char c=getchar();
    while(c>'9'||c<'0'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();
    return i*f;
}
int dis[T],q[T],vis[T],from[T],pre[T];
int head[T],nxt[T*T],lst[T*T],c[T*T],v[T*T];
int n,S,W,tot=1,X,ans,m;
void insert(int x,int y,int z,int a)
{
    lst[++tot]=y; nxt[tot]=head[x]; c[tot]=z; v[tot]=a; head[x]=tot;
    lst[++tot]=x; nxt[tot]=head[y]; c[tot]=0; v[tot]=-a;head[y]=tot;
}
bool spfa()
{
    for(int i=1;i<=W;i++) dis[i]=MX; dis[S]=0;
    int l=1,r=2; q[1]=S;
    while(l!=r)
    {
        int x=q[l++]; l%=T; vis[x]=0;
        for(int i=head[x];i;i=nxt[i])
            if(c[i]&&dis[lst[i]]>dis[x]+v[i])
            {
                dis[lst[i]]=dis[x]+v[i];
                from[lst[i]]=x;
                pre[lst[i]]=i;
                if(!vis[lst[i]])
                {
                    vis[lst[i]]=1;
                    q[r++]=lst[i];
                    r%=T;
                }
            }
    }
    return dis[W]!=MX;
}
void mcf()
{
    int cc=MX;
    for(int i=W;i!=S;i=from[i])cc=min(cc,c[pre[i]]);
    for(int i=W;i!=S;i=from[i])
    {
        int k=pre[i];
        ans+=v[k]*cc;
        c[k]-=cc;
        c[k^1]+=cc;
    }
}
int main()
{
    n=sc(),m=sc(),X=sc();
    S=n+1,W=S+1;
    for(int i=1;i<=n;i++)
    {
        int x=sc();
        insert(i,W,x,0);
    }
    for(int i=1;i<=n;i++)
    {
        int x=sc();
        insert(S,i,MX,x);
        if(i!=n)insert(i,i+1,X,m);
    }
    while(spfa())mcf();
    cout<<ans;
    return 0;
}

你可能感兴趣的:(费用流)