洛谷4307 [JSOI2009]球队收益

Luogu

BZOJ

分析

这题好神仙啊qwq

考虑费用流。

先从源点向每场比赛连容量为$1$,费用为$0$的边,从每场比赛都参加这场比赛的两方连容量为$1$,费用为$0$的边。

先假设所有队在所有比赛中全负。再设第$i$支球队胜$a[i]$场,负$b[i]$场。

每个球队多赢一场,预算会增加$c\times(2\times a[i]+1)-d\times(2\times b[i]-1)$。

然后每支球队向汇点连$b[i]$条容量为$1$,费用为上面这个东西的边,每连一条边就将$a[i]$加$1$,$b[i]$减$1$。

然后跑一遍费用流,最小费用加上全负的花费就是答案。

代码

#include <algorithm>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <queue>
#define re register
using namespace std;

inline int read() {
    int X=0,w=1; char c=getchar();
    while (c<'0'||c>'9') { if (c=='-') w=-1; c=getchar(); }
    while (c>='0'&&c<='9') X=X*10+c-'0',c=getchar();
    return X*w;
}

const int N=5000+10;
const int INF=0x3f3f3f3f;

int a[N],b[N],c[N],d[N],sum[N];
int n,m,s,t;

struct Edge { int v,w,c,nxt; };
Edge e[1000000];
int head[10000],cnt=2;

inline void addEdge(int u,int v,int w,int c) {
    e[cnt].v=v,e[cnt].w=w,e[cnt].c=c,e[cnt].nxt=head[u],head[u]=cnt++;
    e[cnt].v=u,e[cnt].w=0,e[cnt].c=-c,e[cnt].nxt=head[v],head[v]=cnt++;
}
int dis[10000],inq[10000],last[10000];

inline int SPFA() {
    memset(inq,0,sizeof(inq)); memset(last,0,sizeof(last));
    memset(dis,0x3f,sizeof(dis)); dis[s]=0;
    queue<int> Q; Q.push(s);
    while (!Q.empty()) {
        int u=Q.front(); Q.pop(); inq[u]=0;
        for (re int i=head[u];i;i=e[i].nxt) {
            int v=e[i].v,c=e[i].c;
            if (e[i].w&&dis[u]+c<dis[v]) {
                dis[v]=dis[u]+c,last[v]=i;
                if (!inq[v]) inq[v]=1,Q.push(v);
            }
        }
    }
    return last[t]!=0;
}

int main() {
    n=read(),m=read(),s=0,t=n+m+1;
    for (re int i=1;i<=n;++i)
        a[i]=read(),b[i]=read(),c[i]=read(),d[i]=read();
    for (re int i=1;i<=m;++i) {
        int u=read(),v=read();
        ++sum[u],++sum[v],++b[u],++b[v];
        addEdge(s,i,1,0); addEdge(i,u+m,1,0); addEdge(i,v+m,1,0);
    }
    int ans=0;
    for (re int i=1;i<=n;++i) ans+=c[i]*a[i]*a[i]+d[i]*b[i]*b[i];
    for (re int i=1;i<=n;++i)
        for (re int j=1;j<=sum[i];++j) {
            addEdge(i+m,t,1,c[i]*(2*a[i]+1)-d[i]*(2*b[i]-1));
            ++a[i],--b[i];
        }
    while (SPFA()) {
        int flow=INF;
        for (re int i=last[t];i;i=last[e[i^1].v])
            flow=min(flow,e[i].w);
        for (re int i=last[t];i;i=last[e[i^1].v])
            e[i].w-=flow,e[i^1].w+=flow;
        ans+=dis[t]*flow;
    }
    printf("%d\n",ans);
    return 0;
}
最后修改:2019 年 09 月 24 日 10 : 04 PM

发表评论