P3356 火星探险问题

• 点权版的深海机器人问题，主意输出方案的判断条件不要写错！这里的方法是把残余网络的剩余流量转换成实际流量。

#include <iostream>
#include <cstdio>
#include <queue>
#include <cstring>
using namespace std;
const int maxn=4005,maxm=20005,inf=0x7f7f7f7f;
int n,a,b,g[40][40],last[maxn],cur[maxn],flow,ans,vis[maxn],num,st=4000,cnt=1,d[maxn],inq[maxn],po[maxn];
struct edge{
    int v,w,f,next,ff;
}e[maxm];
inline void add(int u,int v,int w,int f)
{
    e[++cnt].v=v;
    e[cnt].w=w;
    e[cnt].f=f;
    e[cnt].ff=f;
    e[cnt].next=last[u];
    last[u]=cnt;
    e[++cnt].v=u;
    e[cnt].w=-w;
    e[cnt].f=0;
    e[cnt].ff=0;
    e[cnt].next=last[v];
    last[v]=cnt;
}
bool spfa()
{
    queue<int>q;
    for(int i=0;i<=st;i++)
    {
        d[i]=-inf;
        inq[i]=0;
    }
    d[0]=0;
    q.push(0);
    while(!q.empty())
    {
        int u=q.front();q.pop();
        inq[u]=0;
        for(int i=last[u];i;i=e[i].next)
        {
            int v=e[i].v,f=e[i].f,w=e[i].w;
            if(f&&d[v]<d[u]+w)
            {
                d[v]=d[u]+w;
                if(!inq[v])
                {
                    inq[v]=1;
                    q.push(v);
                }
            }
        }
    }
    return d[st]!=-inf;
}
int dfs(int u,int dis)
{
    vis[u]=1;
    if(u==st||!dis)
    return dis;
    for(int i=cur[u];i;i=e[i].next)
    {
        cur[u]=e[i].next;
        int v=e[i].v,f=e[i].f,w=e[i].w;
        if(d[v]==d[u]+w&&f&&!vis[v])
        {
            int di=dfs(v,min(dis,f));
            if(di>0)
            {
                e[i].f-=di;
                e[i^1].f+=di;
                return di;
            }
        }
    }
    return 0;
}
void dinic()
{
    while(spfa())
    {
        for(int i=0;i<=st;i++)
        cur[i]=last[i];
        memset(vis,0,sizeof(vis));
        while(int dd=dfs(0,inf))
        {
            flow+=dd;
            ans+=d[st];
            memset(vis,0,sizeof(vis));
        }
    }
}
bool ot(int u,int fa,int nu)
{
    if(u==num||u==num*2) return 1;
    for(int i=last[u];i;i=e[i].next)
    {
        int v=e[i].v,f=e[i].f;
        if(v==fa) continue;
        if(f>0)
        {
            int uu=u>num?u-num:u,vv=v>num?v-num:v;
            if(vv-uu!=1&&vv-uu!=a&&uu!=vv) continue;
            e[i].f--;
            bool p=0,o=0;
            if(uu==vv||u==0||v==0)
            o=1;
            int dire=vv-uu==1?1:0;
            if(!o)
            cout<<nu<<' '<<dire<<endl;
            p=ot(v,u,nu);
            if(p) return 1;

        }
    }
}
int main()
{
    scanf("%d%d%d",&n,&a,&b);
    num=a*b;
    for(int i=1;i<=b;i++)
    for(int j=1;j<=a;j++)
    scanf("%d",&g[i][j]);
    add(0,1,0,n);
    add(num,st,0,n);
    add(num*2,st,0,n);
    for(int i=1;i<=b;i++)
    {
        int poi=(i-1)*a;
        for(int j=1;j<=a;j++)
        {
            if(g[i][j]==1) continue;
            if(g[i][j]==2)
            {
                add(poi+j,poi+num+j,1,1);
                if(j<a&&g[i][j+1]!=1)
                add(poi+num+j,poi+j+1,0,inf);
                if(i<b&&g[i+1][j]!=1)
                add(poi+num+j,poi+a+j,0,inf);
            }
            if(j<a&&g[i][j+1]!=1)
            add(poi+j,poi+j+1,0,inf);
            if(i<b&&g[i+1][j]!=1)
            add(poi+j,poi+a+j,0,inf);
        }
    }
    dinic();
    for(int i=1;i<=cnt;i++)
    {
        e[i].f=e[i].ff-e[i].f;
        po[e[i].v]=e[i].f;
    }
    for(int i=1;i<=n;i++)
    ot(0,-1,i);
    return 0;
}