[CodeVs1227]方格取数2(最大费用最大流)

　　网络流24题的坑还没填完，真的要TJ？

　　题目大意：一个n*n的矩阵,每格有点权,从(1,1)出发,可以往右或者往下走,最后到达(n,n),每达到一格,把该格子的数取出来,该格子的数就变成0,这样一共走K次,现在要求K次所达到的方格的数的和最大。

　　啊简单的费用流。每个点i拆成i和i'，连一条容量为1的边价值为点权，再连一条容量inf的边价值为0来让这个点能被经过，然后S连(1,1)容量k价值0，i'和右、下的点连容量inf价值0的边，(n,n)'连T容量inf价值0，跑最大费用最大流。

　　MDZZ看见n50我就开50，这是矩阵啊喂

#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<queue>
using namespace std;
const int inf=,maxn=;
struct poi{int dis,pos;};
struct zs{int too,pre,c,cf,v,f;}e[];
priority_queue<poi>q;
bool operator <(poi a,poi b){return a.dis<b.dis;};
int n,k,z,tot,sum,ans;
int last[maxn],dist[maxn],pre[maxn];
bool v[maxn];
void read(int &k)
{
    k=;int f=;char c=getchar();
    while(c<''||c>'')c=='-'&&(f=-),c=getchar();
    while(c<=''&&c>='')k=k*+c-'',c=getchar();
    k*=f;
}
void add(int x,int y,int c,int v)
{
    e[++tot].too=y;e[tot].c=e[tot].cf=c;e[tot].v=v;
    e[tot].pre=last[x];last[x]=tot;
    e[++tot].too=x;e[tot].v=-v;
    e[tot].pre=last[y];last[y]=tot;
}
void spfa(int x)
{
    for(int i=;i<=sum;i++)pre[i]=-,dist[i]=-inf,v[i]=;
    dist[x]=;v[x]=;q.push((poi){,x});
    while(!q.empty())
    {
        int now=q.top().pos;q.pop();
        for(int i=last[now],too=e[i].too;i;i=e[i].pre,too=e[i].too)
        if(e[i].cf)
        if(dist[too]<dist[now]+e[i].v)
        {
            dist[too]=dist[now]+e[i].v;pre[too]=i;
            if(!v[too])v[too]=,q.push((poi){dist[too],too});
        }
        v[now]=;
    }
}
void ford(int s,int t)
{
    spfa(s);
    while(pre[t]!=-)
    {
        int mincf=inf;
        for(int i=pre[t];i!=-;i=pre[e[i^].too])
        mincf=min(mincf,e[i].cf);
        ans+=dist[t]*mincf;
        for(int i=pre[t];i!=-;i=pre[e[i^].too])
        {
            e[i].f+=mincf;e[i^].f=-e[i].f;
            e[i].cf-=mincf;e[i^].cf+=mincf;
        }
        spfa(s);
    }
}
int num(int x,int y){return (x-)*n+y;}
int main()
{
    tot=;
    read(n);read(k);sum=*n*n+;
    add(,,k,);add(*n*n,sum,k,);
    for(int i=;i<=n;i++)
    for(int j=;j<=n;j++)
    {
        read(z);
        add(num(i,j),num(i,j)+n*n,,z);
        add(num(i,j),num(i,j)+n*n,inf,);
        if(i!=n)add(num(i,j)+n*n,num(i+,j),inf,);
        if(j!=n)add(num(i,j)+n*n,num(i,j+),inf,);
    }
    ford(,sum);
    printf("%d\n",ans);
}