UPC:2225 The number of steps(概率DP)

題意：按題目要求問走出迷宮的期望。

思路：比較明顯的概率DP。

#include <iostream>
#include <vector>
#include <cstring>
#include <cstdio>
using namespace std;
double f[50][50];
bool vis[50][50];
int n;
double a,b,c,d,e;
double dp(int i,int j)
{
    if(vis[i][j]) return f[i][j];
    vis[i][j]=true;
    double &ans=f[i][j];
    ans=0;
    if(j-1>=0&&i+1>n) ans=dp(i,j-1)+1;
    else if(j-1<0&&i+1<=n) ans=a*dp(i+1,j)+b*dp(i+1,j+1)+1;
    else if(j-1>=0&&i+1<=n) ans=e*dp(i,j-1)+c*dp(i+1,j)+d*dp(i+1,j+1)+1;
    return ans;
}
int main()
{
    while(scanf("%d",&n)!=EOF&&n)
    {
        scanf("%lf%lf%lf%lf%lf",&a,&b,&c,&d,&e);
        memset(vis,0,sizeof(vis));
        f[n][0]=0;
        printf("%.2lf\n",dp(1,0));
    }
    return 0;
}