【bzoj3442】 学习小组

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。

【解析】

拆边+费用流

这道题建边好巧妙*-*

因为有Ci*a^2的存在，使得正常加边的费用流无法处理

然后建容量为1，费用为Ci*1、Ci*3、Ci*5、Ci*7、...的一堆边

这样在最小费用的前提下总花费满足题意啦……

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
struct node
{
	int x,y,c,d,next,other;
}a[110000];int len,last[210];
int st,ed;
void ins(int x,int y,int c,int d)
{
	int k1,k2;
	len++;k1=len;
	a[len].x=x;a[len].y=y;a[len].c=c;a[len].d=d;
	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].d=-d;
	a[len].next=last[y];last[y]=len;
	
	a[k1].other=k2;
	a[k2].other=k1;
}
int list[210],head,tail,d[210];
int pre[210];
bool v[210];
bool spfa()
{
	for(int i=1;i<=ed;i++) d[i]=999999999;
	d[st]=0;
	memset(v,false,sizeof(v));v[st]=true;
	list[1]=st;head=1;tail=2;
	while(head!=tail)
	{
		int x=list[head];
		for(int k=last[x];k;k=a[k].next)
		{
			int y=a[k].y;
			if(a[k].c>0 && d[y]>d[x]+a[k].d)
			{
				d[y]=d[x]+a[k].d;
				pre[y]=k;
				if(v[y]==false)
				{
					v[y]=true;
					list[tail++]=y;
					if(tail==ed+1) tail=1;
				}
			}
		}
		head++;
		if(head==ed+1) head=1;
		v[x]=false;
	}
	if(d[ed]>=999999999) return false;
	else return true;
}
int ans=0;
void get()
{
	int x=ed,minn=999999999;
	while(x!=st)
	{
		int k=pre[x];
		minn=min(minn,a[k].c);
		x=a[k].x;
	}
	x=ed;
	while(x!=st)
	{
		int k=pre[x];
		a[k].c-=minn;
		a[a[k].other].c+=minn;
		ans+=minn*a[k].d;
		x=a[k].x;
	}
}
int c[100],f[100];
int main()
{
	int n,m,k;
	char ss[100];
	scanf("%d%d%d",&n,&m,&k);
	for(int i=1;i<=m;i++) scanf("%d",&c[i]);
	for(int i=1;i<=m;i++) scanf("%d",&f[i]);
	
	st=n+m+1;ed=st+1;
	for(int i=1;i<=m;i++)
	{
		for(int j=1;j<=n;j++) ins(i+n,ed,1,c[i]*(2*j-1));
	}
	
	for(int i=1;i<=n;i++)
	{
		scanf("%s",ss+1);
		ins(st,i,k,0);ins(i,ed,k-1,0);
		for(int j=1;j<=m;j++)
		{
			if(ss[j]=='1') ins(i,j+n,1,-f[j]);
		}
	}
	while(spfa()==true) get();
	printf("%d\n",ans);
	return 0;
}