原题链接:https://leetcode-cn.com/problems/new-21-game/
1、递归
double new21Game(int N, int K, int W) {
if (K<=0){
return (N>=0)?1:0;
}else{
double s=0;
for (int i=1;i<=W;i++){
s+=new21Game(N-i,K-i,W);
}
return s/W;
}
}
2、动态规划
动态规划必须从后往前,结果是dp[0],因为此类题目的最终状态是已知的
dp[i]表示拥有i个分数的概率
落在K~N之间的点的概率是1,初始化dp,状态转移为
dp[i]=(dp[i+1]+dp[i+2]...+dp[i+W])/W;
因为每回合的概率依赖下一回合抽的数,听起来有点怪,但是概率计算是这样的,这也是这题的难点。知道这个之后,写法就因人而异了,高效一点的做法就是如下面代码中的写法,可以在常数时间,而不用每次都计算概率和。
double new21Game(int N, int K, int W) {
vector<double> dp(N+W+1,0);
for(int i=K;i<=N;i++){
dp[i]=1.0;//不小于K就停止抽取了,K~N直接的数必然符合条件
}
double sum=min(N-K+1,W);
for(int k=K-1;k>=0;k--){
dp[k]=sum/W;
sum+=dp[k]-dp[k+W];
}
return dp[0];
}