3442: 学习小组

3442: 学习小组

Time Limit: 5 Sec   Memory Limit: 128 MB
Submit: 439   Solved: 212
[ Submit][ Status][ Discuss]

Description

【背景】

坑校准备鼓励学生参加学习小组。

【描述】

    共有n个学生，m个学习小组，每个学生有一定的喜好，只愿意参加其中的一些学习小组，但是校领导为学生考虑，规定一个学生最多参加k个学习小组。财务处的大叔就没那么好了，他想尽量多收钱，因为每个学生参加学习小组都要交一定的手续费，不同的学习小组有不同的手续费。然而，事与愿违，校领导又决定对学习小组组织者进行奖励，若有a个学生参加第i个学习小组，那么给这个学习小组组织者奖励Ci*a^2元。在参与学生（而不是每个学习小组的人数总和）尽量多的情况下，求财务处最少要支出多少钱（若为负数，则输出负数）（支出=总奖励费-总手续费）。

Input

输入有若干行，第一行有三个用空格隔开的正整数n、m、k。接下来的一行有m个正整数，表示每个Ci。第三行有m个正整数，表示参加每个学习小组需要交的手续费Fi。再接下来有一个n行m列的矩阵，表若第i行j列的数字是1，则表示第i个学生愿意参加第j个学习小组，若为0，则为不愿意。

Output

 

输出只有一个整数，为最小的支出。

Sample Input

3 3 1
1 2 3
3 2 1
111
111
111

Sample Output

-2
【样例解释】
参与学生最多为3，每个学生参加一个学习小组，若有两个学生参加第一个学习小组，一个学生参加第二个学习小组（一定要有人参加第二个学习小组），支出为-2，可以证明没有更优的方案了。
【数据范围与约定】
100%的数据，0＜n≤100，0＜m≤90，0＜k≤m，0＜Ci≤10，0＜Fi≤100。

题解：

st->每个学生，流量为k，费用为0。表示，最多可以选k门课。

每个学生->他喜欢的学习小组，流量为1，费用为-F[i]，表示该学生选这个小组学校可以赚多少钱

每个小组->ed（建n条边） 流量为1，费用为：C[i]*（2*a-1），表示这个小组有i个学生要给多少钱。

因为是最小费用，所以选i的时候，i-1一定已经选了。

那么每增加一个学生，费用增加C[i]*a^2-C[i]*(a-1)^2

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<queue>
#define INF 0x7fffffff
using namespace std;
const int N=41000;
const int M=1100;
int n,m,k;
struct node{
    int x,y,z,next,other,cost;
}sa[N<<1];int len=0,first[M];
int C[M],F[M],st,ed;
void ins(int x,int y,int z,int w)
{
    len++;
    sa[len].x=x;
    sa[len].y=y;
    sa[len].z=z;
    sa[len].cost=w;
    sa[len].next=first[x];
    sa[len].other=len+1;
    first[x]=len;
    len++;
    sa[len].x=y;
    sa[len].y=x;
    sa[len].z=0;
    sa[len].cost=-w;
    sa[len].next=first[y];
    sa[len].other=len-1;
    first[y]=len;
}
char map[M][M];
int pre[M],dis[M],frpe[M],vis[M];
bool spfa()
{
    memset(pre,-1,sizeof(pre));pre[st]=0;
    memset(vis,0,sizeof(vis));
    for(int i=0;i<=ed;i++) dis[i]=INF;
    dis[st]=0;
    vis[st]=1;
    queue<int>q;
    q.push(st);
    while(!q.empty())
    {
        int x=q.front();
        q.pop();vis[x]=0;
        for(int i=first[x];i!=-1;i=sa[i].next)
        {
            int y=sa[i].y;
            if(sa[i].z && dis[y]>dis[x]+sa[i].cost)
            {
                dis[y]=dis[x]+sa[i].cost;
                if(!vis[y])
                {
                    vis[y]=1;
                    q.push(y);
                }
                pre[y]=x;
                frpe[y]=i;
            }
        }
    }
    if(dis[ed]>=INF) return false;
    return true;
}
void MCMF()
{
    int ans=0;
    while(spfa())
    {
        int minl=INF;
        for(int i=ed;i!=st;i=pre[i])
        {
            if(minl>sa[frpe[i]].z)
            minl=sa[frpe[i]].z;
        }
        ans+=minl*dis[ed];
        for(int i=ed;i!=st;i=pre[i])
        {
            sa[frpe[i]].z-=minl;
            sa[sa[frpe[i]].other].z+=minl;
        }
    }
    printf("%d\n",ans);
}
int main()
{
    scanf("%d%d%d",&n,&m,&k);
    memset(first,-1,sizeof(first));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++)
    {
        scanf("%d",&C[i]);
        for(int j=1;j<=n;j++)
        ins(i+n,ed,1,C[i]*(2*j-1));
    }
    for(int i=1;i<=m;i++)
    scanf("%d",&F[i]);
     
    for(int i=1;i<=n;i++)
    {
        scanf("%s",map[i]+1);
        for(int j=1;j<=m;j++)
        {
            if(map[i][j]=='1')
            {
                ins(i,n+j,1,-F[j]);
                //ver[j]++;
            }
        }
    }
    /*for(int i=1;i<=m;i++)
    {
        for(int j=1;j<=ver[j];j++)
        ins(i+n,ed,1,C[i]*(2*j-1));
    }*/
    MCMF();
}