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
题解:
好题,一道非常好的思考题。
看似一道线性代数题,但是如果写出要求的东西的方程:
它就等价于有n个物品,若同时选i,j,会得到b(i,j)的奖励,但每个物品有ci的代价,求最大收益。
这时候就是一道裸的最小割了。
代码如下:
#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<math.h>
#include<queue>
#define ll long long
#define inf 0x7f7f7f7f
#define N 300005
#define M 2000005
using namespace std;
ll tot=1,a[M],p[N],next[M],cap[M];
int cur[N],level[N];
int n,T;
ll ans,sum;
void add(int x,int y,int z)
{
next[++tot]=p[x];a[tot]=y;p[x]=tot;cap[tot]=z;
next[++tot]=p[y];a[tot]=x;p[y]=tot;cap[tot]=0;
}
bool bfs()
{
memset(level,-1,sizeof(level));
queue<int> q;
level[0]=0;
q.push(0);
for(int i=0;i<=T;i++) cur[i]=p[i];
while(!q.empty())
{
int u=q.front();q.pop();
for(int v=p[u];v;v=next[v])
{
if(cap[v] && level[a[v]]<0)
{
level[a[v]]=level[u]+1;
q.push(a[v]);
}
}
}
if(level[T]<0) return 0;
return 1;
}
ll dfs(int now,int t,ll f)
{
if (!f || now==t) return f;
int s1=0,s;
for(int u=cur[now];u;u=next[u])
{
cur[now]=u;
if(level[a[u]]==level[now]+1 && (s=dfs(a[u],t,min(f,cap[u]))))
{
s1+=s;f-=s;
cap[u]-=s;cap[u^1]+=s;
if(!f) break;
}
}
return s1;
}
ll dinic(int s,int t)
{
ll flow=0,f=0;
while(bfs())
while((f=dfs(s,t,inf))>0) flow+=f;
return flow;
}
int main()
{
scanf("%d",&n);T=n*n+n+1;
int id=n,x;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
scanf("%d",&x);id++;
add(i,id,inf);add(j,id,inf);
add(id,T,x);sum+=(ll)x;
}
for(int i=1;i<=n;i++)
{
scanf("%d",&x);
add(0,i,x);
}
printf("%lld\n",sum-dinic(0,T));
return 0;
}