TJU-3977-Probability I（概率DP）

首先是关于概率DP的入门文章：

http://www.cnblogs.com/kuangbin/archive/2012/10/02/2710606.html

设i点原地停留、前进一步、前进两步的概率为p[i][0],p[i][1],p[i][2]

关于是否可以到达n点的判断，开一个bool数组vis，首先全部置0，vis[1]=1,扫描一次，如果p[i][1]不为0则vis[i+1]=1,p[i][2]!=0则vis[i+2]=1，最后看vis[n]即可。

一般概率DP求概率是正推，求期望是逆推。

设f[i]为从第i个格子走到终点的期望值，则有下面的方程:

f[i]=(f[i]+1)*p[i][0]+(f[i+1]+1)*p[i][1]+(f[i+2]+1)*p[i][2]

初始条件为f[n]=0,答案为f[1]

#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cstring>
using namespace std;
const int N=100;

double p[N+5][3];
double f[N+5];
bool   vis[N+5];

void work(int n)
{
    int i;

    for(i=1;i<n;i++)
        scanf("%lf%lf%lf",&p[i][0],&p[i][1],&p[i][2]);
    
    memset(f,0,sizeof(f));
    memset(vis,0,sizeof(vis));
    
    //判断是否可达
    vis[1]=1;
    for(i=1;i<n;i++)
    {
        if(vis[i])
        {
            if(p[i][1]>0) vis[i+1]=1;
            if(p[i][2]>0) vis[i+2]=1;
        }
    }
    if(!vis[n])
    {
        printf("impossible\n");
        return ;
    }

    for(i=n-1;i>=1;i--)
    {
        f[i]=p[i][0]+(f[i+1]+1)*p[i][1]+(f[i+2]+1)*p[i][2];
        //f[i]=f[i]/(1-p[i][0]);
        if(fabs(p[i][0]-1)>1e-6) f[i]=f[i]/(1-p[i][0]); 
        	else f[i]=1e9;
        /*如果出现某点原地踏步的概率为1，那么这点到终点的期望为无穷，
        为避免发生/0的运算错误,要进行一下处理
        */
    }
    
    printf("%0.2lf\n",f[1]);

}

int main()
{
    int n;
    while(~scanf("%d",&n))
        work(n);
    return 0;
}