Chocolate(DP)

题意:一共有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]);
	}
	
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值