hdu 5013 优化疑问+dp

来源:互联网 发布:银川市大数据管理局 编辑:程序博客网 时间:2024/06/05 14:22

http://acm.hdu.edu.cn/showproblem.php?pid=5013

m个游客,n座城市(m, n <= 16), 每个人从1走到n, 每次有一定概率停在原地,然后以后就不前进了。一个人到过一个城会得到一定的愉悦度,对于相邻的两座城,会额外产生Cj / Cj - 1 * Hj的愉悦度,Cj是到过j城的人数,Hj是到过j城的人在这里获得的愉悦度之和。求期望的总愉悦度。

根据题解给出的解法

http://blog.csdn.net/oilover/article/details/39526899

需要跑3s左右

优化成纯dp能变成78ms,但是看不懂别人的代码...

根据期望的线性性,分别求每个人每天的值的期望

f[i][j][x][y]表示第i天,前j个人,前一天x,当前天y的概率

g[i][j][x][y]表示第i天,前j个人,前一天x,当前天y的期望

那么

f[i][j][x][y]=f[i][j-1][x][y]*(1-p[j]^(i-2))+f[i][j-1][x-1][y]*p[j]^(i-2)*(1-p[j])+f[i][j-1][x-1][y-1]*p[j]^(i-1)


g[i][j][x][y]=g[i][j-1][x][y]*(1-p[j]^(i-2))+g[i][j-1][x-1][y]*p[j]^(i-2)*(1-p[j])+(g[i][j-1][x-1][y-1]+f[i][j-1][x-1][y-1]*h[j][i])*p[j]^(i-1)

状态方程不懂,求各位大神解释..

#include <cstdio>#include <cstdlib>#include <cmath>#include <cstring>#include <string>#include <queue>#include <stack>#include <iostream>#include <algorithm>using namespace std;#define RD(x) scanf("%d",&x)#define RD2(x,y) scanf("%d%d",&x,&y)#define RD3(x,y,z) scanf("%d%d%d",&x,&y,&z)#define clr0(x) memset(x,0,sizeof(x))typedef long long LL;const int maxn = 20;int n, m;double p[maxn][maxn],h[maxn][maxn],f[maxn][maxn][maxn],g[maxn][maxn][maxn];int main() {    while (~RD2(m,n)) {        for (int i = 1; i <= m; i++) {            scanf("%lf", &p[i][1]);            p[i][0] = 1.0;            for (int j = 2; j <= n; j++) {                p[i][j] = p[i][j - 1] * p[i][1];            }        }        for (int i = 1; i <= m; i++) {            for (int j = 1; j <= n; j++) {                scanf("%lf", &h[i][j]);            }        }        clr0(f),clr0(g);        double ans = 0;        for (int i = 2; i <= n; i++) {            f[i][0][0] = 1;            for (int j = 1; j <= m; j++) {                for (int x = m; x >= 0; x--) {                    for (int y = x; y >= 0; y--) {                        f[i][x][y] *= (1.0 - p[j][i - 2]);                        if (x > 0) {                            f[i][x][y] += f[i][x - 1][y] * p[j][i - 2] * (1.0 - p[j][1]);                            if (y > 0) {                                f[i][x][y] += f[i][x - 1][y - 1] * p[j][i - 1];                            }                        }                        g[i][x][y] *= (1.0 - p[j][i - 2]);                        if (x > 0) {                            g[i][x][y] += g[i][x - 1][y] * p[j][i - 2] * (1.0 - p[j][1]);                            if (y > 0) {                                g[i][x][y] += (g[i][x - 1][y - 1] + f[i][x - 1][y - 1] * h[j][i]) * p[j][i - 1];                            }                        }                    }                }            }            for (int x = 1; x <= m; x++) {                for (int y = 1; y <= x; y++) {                    ans += g[i][x][y] * ((double)y / x + 1.0);                }            }        }        for (int i = 1; i <= m; i++) {            ans += h[i][1];        }        printf("%.10f\n", ans);    }    return 0;}


0 0
原创粉丝点击