cogs742 [网络流24题]深海机器人

一道很简单的费用流 这个题的坐标和一般题好像不太一样 处理的时候要注意一下
其实不太正常的坐标就是这个题唯一的难点了 HHH

建图非常容易想 把原图里的边变成网络里的两条边 一条费用为原图中边权 容量为1 另一条费用为0 容量无穷大 然后原点向所有出发点连边 容量k费用0 所有到达点向汇点连边 容量r费用0

这个题我莫名其妙的WA了一个点 调了半天没找出来 结果把数组大小乘以4就过了
我果然还是太弱了啊= =
做了这么多网络流发现其实数组大小是个很关键的问题 能开大就尽量开大 不要相信自己的直觉 浪费内存又不扣分对吧对吧QvQ

最后我必须吹一波cogs这个OJ 好用美观 题也很全 =A=

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cstdlib>

using namespace std;

const int N=400*4,M=N*4,S=N-1,T=N-2,INF=1e9;

int a=0,b=0,p=0,q=0;
int head[N],next[M],to[M],from[M],c[M],f[M],cost[M],edge=0;
int queue[M],front=0,back=0,vis[N],fr[N],dis[N];

inline void addEdge(int u,int v,int ac,int acost) {
    to[edge]=v,from[edge]=u,c[edge]=ac,cost[edge]=acost,next[edge]=head[u],head[u]=edge++;  
    to[edge]=u,from[edge]=v,c[edge]=0,cost[edge]=-acost,next[edge]=head[v],head[v]=edge++;
}

inline int num(int i,int j) {
    ++i,++j;
    int ret=i+(j-1)*(q+1);
    return ret;
}

inline void build() {
    for (int i=0;i<=p;++i)
        for (int j=0;j<q;++j) {
            int x=0;
            scanf("%d",&x);
            int u=num(i,j),v=num(i,j+1);
            addEdge(u,v,1,x);
            addEdge(u,v,INF,0);
        }
    for (int i=0;i<=q;++i)
        for (int j=0;j<p;++j) {
            int x=0;
            scanf("%d",&x);
            int u=num(j,i),v=num(j+1,i);
            addEdge(u,v,1,x);
            addEdge(u,v,INF,0);
        }
    while (a--) {
        int x=0,y=0,k=0;
        scanf("%d %d %d",&k,&x,&y);
        addEdge(S,num(x,y),k,0);
    }
    while (b--) {
        int x=0,y=0,r=0;
        scanf("%d %d %d",&r,&x,&y);
        addEdge(num(x,y),T,r,0);
    }
} 

inline int spfa() {
    memset(vis,0,sizeof(vis));
    front=back=0;
    for (int i=1;i<N;++i)
        dis[i]=-INF;
    queue[back++]=S;
    dis[S]=0;
    vis[S]=1;   
    memset(fr,-1,sizeof(fr));
    while (front<back) {
        int x=queue[front++];
        for (int e=head[x];~e;e=next[e]) {
            int v=to[e];
            if (c[e]>f[e] && dis[x]+cost[e]>dis[v]) {
                dis[v]=dis[x]+cost[e];
                fr[v]=e;
                if (!vis[v]) {
                    queue[back++]=v;
                    vis[v]=1;
                }
            }
        }
        vis[x]=0;
    }
    return dis[T]!=-INF;
}

inline int augment() {
    int mnc=INF,ret=0;
    for (int e=fr[T];~e;e=fr[from[e]])
        mnc=min(mnc,c[e]-f[e]);
    for (int e=fr[T];~e;e=fr[from[e]]) {
        ret+=mnc*cost[e];
        f[e]+=mnc;
        f[e^1]-=mnc;
    }
    return ret;
}

inline int mcf() {
    int ret=0; 
    while (spfa())
        ret+=augment();
    return ret;
}

int main(void) {
    freopen("shinkai.in","r",stdin);
    freopen("shinkai.out","w",stdout);
    scanf("%d %d",&a,&b);
    scanf("%d %d",&p,&q);
    memset(head,-1,sizeof(head));
    build(); 
    printf("%d\n",mcf());
    fclose(stdin);
    fclose(stdout);
    return 0;
}