[BZOJ3280]小R的烦恼（费用流）

题目描述

传送门

题解

设大学zi，将每一天拆点xiyi，分别表示濒死的和健康的。
对于每一个大学i，s->zi,li,0
对于每一个x，s->xi,ai,0，表示一天最多产生ai和濒死的人。
对于每一个y，yi->t,ai,0，表示每天需要的人。
每一个z->y,INF,pi，表示每天新买进的人。
xi->yi+di+1,INF,qi，表示医治好的人。
xi->xi+1,INF,0，表示延期医治的人。

代码

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
using namespace std;

const int max_n=100;
const int max_N=500;
const int max_m=max_N*max_N;
const int max_e=max_m*2;
const int INF=1e9;

int T,n,m,k,N,sum,maxflow,mincost,a[max_n],l[max_n],p[max_n],d[max_n],qq[max_n];
int tot,point[max_N],nxt[max_e],v[max_e],remain[max_e],c[max_e];
int dis[max_N],last[max_N]; bool vis[max_N];
queue <int> q;

inline void clear()
{
    n=m=k=N=sum=maxflow=mincost=0;
    tot=-1;memset(point,-1,sizeof(point));memset(nxt,-1,sizeof(nxt));
    memset(v,0,sizeof(v));memset(remain,0,sizeof(remain));memset(c,0,sizeof(c));
    memset(a,0,sizeof(a));memset(l,0,sizeof(l));memset(d,0,sizeof(d));memset(p,0,sizeof(p));memset(qq,0,sizeof(qq));
    memset(last,0,sizeof(last));
}
inline void addedge(int x,int y,int cap,int z)
{
    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y; remain[tot]=cap; c[tot]=z;
    ++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x; remain[tot]=0; c[tot]=-z;
}
inline int addflow(int s,int t)
{
    int now=t,ans=INF;
    while (now!=s)
    {
        ans=min(ans,remain[last[now]]);
        now=v[last[now]^1];
    }
    now=t;
    while (now!=s)
    {
        remain[last[now]]-=ans;
        remain[last[now]^1]+=ans;
        now=v[last[now]^1];
    }
    return ans;
}
inline bool bfs(int s,int t)
{
    memset(dis,0x7f,sizeof(dis)); dis[s]=0;
    memset(vis,0,sizeof(vis)); vis[s]=true;
    while (!q.empty()) q.pop(); q.push(s);

    while (!q.empty())
    {
        int now=q.front(); q.pop();
        vis[now]=false;
        for (int i=point[now];i!=-1;i=nxt[i])
            if (dis[v[i]]>dis[now]+c[i]&&remain[i])
            {
                dis[v[i]]=dis[now]+c[i];
                last[v[i]]=i;
                if (!vis[v[i]])
                {
                    vis[v[i]]=true;
                    q.push(v[i]);
                }
            }
    }
    if (dis[t]>INF) return false;
    int flow=addflow(s,t);
    maxflow+=flow;
    mincost+=dis[t]*flow;
    return true;
}
inline void mincostflow(int s,int t)
{
    while (bfs(s,t));
}
int main()
{
    scanf("%d",&T);
    for (int t=1;t<=T;++t)
    {
        clear();
        scanf("%d%d%d",&n,&m,&k);
        for (int i=1;i<=n;++i) scanf("%d",&a[i]),sum+=a[i];
        for (int i=1;i<=m;++i) scanf("%d%d",&l[i],&p[i]);
        for (int i=1;i<=k;++i) scanf("%d%d",&d[i],&qq[i]);
        N=m+n*2+2;
        for (int i=1;i<=m;++i)
            addedge(1,1+i,l[i],0);
        for (int i=1;i<=n;++i)
            addedge(1,1+m+i,a[i],0);
        for (int i=1;i<=n;++i)
            addedge(1+m+n+i,N,a[i],0);
        for (int i=1;i<=m;++i)
            for (int j=1;j<=n;++j)
                addedge(1+i,1+m+n+j,INF,p[i]);
        for (int i=1;i<=n;++i)
            for (int j=1;j<=k;++j)
                if (i+d[j]+1<=n)
                    addedge(1+m+i,1+m+n+i+d[j]+1,INF,qq[j]);
        for (int i=1;i<n;++i)
            addedge(1+m+i,1+m+i+1,INF,0);

        mincostflow(1,N);
        if (maxflow<sum) printf("Case %d: impossible\n",t);
        else printf("Case %d: %d\n",t,mincost);
    }
}