UVa11082——矩阵解压——网络流学习

问题来源：刘汝佳紫书P374——矩阵解压。
思路：还是调用刘汝佳大大的思路建模：建立二分图，X节点为第i行的和，Y节点为第i列的和，再建立一个超级源点和超级汇点，每个X与源点建边，容量为每行之和减C，每个Y与汇点建边，容量为每列之和减R，再将每个X与Y建边，容量为19（这样的好处看后面），这样，求出网络流所有的最大流，如果最后每条由S-T的流都是满载，则说明有解，解就是每个X与每个Y上的流量。
为什么这样是对的呢？这里我自己还是没分析出来，借用大佬们的思路吧。

1. 建边容量的选择
   因为网络流中是允许0流的，所以我们将所有元素减1，对应每行之和就是减C，每列减R，中间最大容量就是0~19，输出时加一即可。

2. 模型建立的选择
   这样建模是常用的矩阵建模之一，学习学习~
   首先，每个X节点只有一个入流，多个出流，出流之和等于入流，也就是第i行之和；每个Y节点有多个入流，只有一个出流，入流之和等于出流，也就是第i列之和。
   再考虑每一类弧的意义：第一类弧，以每一行所有元素之和为容量；第三类弧，以每一列所有元素之和为容量；第二类弧，最终流量为矩阵中的每一个元素。也就是说，第一类弧分成多个分支，每一个第一类弧都分出一个分支汇到同一个第三类弧。再考虑矩阵：把每一行的和分成多个元素，每一行的和都会分出一个元素排列在同一列，组成这一列的和。其中有很大的相似性，其实就是问题的变形，只要求出每一个第二类弧的流量，就是求出了矩阵中的每一个元素。

3. 算法实现的选择
   这里我还重温了Dinic算法，之前都是看模板，自己建模应用时还是有问题的。
   为什么用Dinic呢？Dinic会用DFS遍历层次图，直到形成阻塞流，而此题就要求必须每个从S—T的网络流都必须满载，这样就可以用Dinic构造的阻塞流判断每个流的cap是否等于flow，如果是，输出每个第二类弧的流量，如果否就是无解.

#include<iostream>
#include<queue>
#include<cstring>
#define maxn 1000

using namespace std;

const int INF=0x3f3f3f3f;

struct Edge
{
    int from,to,cap,flow;
    Edge(int u,int v,int c,int f):
        from(u),to(v),cap(c),flow(f) {}
};

struct Dinic
{
    int n,m,s,t;
    vector<Edge> edges;
    vector<int> G[maxn];
    int d[maxn];
    int cur[maxn];
    bool vis[maxn];

    void init(int n)
    {
        for (int i=0; i<n; i++)
            G[i].clear();
        edges.clear();
    }

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

    bool BFS()
    {
        memset(vis,false,sizeof(vis));
        for (int i=0; i<n; i++) d[i] = INF;
        d[s] = 0; vis[s] = true;

        queue<int> Q;
        Q.push(s);
        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] = true;
                    d[e.to] = d[x]+1;
                    Q.push(e.to);
                }
            }
        }
        return vis[t];
    }

    int DFS(int x,int a)
    {
        if (x == t || a == 0)
            return a;
        int flow = 0,f;
        for (int i=cur[x]; i<G[x].size(); i++)
        {
            Edge& e = edges[G[x][i]];
            if (d[e.to] == d[x]+1 && (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 == 0)
                    break;
            }
        }
        return flow;
    }

    bool OKA()
    {
        for (int i=0; i<G[s].size(); i++)
        {
            Edge e = edges[G[s][i]];
            if (e.cap!=e.flow)
                return false;
        }
        return true;
    }

    bool OKB(int R,int C)
    {
        for (int j=R+1; j<=R+C; j++)
        {
            Edge& e = edges[G[j][0]];
            if (e.cap!=e.flow)
                return false;
        }
        return true;
    }

    void Maxflow(int t,int R,int C)
    {
        int flow = 0;
        while (BFS())
        {
            memset(cur,0,sizeof(cur));
            flow += DFS(s,INF);
        }

        cout<<"Matrix "<<t<<endl;
        if (OKA() && OKB(R,C))
        {
            for (int i=1; i<=R; i++)
            {
                int j;
                for (j=1; j<G[i].size()-1; j++)
                    cout<<edges[G[i][j]].flow+1<<' ';
                cout<<edges[G[i][j]].flow+1<<endl;
            }
        }
        cout<<endl;
    }
};

int main()
{
    Dinic aa;
    int T,R,C,tmp;
    int a[30],b[30],c[30],d[30];

    cin>>T;
    tmp = T;
    while (T>0)
    {
        T--;
        aa.init(maxn);
        cin>>R>>C;
        for (int i=1; i<=R; i++) cin>>a[i];
        for (int i=1; i<=C; i++) cin>>b[i];
        for (int i=1; i<=R; i++) c[i] = a[i]-a[i-1]-C;
        for (int i=1; i<=C; i++) d[i] = b[i]-b[i-1]-R;

        for (int i=1; i<=R; i++)
            aa.Addedge(0,i,c[i]);
        for (int i=1; i<=C; i++)
            aa.Addedge(R+i,R+C+1,d[i]);
        for (int i=1; i<=R; i++)
            for (int j=1; j<=C; j++)
                aa.Addedge(i,R+j,19);
        aa.s = 0; aa.t = R+C+1;
        aa.Maxflow(tmp-T,R,C);
    }
    return 0;
}