题意:包里有无穷多个巧克力,巧克力有c种颜色,每次从包里拿出不同颜色巧克力的概率都是相等的,桌面的巧克力不允许颜色相同,若某次拿出的巧克力与桌上的巧克力颜色相同了,则将两颗巧克力都吃掉。计算进行n次拿巧克力的操作后,桌上有m颗巧克力的概率,输出结果精确到小数点后三位。
用dp[ i ][ j ] 表示进行 i 次操作后桌面上有 j 颗巧克力的概率,状态转移方程:dp[ i ][ j ] += dp[ i - 1 ][ j - 1 ] * ( ( c - j + 1.0 ) / c ) + dp[ i - 1 ][ j + 1 ] *( ( j + 1.0 ) / c );
方程的前半部分表示新拿出的巧克力与桌面没有相同颜色, 后半部分表示有相同颜色并吃掉了相同颜色的两颗。
由方程可以看出,第 i 行仅与 i - 1行有关,所以可用滚动数组。
由于dp[ 1 ][ 0 ] = 0 (客观事实), 再根据状态转移方程知:当i + j为奇数时,dp[ i ][ j ] == 0,因为不可能拿奇数次让桌上有偶数颗,也不可能拿偶数次让桌上有奇数颗。
以前不知道有这样的处理方式,数据很大,根据输出为近似值直接压缩了数据,只要满足输出即可。
#include <iostream>
using namespace std;
double dp[2][102];
int main()
{
int c, n, m, i, j;
while(scanf("%d", &c) && c)
{
scanf("%d %d", &n, &m);
if(m > n || m > c || (m + n) % 2){
printf("0.000\n");
continue;
}
if(n > 1001)
n = n % 2 ? 1001 : 1000;
memset(dp, 0, sizeof(dp));
dp[0][0] = 1.0;
for(i = 1; i <= n; i++){
for(j = 0; j <= i && j <= c; j++){
dp[i%2][j] = 0.0;
if((i + j) % 2) continue;
if(j > 0)
dp[i%2][j] += dp[1-i%2][j-1] * ((c-j+1.0)*1.0/c);
if(j+1 <= i-1)
dp[i%2][j] += dp[1-i%2][j+1] * ((j+1.0)*1.0/c);
}
}
printf("%.3lf\n", dp[n%2][m]);
}
return 0;
}