ZOJ 3582 Back to the Past (概率dp)

OJ题目：click here~~

题目分析：有两排灯，每排n个，每个灯亮的概率为p，每个灯之间互不影响，亮了的灯不再灭，问两排中，每排有大于等于m个灯亮的概率。

设dp[ i ][ j ]为第一排亮了i个灯，第二排亮了j个灯，距离目标状态的期望天数。显然 i >= m ,j >= m时 ， dp[ i ][ j ] = 0 。

状态转移 :

第一排亮了a个灯，a 在[ 0 , n - i] 之间，第二排亮了b个灯 ， b 在[ 0 , n - j ] 之间。

则：

其中概率：

将dp[ i ][ j ]整理到等式的左边即可求出。

AC_CODE

四个for解决：

const int inf = 1<<30;
int n , m ;
double p;
double c[54][54] ;
double powp[52] , pow1_p[52] , dp[52][52];
int main()
{
    int i , j ,a , b;
    for(i = 0;i <= 52;i++)
    {
        c[i][0] = 1;
        c[i][i] = 1;
        for(j = 1;j < i;j++)
            c[i][j] = c[i - 1][j] + c[i - 1][j - 1];
    }
    while(scanf("%d%d%lf",&n,&m,&p))
    {
        if(n == 0 && m == 0 && p == 0) break;
        powp[0] = 1;
        for(i = 1;i <= 50;i++)
            powp[i] = powp[i - 1]*p;
        pow1_p[0] = 1;
        for(i = 1;i <= 50;i++)
            pow1_p[i] = pow1_p[i - 1]*(1 - p);
        memset(dp , 0 , sizeof(dp));
        for(i = n;i >= 0;i--)
            for(j = n;j >= 0;j--)
        {
            if(i >= m && j >= m) continue;
            double sum = 0;
            for(a = 0;a <= (n - i);a++)
                for(b = 0;b <= (n - j);b++)
            {
                if(a == 0 && b == 0) continue;
                sum += c[n-i][a]*powp[a]*pow1_p[n-i-a]*c[n-j][b]*powp[b]*pow1_p[n -j-b]*dp[i+a][j+b];
            }
            dp[i][j] = (sum + 1)/(1 - pow1_p[n-i]*pow1_p[n-j]);
        }
        printf("%.6lf\n",dp[0][0]);
    }
    return 0;
}

记忆化搜索解决：

int n , m ;
double p;
double c[54][54] ;
double powp[52] , pow1_p[52] , dp[52][52];

double dfs(int i , int j)
{
    if(i >= m && j >= m) return dp[i][j] = 0.0;
    if(dp[i][j] != -1) return dp[i][j];
    double sum = 0;
    int a , b;
    for(a = 0;a <= (n - i);a++)
        for(b = 0;b <= (n - j);b++)
    {
        if(a == 0 && b == 0) continue;
        sum += c[n-i][a]*powp[a]*pow1_p[n-i-a]*c[n-j][b]*powp[b]*pow1_p[n -j-b]*dfs(i+a , j+b);
    }
    dp[i][j] = (sum + 1)/(1 - pow1_p[n-i]*pow1_p[n-j]);
    return dp[i][j];
}

int main()
{
    int i , j;
    for(i = 0;i <= 52;i++)
    {
        c[i][0] = 1;
        c[i][i] = 1;
        for(j = 1;j < i;j++)
            c[i][j] = c[i - 1][j] + c[i - 1][j - 1];
    }
    while(scanf("%d%d%lf",&n,&m,&p))
    {
        if(n == 0 && m == 0 && p == 0) break;
        powp[0] = 1;
        for(i = 1;i <= 50;i++)
            powp[i] = powp[i - 1]*p;
        pow1_p[0] = 1;
        for(i = 1;i <= 50;i++)
            pow1_p[i] = pow1_p[i - 1]*(1 - p);
        for(i = 0;i <= n;i++)
            for(j = 0;j <= n;j++)
            dp[i][j] = -1; //这里不能用memset赋值
        printf("%.6lf\n",dfs(0 , 0));
    }
    return 0;
}