CSU 2073: Tile Cut （最大流）

题目链接：http://acm.csu.edu.cn/csuoj/problemset/problem?pid=2073

题目大意：给出一个m*n的矩阵，矩阵中只有‘W’，‘I’，‘N’，三种字符，要你求出在这个m*n的矩阵中能连出多少个独立的‘WIN’（一个字符只能向上、下、左、右四个方向连接，并且一个字符只能使用一次）；

题目思路：本题就是一个很明显的最大流问题，首先将m*n的图上的每个点都标上一个独立的编号。大致建图思路就是建一个超级源点，在超级源点和所有的‘W’连上一条权值为1的边，因为一个字符只能用一次，那么我们就可以将字符为‘I’的点拆成两个点（一个入点，一个出点），在两点之间建一条权值为1的边，这样就可以限制每个字符的使用次数了，接着就把所有的‘W’和所有‘I’的入点连接上一条权值为1的边，把所有的‘N’和所有的‘I’的出点连上一条边，最后再建一个超级汇点，把所有的‘N’和超级汇点之间建立一条权值为1的边。从超级源点到超级汇点之间跑一次最大流就可以得到结果了。

ps：这题的读入有点恶心，研究 了半天才搞懂要怎么弄，不懂得小伙伴可以稍微借鉴下我的写法（当然如果有更好的写法可以相互交流下）

AC代码如下：

#include <map>
#include <set>
#include <cmath>
#include <vector>
#include <queue>
#include <stack>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long LL;
typedef pair<int,int>pii;
const int inf = 0x3f3f3f3f;
const int MX = 2005;
const int MXE = 40 * MX;
struct MaxFlow {
    struct Edge {
        int v, w, nxt;
    } edge[MXE];
    int tot, num, s, t;
    int head[MX];
    void init() {
        memset(head, -1, sizeof(head));
        tot = 0;
    }
    void add(int u, int v, int w) {
        edge[tot].v = v;
        edge[tot].w = w;
        edge[tot].nxt = head[u];
        head[u] = tot++;

        edge[tot].v = u;
        edge[tot].w = 0;
        edge[tot].nxt = head[v];
        head[v] = tot++;
    }
    int  d[MX], vis[MX], gap[MX];
    void bfs() {
        memset(d, 0, sizeof(d));
        memset(gap, 0, sizeof(gap));
        memset(vis, 0, sizeof(vis));
        queue<int>q;
        q.push(t);
        vis[t] = 1;
        while (!q.empty()) {
            int u = q.front();
            q.pop();
            for (int i = head[u]; ~i; i = edge[i].nxt) {
                int v = edge[i].v;
                if (!vis[v]) {
                    d[v] = d[u] + 1;
                    gap[d[v]]++;
                    q.push(v);
                    vis[v] = 1;
                }
            }
        }
    }
    int last[MX];
    int dfs(int u, int f) {
        if (u == t) return f;
        int sap = 0;
        for (int i = last[u]; ~i; i = edge[i].nxt) {
            int v = edge[i].v;
            if (edge[i].w > 0 && d[u] == d[v] + 1) {
                last[u] = i;
                int tmp = dfs(v, min(f - sap, edge[i].w));
                edge[i].w -= tmp;
                edge[i ^ 1].w += tmp;
                sap += tmp;
                if (sap == f) return sap;
            }
        }
        if (d[s] >= num) return sap;
        if (!(--gap[d[u]])) d[s] = num;
        ++gap[++d[u]];
        last[u] = head[u];
        return sap;
    }
    int solve(int st, int ed, int n) {
        int flow = 0;
        num = n;
        s = st;
        t = ed;
        bfs();
        memcpy(last, head, sizeof(head));
        while (d[s] < num) flow += dfs(s, inf);
        return flow;
    }
} F;

char mp[105][105];
int vis[105][105];
int to[4][2]={0,1,0,-1,1,0,-1,0};

int main(){
    //freopen("in.txt","r",stdin);
    int flag=1;
    while(flag&&gets(mp[0])){
        int n=1,m;
        flag=0;
        while(gets(mp[n])){
            if(mp[n][0]==0){
                flag=1;
                break;
            }
            if(mp[n][0]==EOF) break;
            n++;
        }
        m=strlen(mp[0]);
        memset(vis,0,sizeof(vis));
        F.init();
        int st=0,ed=2*n*m+1;
        int cnt=0;
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                if(mp[i][j]=='W')
                    F.add(st,i*m+j+1,1);
                if(mp[i][j]=='N')
                    F.add(i*m+j+1,ed,1);
                if(mp[i][j]=='I'){
                    F.add(i*m+j+1,i*m+j+n*m+1,1);
                    for(int k=0;k<4;k++){
                        int nx=i+to[k][0];
                        int ny=j+to[k][1];
                        if(nx<0||nx>=n||ny<0||ny>=m) continue;
                        if(mp[nx][ny]=='W'){
                            F.add(nx*m+ny+1,i*m+j+1,1);
                            vis[i][j]=1;
                            vis[nx][ny]=1;
                        }
                        if(mp[nx][ny]=='N'){
                            F.add(i*m+j+1+n*m,nx*m+ny+1,1);
                            vis[i][j]=1;
                            vis[nx][ny]=1;
                        }
                    }
                }
            }
        }
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                if(vis[i][j]){
                    if(mp[i][j]=='I') cnt+=2;
                    else cnt++;
                }
            }
        }
        int ans=F.solve(st,ed,cnt+2);
        printf("%d\n",ans);
    }
    return 0;
}
/**********************************************************************
	Problem: 2073
	User: 1264082032
	Language: C++
	Result: AC
	Time:8 ms
	Memory:3064 kb
**********************************************************************/