[网络流] bzoj1449: [JSOI2009]球队收益

bzoj1449: [JSOI2009]球队收益 http://www.lydsy.com/JudgeOnline/problem.php?id=1449

因为输赢都有收益 所以先全部搞成输的
然后拆边 一个比赛一个比赛的加进去
然后算最小费用嘛
表示不会最大流

调试调了一个小时
发现只是打多了一个int
-打了+
菜啊

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
struct node
{
    int x,y,c,cost,next,other;
}a[21000];
int last[11000],len;
int win[5100],lose[5100];
int x[1100],y[1100],c[5100],d[5100];
int list[1100000],f[11000],pre[11000];
bool v[11000];
int ans=0,st,ed;
int sqr(int x) {return x*x;}
void build(int x,int y,int c,int cost)
{
    int k1,k2;
    len++;k1=len;
    a[len].x=x;a[len].y=y;a[len].c=c;a[len].cost=cost;a[len].next=last[x];last[x]=len;
    len++;k2=len;
    a[len].x=y;a[len].y=x;a[len].c=0;a[len].cost=-cost;a[len].next=last[y];last[y]=len;
    a[k1].other=k2;a[k2].other=k1;
}
bool spfa()
{
    memset(v,true,sizeof(v));
    memset(f,63,sizeof(f));
    int head=1,tail=1;
    list[1]=st;v[st]=false;f[st]=0;
    while (head<=tail)
    {
        int x=list[head];
        for (int k=last[x];k;k=a[k].next)
        {
            int y=a[k].y;
            if (f[y]>f[x]+a[k].cost&&a[k].c>0)
            {
                f[y]=f[x]+a[k].cost;
                pre[y]=k;
                if (v[y])
                {
                    v[y]=false;
                    list[++tail]=y;
                }
            }
        }
        head++;
        v[x]=true;
    }
    if (f[ed]>=999999999) return 0;
    else 
    {
        int x=ed;
        int sum=999999999;
        while (x!=st)
        {
            int k=pre[x];
            sum=min(sum,a[k].c);
            x=a[k].x;
        }
        x=ed;
        while (x!=st)
        {
            int k=pre[x];
            a[k].c-=sum;
            a[a[k].other].c+=sum;
            x=a[k].x;
        }
        ans+=f[ed]*sum;
        return 1;
    }

}
int main()
{
    // 因为输赢都有收益 所以不能正常费用流 
    // 所以先把全部人都搞成输的 然后拆边 
    int n,m;
    scanf("%d%d",&n,&m);
    st=n+m+1;ed=st+1;
    for (int i=1;i<=n;i++){scanf("%d%d%d%d",&win[i],&lose[i],&c[i],&d[i]);}
    for (int i=1;i<=m;i++){scanf("%d%d",&x[i],&y[i]);lose[x[i]]++;lose[y[i]]++;}
    for (int i=1;i<=n;i++) ans+=sqr(win[i])*c[i]+sqr(lose[i])*d[i];
    for (int i=1;i<=m;i++)
    {
        build(st,i,1,0); //和每一场比赛连边 
        build(i,x[i]+m,1,0); //每一场比赛和他们的胜负方连边 
        build(i,y[i]+m,1,0);
        // c[i]*(win[i]+1)*(win[i]+1)+d[i]*(lose[i]-1)*(lose[i]-1)-(c[i]*win[i]*win[i]+d[i]*lose[i]*lose[i])
        //=c[i]*(win[1]^2+2*win[1]+1)+d[i]*(lose[i]^2-2*lose[i]+1)-c[i]*win[i]*win[i]-d[i]*lose[i]*lose[i]
        //=2*c[i]*win[i]+c[i]-d[i]*2*lose[i]+d[i]
        int sum=2*c[x[i]]*win[x[i]]+c[x[i]]-2*d[x[i]]*lose[x[i]]+d[x[i]];//a[i]赢这场比赛之后增加的收益 
        build(x[i]+m,ed,1,sum);
        win[x[i]]++;lose[x[i]]--;
        sum=2*c[y[i]]*win[y[i]]+c[y[i]]-2*d[y[i]]*lose[y[i]]+d[y[i]];//b[i]赢这场比赛增加的收益
        build(y[i]+m,ed,1,sum);
        win[y[i]]++;lose[y[i]]--; 
    }
    while (spfa()) {}
    printf("%d\n",ans);
    return 0;
}