首页 > 代码库 > 【bzoj3442】学习小组 费用流
【bzoj3442】学习小组 费用流
原文地址:http://www.cnblogs.com/GXZlegend/p/6809670.html
题目描述
【背景】
坑校准备鼓励学生参加学习小组。
【描述】
共有n个学生,m个学习小组,每个学生有一定的喜好,只愿意参加其中的一些学习小组,但是校领导为学生考虑,规定一个学生最多参加k个学习小组。财务处的大叔就没那么好了,他想尽量多收钱,因为每个学生参加学习小组都要交一定的手续费,不同的学习小组有不同的手续费。然而,事与愿违,校领导又决定对学习小组组织者进行奖励,若有a个学生参加第i个学习小组,那么给这个学习小组组织者奖励Ci*a^2元。在参与学生(而不是每个学习小组的人数总和)尽量多的情况下,求财务处最少要支出多少钱(若为负数,则输出负数)(支出=总奖励费-总手续费)。
输入
输入有若干行,第一行有三个用空格隔开的正整数n、m、k。接下来的一行有m个正整数,表示每个Ci。第三行有m个正整数,表示参加每个学习小组需要交的手续费Fi。再接下来有一个n行m列的矩阵,表若第i行j列的数字是1,则表示第i个学生愿意参加第j个学习小组,若为0,则为不愿意。
输出
输出只有一个整数,为最小的支出。
样例输入
3 3 1
1 2 3
3 2 1
111
111
111
样例输出
-2
题解
拆边+费用流
由于有Ci*a^2的存在,使得正常加边的费用流无法处理。我们可以加容量为1,费用为Ci*1、Ci*3、Ci*5、Ci*7、...的一堆边,这样在最小费用的前提下总花费满足题意。
每个学习小组向T连上述所说的边,S向每个学生连一条容量为k,费用为0的边,每个学生向其能参加的学习小组连一条容量为1,费用为Fi的边。
因为题目中描述“在参与学生(而不是每个学习小组的人数总和)尽量多的情况下”,指的是所有学生必须有流通过,但不必满流。
所以还要从每个学生向T连一条容量为k-1,费用为0的边,保证费用最小。
然后跑最小费用最大流即可。
#include <cstdio> #include <cstring> #include <queue> using namespace std; queue<int> q; int f[110] , head[210] , to[100000] , val[100000] , cost[100000] , next[100000] , cnt = 1 , s , t , dis[210] , from[210] , pre[210]; char str[110]; void add(int x , int y , int v , int c) { to[++cnt] = y , val[cnt] = v , cost[cnt] = c , next[cnt] = head[x] , head[x] = cnt; to[++cnt] = x , val[cnt] = 0 , cost[cnt] = -c , next[cnt] = head[y] , head[y] = cnt; } bool spfa() { int x , i; memset(dis , 0x3f , sizeof(dis)); memset(from , -1 , sizeof(from)); dis[s] = 0 , q.push(s); while(!q.empty()) { x = q.front() , q.pop(); for(i = head[x] ; i ; i = next[i]) if(val[i] && dis[to[i]] > dis[x] + cost[i]) dis[to[i]] = dis[x] + cost[i] , from[to[i]] = x , pre[to[i]] = i , q.push(to[i]); } return ~from[t]; } int mincost() { int i , k , ans = 0; while(spfa()) { k = 0x3f3f3f3f; for(i = t ; i != s ; i = from[i]) k = min(k , val[pre[i]]); ans += k * dis[t]; for(i = t ; i != s ; i = from[i]) val[pre[i]] -= k , val[pre[i] ^ 1] += k; } return ans; } int main() { int n , m , k , i , j , x; scanf("%d%d%d" , &n , &m , &k); s = 0 , t = n + m + 1; for(i = 1 ; i <= m ; i ++ ) { scanf("%d" , &x); for(j = 1 ; j <= n ; j ++ ) add(i + n , t , 1 , x * (2 * j - 1)); } for(i = 1 ; i <= m ; i ++ ) scanf("%d" , &f[i]); for(i = 1 ; i <= n ; i ++ ) { add(s , i , k , 0) , add(i , t , k - 1 , 0); scanf("%s" , str + 1); for(j = 1 ; j <= m ; j ++ ) if(str[j] == ‘1‘) add(i , j + n , 1 , -f[j]); } printf("%d\n" , mincost()); return 0; }
【bzoj3442】学习小组 费用流
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。