Robot(简单概率dp)

Sample Input

 

3 1 1 2
1
5 2 4 4
1
2
0 0 0 0

Sample Output

0.5000
0.2500

Robot

题意:

一个环形的数字圈,起始点为1,每次走x步

随机向前向后,概率各位0.5,询问y点的概率

解析:

简单的一道概率dp题

概率dp一般从前往后推

这题数据非常大,要用滚动数组优化

注意顺序,最优子结构,处理好边界

ac:

#include<bits/stdc++.h>
#define ll long long
#define MAXN 505
using namespace std;
double dp[2][MAXN]={0};

int main()
{
    int n,m,l,r,x;
    while(scanf("%d%d%d%d",&n,&m,&l,&r)!=EOF)
    {
        if(n==0&&m==0&&l==0&&r==0)
            break;
        for(int i=0;i<n;i++)
            dp[0][i]=0;
        dp[0][0]=1.00;
        int now=0;
        while(m--)
        {
            scanf("%d",&x);
            now=now^1;//滚动数组优化
            for(int i=0;i<n;i++)//滚动数组,重新计算,要清零
                dp[now][i]=0;
            for(int i=0;i<n;i++)
            {
                if(dp[now^1][i]==0)
                    continue;
                dp[now][(i+x)%n]+=0.5*dp[now^1][i];
                dp[now][(i+n-x)%n]+=0.5*dp[now^1][i];
            }
        }
        double ans=0;
        for(int i=l-1;i<r;i++)
            ans+=dp[now][i];
        printf("%.4f\n",ans);
    }
    return 0;
}