题意就是
给一个50 * 50的矩阵
然后给出每行每列元素的和
和一个初始矩阵
矩阵中有些是未知,有些是已知
然后我们求目标矩阵就是把能确定的元素的值求出来,实在不能确定的就置为-1
所有矩阵元素的值在0-100之间
看到范围很小。
第一反应是求一个最大流
先把已经给出的元素都从每行每列的和中减掉。
然后左边为行结点,右边为列结点
然后源点向行结点连边
列结点向汇点连边
行和列中如果对应的元素未知就连一下,流向上限是100
然后这样我们就得到了一个可行解
但是可能有多解怎么办
对于一个可能多解的元素
如果我们将这个元素的值固定住。
然后建立一个超级源点与该行结点连边。
该列结点与超级汇点连边。
流量都是1,
跑一遍看看有没有增广路。
如果有,显然这个位置的值是可以改变的,就是多解,然后我们把这个位置的元素值-1,因为我们刚才增广了,其他有元素的值增加了1,所以
为了保持流量的平衡,这个位置的元素要减1
但是这样还不行。
我们想想。
如果该位置的值现在是0怎么办。
他没法减掉1。
所以我们就要想想残余网络了。
既然他没法减掉1,就让他想办法+1,让别的元素-1去
那么我们可以用一个超级源点连接列结点。
行结点连接超级汇点
跑最大流,看有没有增广路。
也就是看他的残余网络能不能减掉1.即它自身+1
如果有增广路,跟之前一样,更新一下边
#include <iostream> #include <cstdio> #include <cstring> #include <string> #include <algorithm> #include <cstdlib> #include <cmath> #define MAXN 106 #define MAXM 211111 #define INF 1111111111 using namespace std; struct EDGE { int v, next; int w; }edge[MAXM]; int head[MAXN], e; void init() { memset(head, -1, sizeof(head)); e = 0; } void add(int u, int v, int w) { edge[e].v = v; edge[e].w = w; edge[e].next = head[u]; head[u] = e++; edge[e].v = u; edge[e].w = 0; edge[e].next = head[v]; head[v] = e++; } int n; int h[MAXN]; int gap[MAXN]; int src, des; int tt[111][111]; int dfs(int pos, int cost) { if(pos == des) return cost; int j, minh = n - 1; int lv = cost, d; for(j = head[pos]; j != -1; j = edge[j].next) { int v = edge[j].v; int w = edge[j].w; if(w > 0) { if(h[v] + 1 == h[pos]) { if(lv < edge[j].w) d = lv; else d = edge[j].w; d = dfs(v, d); edge[j].w -= d; edge[j ^ 1].w += d; lv -= d; if(h[src] >= n) return cost - lv; if(lv == 0) break; } if(h[v] < minh) minh = h[v]; } } if(lv == cost) { --gap[h[pos]]; if(gap[h[pos]] == 0) h[src] = n; h[pos] = minh + 1; ++gap[h[pos]]; } return cost - lv; } int sap() { int ret = 0; memset(gap, 0, sizeof(gap)); memset(h, 0, sizeof(h)); gap[0] = n; while(h[src] < n) ret += dfs(src, INF); return ret; } int nt, m; int col[55], row[55]; int val[55][55], id[55][55]; int ans[55][55]; int vis[55][55]; int main() { //freopen("C:/C.in", "r", stdin); //freopen("C:/C2.out", "w", stdout); while(scanf("%d%d", &nt, &m) != EOF) { if(!nt && !m) break; for(int i = 1; i <= nt; i++) for(int j = 1; j <= m; j++) scanf("%d", &val[i][j]); for(int i = 1; i <= nt; i++) scanf("%d", &row[i]); for(int i = 1; i <= m; i++) scanf("%d", &col[i]); memset(ans, -1, sizeof(ans)); for(int i = 1; i <= nt; i++) for(int j = 1; j <= m; j++) { if(val[i][j] != -1) { row[i] -= val[i][j]; col[j] -= val[i][j]; ans[i][j] = val[i][j]; } } init(); src = nt + m + 1; des = nt + m + 2; n = des; int S = nt + m + 3; int T = nt + m + 4; for(int i = 1; i <= nt; i++) for(int j = 1; j <= m; j++) { if(ans[i][j] == -1) { id[i][j] = e; add(i, j + nt, 100); } } for(int i = 1; i <= nt; i++) { add(src, i, row[i]); } for(int j = 1; j <= m; j++) { add(j + nt, des, col[j]); } sap(); n = T; memset(vis, 0, sizeof(vis)); for(int i = 1; i <= nt; i++) for(int j = 1; j <= m; j++) if(ans[i][j] != -1) vis[i][j] = 2; src = S; des = T; for(int i = 1; i <= nt; i++) { for(int j = 1; j <= m; j++) { if(vis[i][j] == 2) continue; int te = id[i][j]; int tmp = edge[te ^ 1].w; edge[te].w = edge[te ^ 1].w = 0; int flag = 1; int le = e; add(src, i, 1); int me = e; add(j + nt, des, 1); if( tmp && sap()) flag = 0, tmp--; edge[le].w = edge[le ^ 1].w = 0; edge[me].w = edge[me ^ 1].w = 0; le = e; add(src, j + nt, 1); me = e; add(i, des, 1); if(flag && 100 - tmp > 0&& sap()) flag = 0, tmp++; edge[le].w = edge[le ^ 1].w = 0; edge[me].w = edge[me ^ 1].w = 0; edge[te ^ 1].w = tmp; edge[te].w = 100 - tmp; vis[i][j] = flag; } } for(int i = 1; i <= nt; i++) for(int j = 1; j <= m; j++) { if(vis[i][j] == 2) { if(ans[i][j] != -1) printf("%d", ans[i][j]); } else { if(vis[i][j]) printf("%d", edge[id[i][j] ^ 1].w); else printf("-1"); } if(j < m) printf(" "); else printf(" "); } } return 0; }