BZOJ3442: 学习小组

最小费用最大流
源连向每个学生流量为k，费用为-Fi
每个学生连向他能去的学习小组，流量为1，费用为0
每个学习小组的支出费用与参加的人数平方相关，所以拆边连向汇（类似石头剪刀布），拆出n条边代表参加的人数为1~n时增加的支出，因为最小费用，所以总是会先选小的边，比如第i条边，选的时候1~i-1一定已经流满了，那这时i的贡献为
$C*i^2-C*(i-1)^2=C*(2*i-1)$
流量为1，费用就是C*(2i-1)

这题有个坑，即要求参加的学生尽量多，并不要求每个学生参加尽量多的小组，只要参加至少一个即可，所以每个学生连k-1的流量到汇，费用0
code：

#include<set>
#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<cmath>
#include<ctime>
#include<bitset>
#include<string>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<climits>
#include<complex>
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;

inline void down(int &x,const int &y){if(x>y)x=y;}
const int maxn = 1100;
const int maxm = 41000;

int n,m,st,ed;

struct edge
{
    int y,c,d,nex;
    edge(){}
    edge(const int _y,const int _c,const int _d,const int _nex){y=_y;c=_c;d=_d;nex=_nex;}
}a[maxm<<1]; int len,fir[maxn];
inline void ins(const int x,const int y,const int c,const int d)
{
    a[++len]=edge(y,c,d,fir[x]); fir[x]=len;
    a[++len]=edge(x,0,-d,fir[y]); fir[y]=len;
}

struct node
{
    int x,i;
    node(){}
    node(const int &_x,const int &_i){x=_x;i=_i;}
};
inline bool operator <(node x,node y){return x.x>y.x;}
priority_queue<node>q;
int d[maxn],pre[maxn],pos[maxn];
bool bfs()
{
    for(int i=1;i<=ed;i++) d[i]=INT_MAX;
    q.push(node(0,st)); d[st]=0;
    while(!q.empty())
    {
        const node x=q.top(); q.pop();
        if(x.x!=d[x.i]) continue;
        for(int k=fir[x.i];k;k=a[k].nex)
        {
            const int y=a[k].y;
            if(a[k].c&&d[y]>x.x+a[k].d)
            {
                d[y]=x.x+a[k].d;
                pos[y]=x.i; pre[y]=k;
                q.push(node(d[y],y));
            }
        }
    }
    return d[ed]!=INT_MAX;
}
int Flow()
{
    int re=0;
    while(bfs())
    {
        int c=INT_MAX;
        for(int k=ed;k!=st;k=pos[k]) down(c,a[pre[k]].c);
        re+=d[ed]*c;
        for(int k=ed;k!=st;k=pos[k]) a[pre[k]].c-=c,a[pre[k]^1].c+=c;
    }
    return re;
}
int fi[maxn];

int main()
{
    len=1;
    int K; scanf("%d%d%d",&n,&m,&K); st=n+m+1,ed=st+1;
    for(int i=1;i<=n;i++) ins(st,i,K,0),ins(i,ed,K-1,0);
    for(int i=1;i<=m;i++)
    {
        int x; scanf("%d",&x);
        for(int j=1;j<=n;j++) ins(n+i,ed,1,x*(j*j-(j-1)*(j-1)));
    }
    for(int i=1;i<=m;i++) scanf("%d",&fi[i]);

    char str[maxn];
    for(int i=1;i<=n;i++)
    {
        scanf("%s",str+1);
        for(int j=1;j<=m;j++) if(str[j]=='1')
            ins(i,n+j,1,-fi[j]);
    }

    printf("%d\n",Flow());

    return 0;
}