HDU4576 Robot(概率dp)

HDU4576 Robot

題意

1到n形成一個環，機器人從一號點出發，給定每次走的長度，方向隨機，問最後在某個區域內的概率是多少。

思路

入門級概率dp

把if(dp[id][i])這段去掉就超時了,還是得剪枝一下

代碼

#include<bits/stdc++.h>
using namespace std;
const int maxn=210;
double dp[2][maxn];
int main(){
    int n,m,l,r;
    while(~scanf("%d%d%d%d",&n,&m,&l,&r)){
        if(n==0&&m==0&&l==0&&r==0) break;
        for(int i=0;i<=n;i++)
            dp[0][i]=dp[1][i]=0;
        dp[0][0]=1;
        int id=0;
        for(int i=1;i<=m;i++){
            int x;scanf("%d",&x);
            // for(int i=0;i<n;i++) {
            //     printf("%lf ",dp[id][i]);
            // }puts("");
            for(int i=0;i<n;i++){
                dp[id^1][i]=0;
            }
            for(int i=0;i<n;i++){
                if(dp[id][i]){
                    dp[id^1][(i+x)%n]+=0.5*dp[id][i];
                    dp[id^1][(i-x+n)%n]+=dp[id][i]*0.5;
                }
            }
            id^=1;
            // for(int i=0;i<n;i++) {
            //     printf("%lf ",dp[id][i]);
            // }puts("");
        }
        double ans=0;
        for(int i=l-1;i<r;i++){
            ans+=dp[id][i];
        }
        printf("%.4lf\n",ans);
    }
}