bzoj3280: 小R的烦恼

本题和上题bzoj1221是很类似的啊

注意的是要为每个学校开一个新点，再流向另一边。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
const int inf=999999999;

int n;
struct node
{
    int x,y,c,d,next,other;
}a[2100000];int len,last[21000];
void ins(int x,int y,int c,int d)
{
    int k1,k2;
    
    len++;k1=len;
    a[len].x=x;a[len].y=y;a[len].c=c;a[len].d=d;
    a[len].next=last[x];last[x]=len;
    
    //printf("%d %d %d %d\n",x,y,c,d);
    
    len++;k2=len;
    a[len].x=y;a[len].y=x;a[len].c=0;a[len].d=-d;
    a[len].next=last[y];last[y]=len;
    
    a[k1].other=k2;
    a[k2].other=k1;
}

int C,ans,st,ed;
int list[21000];bool v[21000];
int d[21000],pre[21000],cc[21000];
bool spfa()
{
    memset(d,63,sizeof(d));d[st]=0;
    memset(cc,0,sizeof(cc));cc[st]=inf;
    memset(v,false,sizeof(v));v[st]=true;
    int head=1,tail=2;list[1]=st;
    while(head!=tail)
    {
        int x=list[head];
        for(int k=last[x];k;k=a[k].next)
        {
            int y=a[k].y;
            if(d[y]>d[x]+a[k].d&&a[k].c>0)
            {
                d[y]=d[x]+a[k].d;
                cc[y]=min(cc[x],a[k].c);
                pre[y]=k;
                if(v[y]==false)
                {
                    v[y]=true;
                    list[tail]=y;
                    tail++;if(tail==20500)tail=1;
                }
            }
        }
        v[x]=false;
        head++;if(head==20500)head=1;
    }
    if(d[ed]>=inf)return false;
    else
    {
        C+=cc[ed];
        ans+=cc[ed]*d[ed];
        int y=ed;
        while(pre[y]!=0)
        {
            int k=pre[y];
            a[k].c-=cc[ed];
            a[a[k].other].c+=cc[ed];
            y=a[k].x;
        }
        return true;
    }
}

int main()
{
    int T;
    scanf("%d",&T);
    for(int ttt=1;ttt<=T;ttt++)
    {
        int m,k,x,y,sum=0;
        scanf("%d%d%d",&n,&m,&k);st=n*2+m+1,ed=n*2+m+2;
        len=0;memset(last,0,sizeof(last));
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&x);
            ins(st,i,x,0);
            ins(i+n,ed,x,0);
            sum+=x;
        }
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d",&x,&y);
            ins(st,2*n+i,x,0);
            for(int j=1;j<=n;j++)ins(2*n+i,j+n,x,y);
        }
        for(int i=1;i<=k;i++)
        {
            scanf("%d%d",&x,&y);
            for(int j=1;j<=n;j++)
                if(j+x+1<=n)ins(j,j+n+x+1,inf,y);
                else break;
        }
          for(int i=1;i<n;i++)ins(i,i+1,inf,0);
        
        C=0;ans=0;
        while(spfa()==true);
        printf("Case %d: ",ttt);
        if(C==sum)printf("%d\n",ans);
        else printf("impossible\n");
    }
    return 0;
}