HDU 4576(Robot)

来源:互联网 发布:黑客网络分析软件 编辑:程序博客网 时间:2024/04/28 04:00

题意:杭州邀请赛的一道题;有标记着1~n的环,初始时有个机器人在1的位置,有m个操作,每个操作是使机器人随机的顺时针或逆时针的走w步,求所有操作过后机器人停在区间[l,r]内的概率;

思路:挺明显的概率dp,首先可以很直观的定义出状态dp[i][j],表示第j次操作后停留在第i位置的概率,转移方程是

dp[(i + w) % n][j] = dp[i[j-1] * 0.5;

dp[i-w][j]=dp[i][j-1]*0.5;其中i-w不断的加n使它大于等于0;

但是题目操作数比较大,这样会超内存,观察发现每次转移只和上一次操作有关,所以可以用个滚动数组来节省空间;优化空间后转移方程是

dp[(i + w) % n][j&1] += dp[i][!(j&1)] * 0.5;
dp[x][j&1] += dp[i][!(j&1)] * 0.5;x=i-w,x也要叠加到非负;

最后答案就是所有操作完后,位置l到r的概率总和。

#include <cstdio>#include <iostream>#include <algorithm>using namespace std;const int N = 220;const double eps = 0.0;double dp[N][2];int main(){    int n, m, l, r;        while(cin >> n >> m >> l >> r){        if(n == 0 && m == 0 && l == 0 && r == 0)            break;        int w, i, j;        for(i = 0;i < n;i++)            dp[i][0] = dp[i][1] = eps;        dp[0][0] = 1.0;        for(j = 1;j <= m;j++){            scanf("%d", &w);            for(i = 0;i < n;i++){                dp[(i + w) % n][j&1] += dp[i][!(j&1)] * 0.5;                int x = i - w;                while(x < 0)                x += n;                 dp[x][j&1] += dp[i][!(j&1)] * 0.5;                }            for(i = 0;i < n;i++)                dp[i][!(j&1)] = eps;        }        double res = 0.0;        l--, r--;        for(i = l;i <= r;i++)            res += dp[i][m&1];        printf("%.4lf\n", res);    }        return 0;}




原创粉丝点击