【HDU】4307 Matrix 最小割——项目分配问题

传送门：【HDU】4307 Matrix

题目分析：给这题跪了T U T。。。。

本题之所以可以用最小割解决，是因为他可以求解下列函数的最小值：

分析出自这篇文章：http://blog.csdn.net/weiguang_123/article/details/8077385

还是自己码两行加深一下印象好了= =。。。

因为Xi只能取0或1，所以最终我们可以将所有的Xi分成两个集合：Xi取1的集合，Xi取0的集合。

假设源点S和Xi取1的集合相连，汇点T和Xi取0的集合相连。

一开始，我们先将所有的Xi和源点S以及汇点T相连，表示Xi的取值还未确定。

由公式可知Xi选择1会产生ai的花费，Xi选择0会产生bi的花费，Xi和Xj的选择不同会产生cij的花费。

如果Xi取0，也就是说Xi不可以取1，那么需要割掉Xi->T的边，因为割掉Xi->T的代价为bi，所以Xi->T的边容量为bi。

如果Xi取1，也就是说Xi不可以取0，那么需要割掉S->Xi的边，因为割掉S->Xi的代价为ai，所以S->Xi的边容量为ai。

现在，如果Xi取1，Xj取0，因为i，j不能同属一个集合，所以还需要割掉i->j，由上式可知，割掉i->j的边会产生cij的代价，所以i->j的边容量为cij。

最后跑一遍最小割即可得到该函数的最小值（通过网络流来决定Xi的取值）。

dfs一遍即可求出每个Xi所属于的集合，即S->Xi为割边则Xi取值为1，Xi->T为割边则Xi取值为0。

现在根据题意我们可以很直接的得到一个函数：

然后我们将其继续分解，使其逼近上面的函数：

得到函数一个这样的形式以后，很惊讶的可以发现，等式右端括号里的部分和f函数的定义惊人的相似！

当ai取0的时候，需要割掉代价为sum{bij}(1 <= j <= N)的S->Xi边，当ai取1的时候需要割掉代价为ci的Xi->T边，当ai取1且aj取0时需要割掉代价为bij的i->j边。

这个我们跑一遍最小割即括号中函数的最小值，用bij的总价值减去最小割的值即最大的D。

PS：公式编辑器搞了好久还没搞出来什么像样的东东。。。只能将人家的原样搞过来了。。。T U T。。

代码如下：

#include <cmath>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std ;

#define REP( i , a , b ) for ( int i = a ; i < b ; ++ i )
#define REV( i , a , b ) for ( int i = a - 1 ; i >= b ; -- i )
#define FOR( i , a , b ) for ( int i = a ; i <= b ; ++ i )
#define FOV( i , a , b ) for ( int i = a ; i >= b ; -- i )
#define CLR( a , x ) memset ( a , x , sizeof a )
#define CPY( a , x ) memcpy ( a , x , sizeof a )

typedef long long LL ;
typedef int type_c ;
typedef int type_f ;

const int MAXN = 1005 ;
const int MAXQ = 1005 ;
const int MAXE = 2333333 ;
const int INF = 0x3f3f3f3f ;

struct Edge {
	int v , n ;
	type_c c , rc ;
	Edge () {}
	Edge ( int v , type_c c , int n ) : v ( v ) , c ( c ) , n ( n ) {}
} ;

struct Net {
	Edge E[MAXE] ;
	int H[MAXN] , cntE ;
	int d[MAXN] , cur[MAXN] , pre[MAXN] , num[MAXN] ;
	int Q[MAXQ] , head , tail ;
	int s , t , nv ;
	type_f flow ;
	
	int n ;
	
	void init () {
		cntE = 0 ;
		CLR ( H , -1 ) ;
	}
	
	void addedge ( int u , int v , type_c c , type_c rc = 0 ) {
		E[cntE] = Edge ( v ,  c , H[u] ) ;
		H[u] = cntE ++ ;
		E[cntE] = Edge ( u , rc , H[v] ) ;
		H[v] = cntE ++ ;
	}
	
	void rev_bfs () {
		CLR ( d , -1 ) ;
		CLR ( num , 0 ) ;
		head = tail = 0 ;
		Q[tail ++] = t ;
		d[t] = 0 ;
		num[d[t]] = 1 ;
		while ( head != tail ) {
			int u = Q[head ++] ;
			for ( int i = H[u] ; ~i ; i = E[i].n ) {
				int v = E[i].v ;
				if ( d[v] == -1 ) {
					Q[tail ++] = v ;
					d[v] = d[u] + 1 ;
					num[d[v]] ++ ;
				}
			}
		}
	}
	
	type_f ISAP () {
		CPY ( cur , H ) ;
		rev_bfs () ;
		flow = 0 ;
		int u = pre[s] = s , i , pos , mmin ;
		while ( d[s] < nv ) {
			if ( u == t ) {
				type_f f = INF ;
				for ( i = s ; i != t ; i = E[cur[i]].v )
					if ( f > E[cur[i]].c ) {
						f = E[cur[i]].c ;
						pos = i ;
					}
				for ( i = s ; i != t ; i = E[cur[i]].v ) {
					E[cur[i]].c -= f ;
					E[cur[i] ^ 1].c += f ;
				}
				u = pos ;
				flow += f ;
			}
			for ( i = cur[u] ; ~i ; i = E[i].n )
				if ( E[i].c && d[u] == d[E[i].v] + 1 )
					break ;
			if ( ~i ) {
				cur[u] = i ;
				pre[E[i].v] = u ;
				u = E[i].v ;
			}
			else {
				if ( 0 == -- num[d[u]] )
					break ;
				for ( mmin = nv , i = H[u] ; ~i ; i = E[i].n )
					if ( E[i].c && mmin > d[E[i].v] ) {
						mmin = d[E[i].v] ;
						cur[u] = i ;
					}
				d[u] = mmin + 1 ;
				num[d[u]] ++ ;
				u = pre[u] ;
			}
		}
		return flow ;
	}
	
	void solve () {
		int sum = 0 , x ;
		scanf ( "%d" , &n ) ;
		init () ;
		s = 0 , t = n + 1 , nv = t + 1 ;
		FOR ( i , 1 , n ) {
			int tmp = 0 ;
			FOR ( j , 1 , n ) {
				scanf ( "%d" , &x ) ;
				tmp += x ;
				addedge ( i , j , x ) ;
			}
			sum += tmp ;
			addedge ( s , i , tmp ) ;
		}
		FOR ( i , 1 , n ) {
			scanf ( "%d" , &x ) ;
			addedge ( i , t , x ) ;
		}
		printf ( "%d\n" , sum - ISAP () ) ;
	}
} e ;

int main () {
	int T ;
	scanf ( "%d" , &T ) ;
	while ( T -- )
		e.solve () ;
	return 0 ;
}