3996: [TJOI2015]线性代数

题目链接

题目大意：给出一个N*N的矩阵B和一个1*N的矩阵C。求出一个1*N的01矩阵A，得
D=（A*B-C）*A^T最大。其中A^T为A的转置。输出D

题解：Orz GXZ

我的收获：大力推导……

#include <bits/stdc++.h>
using namespace std;

#define INF 0x7fffffff
const int M=612350;

int n,m,st,ed,flow,ans,t,id;
int head[M],last[M];
int num[M],d[M];
bool Exit;

struct edge{int to,c,nex;}e[M*9];

void add(int u,int v,int w){e[t]=(edge){v,w,head[u]};last[u]=head[u]=t++;}

void insert(int x,int y,int z){add(x,y,z),add(y,x,0);};

int dfs(int x,int in)
{
    if(x==ed) return in;
    int ans=0,f;
    for(int i=last[x];i!=-1;last[x]=i=e[i].nex)
    {
        int v=e[i].to;
        if(e[i].c&&d[v]==d[x]-1){
            f=dfs(v,min(in-ans,e[i].c));
            ans+=f;e[i].c-=f; e[i^1].c+=f;
            if(Exit||ans==in) return ans;
        }
    }
    if(--num[d[x]]==0) Exit=1;
    d[x]++,num[d[x]]++,last[x]=head[x];
    return ans;
}

void isap()
{
    Exit=0;flow=0;
    while(!Exit) flow+=dfs(st,INF);
}

void work()
{
    isap();ans-=flow;
    cout<<ans<<endl;
}

void init()
{
    memset(head,-1,sizeof(head));
    memset(last,-1,sizeof(last));
    int x;scanf("%d",&n);id=n;
    st=0,ed=n+n*n+1;num[0]=ed+1;
    for(int i=1;i<=n;i++) for(int j=1;j<=n;j++)
    {
        scanf("%d",&x);++id;
        insert(st,id,x);
        insert(id,i,INF);
        insert(id,j,INF);ans+=x;
    }
    for(int i=1;i<=n;i++) scanf("%d",&x),insert(i,ed,x);
}

int main()
{
    init();
    work();
    return 0;
}