BZOJ 3442: 学习小组 费用流 拆边

3442: 学习小组

Time Limit: 5 Sec  Memory Limit: 128 MB
Submit: 463  Solved: 217
[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。

最小费用最大流 拆边

如果没有费用为Ci*a^2那就是普通的费用流

现在多了这个要求 如何解决呢

(a+1)^2=a^1+2a+1

所以每个社团向汇点连费用为1*Ci、3*Ci、5*Ci、7*Ci……流量为1的边

这样在最小费用流的前提下即保证了通过人数可以完美对应花费

本题要求每个人都有参与又要费用最小 有每个人向汇点连费用0 流量1的边

多路增广大法好 （循环队列打错调好久TAT）

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<map>
#include<set>
using namespace std;
inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=10*x+ch-'0';ch=getchar();}
	return x*f;
}
inline void print(int x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=210,M=100100,inf=0X3f3f3f3f;

int ecnt=1,last[N];
struct EDGE{int to,nt,val,ct;}e[M];
inline void readd(int u,int v,int val,int ct){e[++ecnt]=(EDGE){v,last[u],val,ct};last[u]=ecnt;}
inline void add(int u,int v,int val,int ct){readd(u,v,val,ct);readd(v,u,0,-ct);}

int n,S,T=N-1;

int q[N],dis[N],ans=0;
bool inq[N],vis[N];
bool spfa()
{
	memset(dis,0X3f,sizeof(dis));
	memset(inq,0,sizeof(inq));
	memset(vis,0,sizeof(vis));
	register int head=0,tail=1;q[0]=0;inq[0]=1;dis[0]=0;
	while(head!=tail)
	{
		int u=q[head++];if(head==N)head=0;inq[u]=0;
		for(int i=last[u];i;i=e[i].nt)
		if(e[i].val&&dis[e[i].to]>dis[u]+e[i].ct)
		{
			dis[e[i].to]=dis[u]+e[i].ct;
			if(!inq[e[i].to]){q[tail++]=e[i].to,inq[e[i].to]=1;if(tail==N)tail=0;}
		}
	}
	return dis[T]<inf;
}

int dfs(int u,int lim)
{
	if(u==T)
	{ans+=dis[u]*lim;return lim;}
	vis[u]=1;
	register int i,tmp,res=0;
	for(i=last[u];i;i=e[i].nt)
	if(e[i].val&&!vis[e[i].to]&&dis[e[i].to]==dis[u]+e[i].ct)
	{
		tmp=dfs(e[i].to,min(lim,e[i].val));
		lim-=tmp;e[i].val-=tmp;e[i^1].val+=tmp;res+=tmp;
		if(!lim)break;
	}
	if(!res)dis[u]=-1;
	return res;
}

void mincf(){while(spfa())dfs(S,inf);}

char s[N];int f[N];
int main()
{
	n=read();int m=read(),k=read();
	register int i,val,j;
	for(i=1;i<=m;++i)
	{val=read();for(j=1;j<=n;++j){add(i+n,T,1,val*(j*2-1));}}
	for(i=1;i<=m;++i){f[i]=read();}
	for(i=1;i<=n;++i)
	{
		scanf("%s",s+1);
		for(j=1;j<=m;++j)
		if(s[j]=='1')add(i,j+n,1,-f[j]);
	}
	for(i=1;i<=n;++i)add(S,i,k,0),add(i,T,k-1,0);
	mincf();print(ans);puts("");
	return 0;
}
/*
3 3 1
1 2 3
3 2 1
111
111
111

-2
*/