BZOJ_3996_[TJOI2015]线性代数_最大权闭合子图

BZOJ_3996_[TJOI2015]线性代数_最大权闭合子图

Description

给出一个N*N的矩阵B和一个1*N的矩阵C。求出一个1*N的01矩阵A.使得

D=（A*B-C）*A^T最大。其中A^T为A的转置。输出D

Input

第一行输入一个整数N，接下来N行输入B矩阵，第i行第J个数字代表Bij. 接下来一行输入N个整数，代表矩阵C。矩阵B和矩阵C中每个数字都是不超过1000的非负整数。

Output

输出最大的D

Sample Input

3
1 2 1
3 1 0
1 2 3
2 3 7

Sample Output

2

HINT

 1<=N<=500

根据乘法分配律可知，对于$b(i,j)$ ，只有$a[i],a[j]$ 都选才会有贡献。

而选择$a[j]$会导致选择$-c[j]$.

可以发现这是个最大权闭合子图的模型。

$S->b[i][j],b[i][j]->c[i],b[i][j]->c[j],c[i]->T$

 

代码:

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define N 300050
#define M 2000050
#define inf 100000000
int head
,to[M],nxt[M],flow[M],cnt=1,sum,n;
int dep
,Q
,l,r,S,T,idx[510][510],c[510];
inline void add(int u,int v,int f) {
to[++cnt]=v; nxt[cnt]=head[u]; head[u]=cnt; flow[cnt]=f;
to[++cnt]=u; nxt[cnt]=head[v]; head[v]=cnt; flow[cnt]=0;
}
bool bfs() {
memset(dep,0,sizeof(dep));
l=r=0;Q[r++]=S;dep[S]=1;
while(l<r) {
int x=Q[l++],i;
for(i=head[x];i;i=nxt[i]) {
if(!dep[to[i]]&&flow[i]) {
dep[to[i]]=dep[x]+1;
if(to[i]==T) return 1;
Q[r++]=to[i];
}
}
}
return 0;
}
int dfs(int x,int mf) {
if(x==T) return mf;
int i,nf=0;
for(i=head[x];i;i=nxt[i]) {
if(dep[to[i]]==dep[x]+1&&flow[i]) {
int tmp=dfs(to[i],min(mf-nf,flow[i]));
if(!tmp) dep[to[i]]=0;
nf+=tmp;
flow[i]-=tmp;
flow[i^1]+=tmp;
if(nf==mf) break;
}
}
return nf;
}
void dinic() {
int f;
while(bfs()) while(f=dfs(S,inf)) sum-=f;
printf("%d\n",sum);
}
int main() {
int i,j,x;
scanf("%d",&n);
S=n*n+n+1;T=S+1;
int tot=0;
for(i=1;i<=n;i++) {
for(j=1;j<=n;j++) {
idx[i][j]=++tot;
scanf("%d",&x);
sum+=x;
add(S,tot,x);
}
}
for(i=1;i<=n;i++) {
scanf("%d",&c[i]);
add(i+n*n,T,c[i]);
}
for(i=1;i<=n;i++) {
for(j=1;j<=n;j++) {
add(idx[i][j],n*n+i,inf);
add(idx[i][j],n*n+j,inf);
}
}
dinic();
}