题意:一共有c种颜色的巧克力,取出n块,当有两块颜色相同就把这两块给恰了,问最后剩下m块的概率
我们来想一想当取出奇数块的时候可能剩下偶数块吗?取出偶数块剩下奇数块呢?答案是都不可能。假设取出奇数块,剩下偶数块,那被恰掉的块数是奇数,取出偶数块剩下奇数块同理。而题目要求恰的话必须恰两块颜色相同的,所以被恰的块数是偶数,相矛盾。
我们用dp[i][j]表示取i块剩下j块的概率,那么dp[i][j]可以由dp[i-1][j-1]和dp[i-1][j+1]得到.
dp[i-1][j-1]->dp[i][j]即新取出的巧克力和现有的巧克力颜色都不同,而发生的概率P=(c-j+1)/c,即在取第i次之前,还没取出的巧克力颜色总数(c-j+1)除以全部巧克力颜色总种数c,则P(dp[i-1][j-1]->dp[i][j])=dp[i-1][j-1]*(c-j+1)/c
dp[I-1][j+1]->dp[i][j]即取出与现有的颜色,发生的概率为现有的巧克力颜色种数/巧克力颜色总种数,即(j+1)/c,可得P(dp[I-1][j+1]->dp[i][j])=dp[i-1][j+1]*(j+1)/c.
综上,dp[i][j=P(dp[i-1][j-1]->dp[i][j])+P(dp[I-1][j+1]->dp[i][j])=dp[i-1][j-1](c-j+1)/c+dp[i-1][j+1](j+1)/c.
同时,要注意空间和时间要求。空间虽然可以用滚动数组来解决,但无法解决时间上的问题,本蒟蒻不会qaq,然后看别人的题解看到的:大佬链接
AC代码
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int N = 1010;
double dp[N][110];
const int mod = 1000;
int main() {
int c,n,m;
while (scanf("%d", &c)) {
if (c == 0) break;
scanf("%d %d", &n, &m);
if (m > n || m > c || (m + n) & 1) {
printf("0.000\n");
continue;
}
if (n >= 1000) {
n = 1000 + n % 2;
}
memset(dp, 0, sizeof(dp));
dp[0][0] = 1;
for (int i = 1; i <=n; i++) {
for (int j = 0; j <= i && j <= c; j++) {
if ((i + j) & 1) {
dp[i][j] = 0;
}
else {
dp[i][j] = dp[(i - 1)][j - 1] * (c - j + 1.0) / c + dp[(i - 1)][j + 1] * (j + 1.0) / c;
}
}
}
printf("%.3lf\n", dp[n][m]);
}
}