gym 101755D Transfer Window 【网络流】【dfs】

【题意】：

链接：gym 101755 D Transfer Window

给你一个n和一个k，表示一共有n个人，由1...n表示。 你有k个人，你想要k个人。第二行给出k个数，表示你有的k个人的id，第三行给出k个数，表示你想要的k个人的id，然后给你n*n的矩阵，第i行的第j个字符为0表示 i 不能换到 j，为1表示 i 能换到 j 。

如果能换到你想要的k个人，输出YES，然后下一行输出一个整数m，接下来输出m行交换方案i j，表示用 i 去换 j 。按照你的方案一步一步换下来可以达到结果，中间不能有非法的操作，不要求交换次数最少。

不能换到则输出NO。

【题解】：

用0表示 源点，用n+1表示汇点。

将0向你有的k个人连接一条容量为1的有向边。将你想要的k个人向n+1连接一条容量为1的有向边。

然后如果 i 能换到 j ，那么从 i 向 j 连接一条容量为无限的有向边，这里的无限可以假定为 n，因为最大的流量为n。

然后跑一个dinic最大流，如果最大流等于k，说明是YES，不然是NO。

如何输出方案？

如果我们直接在原来的网络上进行搜索，因为最多有9000条边，应该会TLE，所以先在之前的所有边里把有流量的边拿出来，重新建图。然后在新的图里进行深搜。

将一开始有的k个人标记为1，表示拥有。然后从 1 for 到 n，如果拥有这个点，那么对这个点进行dfs。

for 一遍 当前点连出去的所有边，如果这个边到达的是汇点，那么continue，看下一条。

如果这条边有流量，那么首先该流量减1（避免死循环，总的流量数象征着总的复杂度），然后看一下连向的点是否已经拥有，如果没有拥有，那么把当前的u 和 v 放到存答案的vector里，然后再搜索v。如果已经拥有v，先搜索v，再把u 和 v 放到vector里。注意放答案和搜索的执行顺序！

最后输出vector里的答案即可。

【代码】：

有点丑，见谅。

#include<stdio.h>
#include<math.h>
#include<vector>
#include<string.h>
#include<queue>
using namespace std;
typedef long long ll;
#define N 1000
#define INF 100000000
ll n,k;
int deg[500];


struct Edge
{
    int from,to,cap,flow;
    Edge(){};
    Edge(int u,int v,int c,int f):from(u),to(v),cap(c),flow(f){}
};
struct myEdge{
    int from,to,flow,nxt;
    myEdge(){};
    myEdge(int u,int v,int f):from(u),to(v),flow(f){}
}myedge[100000];
queue<myEdge> q;
int cnt=1;
int head[1000];
vector<pair<int,int> > myans;
void build(int u,int v,int w)
{
    myedge[cnt]=myEdge(u,v,w);
    myedge[cnt].nxt=head[u];
    head[u]=cnt++;
}
struct Dinic
{
    int n,m,s,t;//结点数，边数（包括反向弧），源点编号，汇点编号
    vector<Edge>edges;//边表，dges[e]和dges[e^1]互为反向弧
    vector<int>G[N];//邻接表，G[i][j]表示结点i的第j条边在e数组中的编号
    bool vis[N]; //BFS的使用
    int d[N]; //从起点到i的距离
    int cur[N]; //当前弧下标

    void addedge(int from,int to,int cap)
    {
        edges.push_back(Edge(from,to,cap,0));
        edges.push_back(Edge(to,from,0,0));
        int  m=edges.size();
        G[from].push_back(m-2);
        G[to].push_back(m-1);
    }

    bool bfs()
    {
        memset(vis,0,sizeof(vis));
        queue<int>Q;
        Q.push(s);
        d[s]=0;
        vis[s]=1;
        while(!Q.empty())
        {
            int x=Q.front();Q.pop();
            for(int i=0;i<G[x].size();i++)
            {
                Edge&e=edges[G[x][i]];
                if(!vis[e.to]&&e.cap>e.flow)//只考虑残量网络中的弧
                {
                    vis[e.to]=1;
                    d[e.to]=d[x]+1;
                    Q.push(e.to);
                }
            }

        }
        return vis[t];
    }

    int dfs(int x,int a)//x表示当前结点，a表示目前为止的最小残量
    {
        if(x==t||a==0)return a;//a等于0时及时退出，此时相当于断路了
        int flow=0,f;
        for(int&i=cur[x];i<G[x].size();i++)//从上次考虑的弧开始，注意要使用引用，同时修改cur[x]
        {
            Edge&e=edges[G[x][i]];//e是一条边
            if(d[x]+1==d[e.to]&&(f=dfs(e.to,min(a,e.cap-e.flow)))>0)
            {
                e.flow+=f;
                edges[G[x][i]^1].flow-=f;
                flow+=f;
                a-=f;
                if(!a)break;//a等于0及时退出，当a!=0,说明当前节点还存在另一个曾广路分支。

            }
        }
        return flow;
    }

    int Maxflow(int s,int t)//主过程
    {
        this->s=s,this->t=t;
        int flow=0;
        while(bfs())//不停地用bfs构造分层网络，然后用dfs沿着阻塞流增广
        {
            memset(cur,0,sizeof(cur));
            flow+=dfs(s,INF);
        }
        return flow;
    }
}dinic;
void dfsans(int rt)
{
    for(int i=head[rt];i;i=myedge[i].nxt){
        if(myedge[i].to==n+1) continue;
        if(myedge[i].flow>0){
            myedge[i].flow--;
            if(deg[myedge[i].to]==1){
                dfsans(myedge[i].to);
                myans.push_back(make_pair(rt,myedge[i].to));
                deg[rt]=0;
                deg[myedge[i].to]=1;
            }
            else{
                myans.push_back(make_pair(rt,myedge[i].to));
                deg[rt]=0;deg[myedge[i].to]=1;
                dfsans(myedge[i].to);
            }
            return;
        }
    }
}
int main(void)
{
    scanf("%lld%lld",&n,&k);
    int temp;
    for(int i=0;i<k;i++){
        scanf("%d",&temp);
        deg[temp]=1;
        dinic.addedge(0,temp,1);
    }
    for(int i=0;i<k;i++){
        scanf("%d",&temp);
        dinic.addedge(temp,n+1,1);
    }
    char ch[500];
    for(int i=0;i<n;i++){
        scanf("%s",ch);
        for(int j=0;j<n;j++){
            if(ch[j]=='1'&&j!=i)
                dinic.addedge(i+1,j+1,n);
        }
    }
    int v=dinic.Maxflow(0,n+1);
    if(v==k){
        printf("YES\n");
        int s=dinic.edges.size();
        for(int i=0;i<s;i++){
            if(dinic.edges[i].flow>0){
                build(dinic.edges[i].from,dinic.edges[i].to,dinic.edges[i].flow);
            }
        }
        for(int i=1;i<=n;i++){
            if(deg[i]==1)
            dfsans(i);
        }
        printf("%d\n",myans.size());
        for(int i=0;i<myans.size();i++){
            printf("%d %d\n",myans[i].first,myans[i].second);
        }
    }
    else{
        printf("NO\n");
    }
    return 0;
}