BOJ)3654 L퍼즐

문제:icpc.me/3654

L퍼즐 모양으로 패턴을 완성시킨다고 가정하였을 경우 B 정점은 인접한 W정점중 하나는 홀수행의 정점에 하나는 짝수행의 정점에 매칭이 되어야 한다.

따라서 두개의 플로우 모델링을 하여 하나는 홀수행의 W들과 B의 최대 매칭 하나는 짝수행의 W들과 B의 최대 매칭을 구한 뒤

두 값과 B의 개수가 같은지와 모든 W의 수와 2*B가 같은지 두 경우를 검사해주면 된다.

#include <cstdio>
#include <algorithm>
#include <vector>
#include <queue>
#include <cstring>
#define INF (1<<30)
using namespace std;
struct Dinic {
    int n;
    struct Edge {
        int v, cap, rev;
        Edge(int v,int cap,int rev) :v(v), cap(cap), rev(rev) {}
    };
    vector<vector<Edge>> vt;
    vector<int> level, work;
    Dinic(int n) :n(n) {
        vt.resize(n + 1);
    }
    void addEdge(int s, int e, int c) {
        vt[s].emplace_back(e, c, vt[e].size());
        vt[e].emplace_back(s, 0, vt[s].size() - 1);
    }
    bool bfs(int src,int sink) {
        level.assign(n + 1, -1);
        queue<int> qu;
        qu.push(src);
        level[src] = 0;
        while (qu.size()) {
            int here = qu.front();
            qu.pop();
            for (auto next : vt[here]) {
                if (level[next.v] == -1 && next.cap > 0) {
                    level[next.v] = level[here] + 1;
                    qu.push(next.v);
                }
            }
        }
        return level[sink] != -1;
    }
    int dfs(int here, int crtcap, int target) {
        if (here == target)return crtcap;
        for (int &i = work[here]; i < vt[here].size(); i++) {
            int there = vt[here][i].v;
            int cap = vt[here][i].cap;
            if (cap > 0 && level[here] + 1 == level[there]) {
                int df = dfs(there, min(crtcap, cap), target);
                if (df > 0) {
                    vt[here][i].cap -= df;
                    vt[there][vt[here][i].rev].cap += df;
                    return df;
                }
            }
        }
        return 0;
    }
    int solve(int src, int sink) {
        int ret = 0;
        while (bfs(src, sink)) {
            work.assign(n + 1, 0);
            while (1) {
                int flow = dfs(src, INF, sink);
                if (!flow)break;
                ret += flow;
            }
        }
        return ret;
    }
};
int t, n, m;
char a[501][501];
int dx[] = { 0,0,1,-1 };
int dy[] = { 1,-1,0,0 };
bool chk(int x, int y) {
    return 0 <= x&&x < n && 0 <= y&&y < m;
}
int main() {
    scanf("%d", &t);
    while (t--) {
        scanf("%d%d", &n, &m);
        for (int i = 0; i < n; i++) {
            getchar();
            for (int j = 0; j < m; j++) 
                scanf("%c", &a[i][j]);
        }
        Dinic f(n*m + 3);
        Dinic g(n*m + 3);
        int srctosink = 0, white = 0;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                if (a[i][j] == 'B') {
                    f.addEdge(n*m + 1, i*m + j, 1);
                    g.addEdge(n*m + 1, i*m + j, 1);
                    srctosink++;
                    for (int k = 0; k < 4; k++) {
                        int cx = i + dx[k];
                        int cy = j + dy[k];
                        if (chk(cx, cy) && a[cx][cy] == 'W') {
                            if (cx % 2)
                                f.addEdge(i*m + j, cx*m + cy, 1);
                            else
                                g.addEdge(i*m + j, cx*m + cy, 1);
                        }
                    }
                }
                else if (a[i][j] == 'W') {
                    white++;
                    if (i % 2)
                        f.addEdge(i*m + j, n*m + 2, 1);
                    else
                        g.addEdge(i*m + j, n*m + 2, 1);
                }
            }
        }
        if (white == 2*srctosink&&f.solve(n*m + 1, n*m + 2) == srctosink&& g.solve(n*m + 1, n*m + 2) == srctosink)
            puts("YES");
        else
            puts("NO");
    }
    return 0;
}