1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136
| #include <bits/stdc++.h>
const int kMaxN = 2e3 + 5, kMaxM = 1e5 + 5; const int kD[][2] = {{0, 1}, {0, -1}, {1, 0}, {-1, 0}};
int n, m, k; int col[45][45], x[kMaxN], y[kMaxN], cnt[kMaxN]; bool del[kMaxN];
int getid(int x, int y) { return (x - 1) * m + y; }
namespace Dinic { struct Node { int v, w, pre;
Node(int _v = 0, int _w = 0, int _pre = 0) : v(_v), w(_w), pre(_pre) {} } e[kMaxM];
int n, s, t, tot = 1, tail[kMaxN], cur[kMaxN], dep[kMaxN]; bool vis[kMaxN];
void addedge(int u, int v, int w) { e[++tot] = {v, w, tail[u]}, tail[u] = tot; } void add(int u, int v, int w) { addedge(u, v, w), addedge(v, u, 0); } void init(int _n, int _s, int _t) { n = _n, s = _s, t = _t, tot = 1, std::fill_n(tail, n + 1, 0); }
bool bfs() { for (int i = 1; i <= n; ++i) cur[i] = tail[i], vis[i] = 0, dep[i] = 1e9; std::queue<int> q; q.emplace(s), dep[s] = 0, vis[s] = 1; for (; !q.empty();) { int u = q.front(); q.pop(); for (int i = tail[u]; i; i = e[i].pre) { int v = e[i].v, w = e[i].w; if (w && !vis[v]) { dep[v] = dep[u] + 1, vis[v] = 1, q.emplace(v); } } } return vis[t]; }
int dfs(int u, int lim) { if (u == t || !lim) return lim; int flow = 0; for (int &i = cur[u]; i; i = e[i].pre) { int v = e[i].v, w = e[i].w; if (w && dep[v] == dep[u] + 1) { int fl = dfs(v, std::min(lim, w)); if (!fl) dep[v] = 1e9; e[i].w -= fl, e[i ^ 1].w += fl; lim -= fl, flow += fl; if (!lim) break; } } return flow; }
int maxflow() { int ans = 0; for (; bfs(); ans += dfs(s, 1e9)) {} return ans; } }
void dickdreamer() { std::cin >> n >> m; int s = n * m + 1, t = n * m + 2; Dinic::init(n * m + 2, s, t); for (int i = 1; i <= n; ++i) { std::string str; std::cin >> str; str = " " + str; for (int j = 1; j <= m; ++j) { if (str[j] == '.') x[0] = i, y[0] = j, del[getid(i, j)] = 1; if (str[j] == 'O') col[i][j] = 0; else col[i][j] = 1; if (!col[i][j]) Dinic::add(s, getid(i, j), 1); else Dinic::add(getid(i, j), t, 1); } } std::cin >> k; for (int i = 1; i <= 2 * k; ++i) { std::cin >> x[i] >> y[i]; del[getid(x[i], y[i])] = 1; } for (int i = 1; i <= n; ++i) { for (int j = 1; j <= m; ++j) { if (col[i][j] || del[getid(i, j)]) continue; for (auto [di, dj] : kD) { int ti = i + di, tj = j + dj; if (col[ti][tj] && !del[getid(ti, tj)]) Dinic::add(getid(i, j), getid(ti, tj), 1); } } } cnt[2 * k + 1] = Dinic::maxflow(); for (int i = 2 * k; ~i; --i) { int u = getid(x[i], y[i]); del[u] = 0; for (auto [di, dj] : kD) { int tx = x[i] + di, ty = y[i] + dj, v = getid(tx, ty); if (tx < 1 || tx > n || ty < 1 || ty > m) continue; if (col[tx][ty] != col[x[i]][y[i]] && !del[v]) { if (!col[x[i]][y[i]]) Dinic::add(u, v, 1); else Dinic::add(v, u, 1); } } cnt[i] = Dinic::maxflow(); } std::vector<int> vec; for (int i = 1; i <= k; ++i) { if (cnt[2 * i - 2] && cnt[2 * i - 1]) vec.emplace_back(i); } std::cout << vec.size() << '\n'; for (auto x : vec) std::cout << x << '\n'; }
int32_t main() { #ifdef ORZXKR freopen("in.txt", "r", stdin); freopen("out.txt", "w", stdout); #endif std::ios::sync_with_stdio(0), std::cin.tie(0), std::cout.tie(0); int T = 1; while (T--) dickdreamer(); return 0; }
|