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
| #include <queue> #include <cstdio> #include <cstring> #include <iostream> #include <algorithm> using namespace std; const int MAXN = 55 * 55; const int INF = 0x3f3f3f3f;
int n, l, r, mid, fa[MAXN]; char opt[MAXN][MAXN]; bool con[MAXN][MAXN];
int st, ed, head[MAXN], ver[MAXN], edge[MAXN], nxt[MAXN], tot, d[MAXN], cur[MAXN]; queue<int> q;
void AddEdge(int u, int v, int c) { ver[++tot] = v, edge[tot] = c, nxt[tot] = head[u], head[u] = tot; ver[++tot] = u, edge[tot] = 0, nxt[tot] = head[v], head[v] = tot; }
bool bfs(int st) { memset(d, 0, sizeof(d)); while (!q.empty()) q.pop(); q.push(st), d[st] = 1, cur[st] = head[st]; while (!q.empty()) { int u = q.front(); q.pop(); for (int i = head[u]; i; i = nxt[i]) { int v = ver[i]; if (!edge[i] || d[v] != 0) continue; q.push(v); cur[v] = head[v]; d[v] = d[u] + 1; if (v == ed) return true; } }
return false;
}
int dinic(int u, int flow) { if (u == ed) return flow; int res = flow, k, i; for (i = cur[u]; i && res; i = nxt[i]) { int v = ver[i]; if (!edge[i] || d[v] != d[u] + 1) continue; k = dinic(v, min(res, edge[i])); if (k == 0) d[v] = 0; edge[i] -= k, edge[i ^ 1] += k; res -= k; cur[u] = i; } return flow - res; }
int ind(int op, int x) { return x + op * n; }
bool check(int mid) {
memset(head, 0, sizeof(head)); tot = 1; for (int i = 1; i <= n; i++) { AddEdge(st, ind(0, i), 1); if (fa[i]) { if (fa[i] > n) AddEdge(ind(0, i), ed, 1); else AddEdge(ind(0, i), ind(1, fa[i]), 1); } else if (con[0][i]) { AddEdge(i, ed, 1); } else { for (int j = 1; j <= n; j++) if (con[i][j]) { AddEdge(ind(0, i), ind(1, j), 1); } } AddEdge(ind(1, i), ed, mid); }
int res = 0, tmp = 0; while (bfs(st)) { while ((tmp = dinic(st, INF))) res += tmp; }
return res == n;
}
int main() {
scanf("%d", &n); st = (n << 1) + 1, ed = (n << 1) + 3;
for (int i = 0; i <= n; i++) { scanf("%s", opt[i] + 1); for (int j = 1; j <= n; j++) { con[i][j] = (opt[i][j] == 'Y'); } } for (int i = 1; i <= n; i++) con[n + 1][i] = con[0][i];
l = 0, r = n; while (l < r) { mid = (l + r) >> 1; if (check(mid)) r = mid; else l = mid + 1; }
for (int i = 1; i <= n; i++) {
bool flag = false; for (int j = 1; j <= n; j++) if (con[i][j]) { fa[i] = j; if (check(l)) { flag = true; break; } } if (!flag) printf("%d ", n), fa[i] = n + 1; else printf("%d ", fa[i] - 1); }
return 0; }
|