sdoi最优图像image(网络流)

来源:互联网 发布:令狐公子 知乎 编辑:程序博客网 时间:2024/06/18 16:40

题意:有一个N*M的01矩阵a,已知每一行有多少个1以及每一列有多少1。现在这个矩阵搞丢了,但是会告诉你a[i,j]是1的概率p[i,j](一个[0,100)的整数表示百分率)。让你还原出一个概率最大的符合条件的01矩阵,任意输出一个。

据说这题只能用zkw费用流过。。这个太奇怪了。

这题调了我一个下午,很有启发意义。

首先这是经典的矩阵还原模型,就是行做X部,列做Y部,之间的连边的流量代表a[i,j]的值,之前做过两道类似的题,因此这里还是想到了的。

然后就是最精华的地方了,概率应该是所有的p[i,j]的乘积,但是费用流的模型是和,我开始试着改一下费用流的写法,就是把中间找最短路的地方改成乘法,这样理论上可行但是乘不到几个数就会严重暴long long了。。然后我就上网搜了,我惊讶地发现竟然没几个人写这道题的题解,并且没人贴程序。这题有这么偏门吗。。找了半天才找到,要把费用设为概率的对数!这个真是太巧妙了,因为两个正数的大小关系完全等同于他们取对数的大小,然后两个数相乘就对应了他们的对数相加!!!然后要注意因为是最大费用,将他们费用全部取反,最后再取反回来即可。

然后我就无脑地写了一发,然后就交了,结果只过了前四个点,后面全T了。。然后我就各种调,我开始以为是连边有问题产生了负权环,然后下了组数据写了个SPFA判定,结果没负环。然后我就翻来覆去看了几遍费用流的模板,觉得没问题,,然后就抓狂了,在各种细节上试,,最后终于试出来了,就是update里面判是否是最短路径上的点那个dis的判定我直接用的等号,但事实上由于精度问题转移几次之后该相等的已经不相等了,于是我写了个eps和自定义了个等号。。好消息,没T了,但是都WA了,,,然后我又找了半天,觉得好像取对数的时候精度不好,于是把概率全部乘了很大一个数之后再取的对数,然后终于过了。以前openjudge上面有道题就是,要先乘一个数再做除法,不然精度要爆。

这题没看到别人写的代码,也完全不知道精度会有这种问题,全是自己摸索的,写出来还是很愉快。

这道题的启发:

1、比较乘积但是太大了,可以换成取对数之后的加法。

2、像什么取对数,开根号之类容易挂精度的,可以先乘一个很大的数。

3、系统的对数函数好像并不是O(1)的(或者是常数有点大)。系统自带的有log,log2,log10,分别是以e,2,10为底,其中以2位底的会比另外两个快三倍,如果只是用来比大小,最后用log2。

#include<cstdio>#include<cstring>#include<cassert>#include<cmath>#include<algorithm>using namespace std;#define DB double#define clr(a) memset(a,0,sizeof a)inline DB min(const DB&a, const DB&b){return a < b ? a : b; }#define rep(a,b,c) for (int a=b;a<=c;++a)const int inf = 0x3f3f3f3f;const int MAXN = 205;const int MAXM = 100000;int N, M;const DB eps = 1e-10;bool cmp(DB a, DB b) //自己定义在精度误差范围内的等号{if (a < b) swap(a, b);return a-b <= eps;}struct Ed {int to, cap;DB cost;Ed*nxt, *back;};struct FlowNet{Ed Edge[MAXM], *ecnt, *adj[MAXN];FlowNet () { ecnt=Edge; }DB dis[MAXN];bool vis[MAXN];int vn, S, T, flow;DB tot;inline void adde(int a, int b, int c, DB d){(++ecnt)->to = b;ecnt->cap = c;ecnt->cost = d;ecnt->nxt = adj[a];ecnt->back = ecnt+1;adj[a] = ecnt;(++ecnt)->to = a;ecnt->cap = 0;ecnt->cost = -d;ecnt->nxt = adj[b];ecnt->back = ecnt-1;adj[b] = ecnt;}void init(int n, int s, int t){tot = 0; flow = 0;vn = n; S = s; T = t;clr(dis), clr(adj);ecnt = Edge;}bool update(){DB tmp = 1e10;rep(i, 1, vn) if (vis[i])for (Ed *p = adj[i]; p; p=p->nxt)if (p->cap > 0 && !vis[p->to])tmp = min(tmp, dis[p->to]-dis[i] + p->cost);if (tmp == 1e10) return 0;for (int i = 1; i<=vn; ++i)if (vis[i]) dis[i] += tmp;return 1;}int aug(int u, int augco){if (u == T){tot += dis[S] * augco;flow += augco;return augco;}vis[u] = 1;int delta, augc = augco;for (Ed*p = adj[u]; p && augc; p=p->nxt){int&v = p->to;if (!vis[v] && p->cap && cmp(dis[u],dis[v]+p->cost)) //写等号要出问题{delta = min(p->cap, augc);delta = aug(v, delta);p->cap -= delta, p->back->cap += delta;augc -= delta;}}return augco - augc;}DB mcmf(){do {do clr(vis); while (aug(S, inf));} while (update());return tot;}void drawMap(int img[105][105]){rep(i, 1, N)for (Ed*p = adj[i]; p; p=p->nxt)if (p->to > i && p->to<=N+M)img[i][p->to - N] = p->back->cap;}} G;int img[105][105];int main(){scanf("%d%d", &N, &M);G.init(N+M+2, N+M+1, N+M+2);DB tmp;rep(i, 1, N) rep(j, 1, M){scanf("%lf", &tmp);G.adde(i, j+N, 1, -log2(tmp*1e6)); //调整精度}int xx;rep(i, 1, N){scanf("%d", &xx);G.adde(G.S, i, xx, 0);}rep(j, 1, M){scanf("%d", &xx);G.adde(N+j, G.T, xx, 0);}G.mcmf();G.drawMap(img);rep(i, 1, N){rep(j, 1, M) printf("%d", img[i][j]);puts("");}return 0;}




0 0