BZOJ1449/2895: [JSOI2009]球队收益

BZOJ1449
BZOJ2895

【分析】

本题中无论输赢都会获得收益，所以按往常的费用流建图，不好完成。
那么我们先假设每场比赛双方全输，计算“初始答案”，然后对于每一场比赛，一方获胜，计算收益改变量。那么多赢一场，产生的收益为：(C*(w+1)^2+D*(l-1)^2)-(C*w^2+D*l^2)=2w*C-2l*D+C+D。每赢一次，w[i]++,l[i]–。初始答案+费用流值=最终答案。
那么为什么答案一定正确呢？因为对于每只球队，图中对于每条该球队连向汇点的边，一定是依次递增的（di<=ci），所以对于该球队的连向汇点的边，一定是从小到大依次增广。并且对于每只球队，不考虑具体赢了谁，只考虑赢了几场就可以了。建图中保证了每场比赛一定有且只有一个胜者。正确性由网络流保证。

【代码】

#include <cstdio>
#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <stack>
#define N 6005
#define M 10005
#define INF 1000000000
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pa;

int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

int n,m,cnt=1,S,T,ans;
int b[M],p[N],nextedge[M],w[M],c[M];
int Dis[N],Pre[N];
int wins[5010],lose[5010],in[5010],C[5010],D[5010];
bool Flag[N];

void Add(int x,int y,int z,int cost)
{
    cnt++;
    b[cnt]=y;
    nextedge[cnt]=p[x];
    p[x]=cnt;
    w[cnt]=z;
    c[cnt]=cost;
}

void Anode(int x,int y,int z,int cost){
    Add(x,y,z,cost);Add(y,x,0,-cost);
}

void Input_Init()
{
    n=read(),m=read();T=n+m+1;
    for(int i=1;i<=n;i++)
        wins[i]=read(),lose[i]=read(),C[i]=read(),D[i]=read();
    for(int i=1;i<=m;i++)
    {
        static int x,y;
        x=read(),y=read();Anode(0,i,1,0);
        Anode(i,x+m,1,0);Anode(i,y+m,1,0);
        in[x]++,in[y]++;lose[x]++,lose[y]++;
    }
    for(int i=1;i<=n;i++) ans+=C[i]*wins[i]*wins[i]+D[i]*lose[i]*lose[i];
    for(int i=1;i<=n;i++)
    for(int j=1;j<=in[i];j++)
    {
        Anode(i+m,T,1,2*C[i]*wins[i]+C[i]+D[i]-2*D[i]*lose[i]);
        wins[i]++,lose[i]--;
    }
}

bool Bfs()
{
    queue<int>q;
    q.push(S);
    for(int i=1;i<=T;i++) Dis[i]=INF;
    while(!q.empty())
    {
        int k=q.front();q.pop();Flag[k]=0;
        for(int i=p[k];i;i=nextedge[i])
        {
            int v=b[i],f=w[i];
            if(Dis[v]>Dis[k]+c[i]&&f)
            {
                Dis[v]=Dis[k]+c[i];
                Pre[v]=i;
                if(!Flag[v])
                {
                    Flag[v]=1;
                    q.push(v);
                }
            }
        }
    }
    return Dis[T]!=INF;
}

void Mcf()
{
    int Maxf=INF;
    for(int i=Pre[T];i;i=Pre[b[i^1]])
        Maxf=min(Maxf,w[i]);
    for(int i=Pre[T];i;i=Pre[b[i^1]])
    {
        w[i]-=Maxf,w[i^1]+=Maxf;
        ans+=c[i]*Maxf;
    }
}

void MCF()
{
    while(Bfs())
        Mcf();
    printf("%d\n",ans);
}

int main()
{
    Input_Init();
    MCF();
    return 0;
}