CF464D World of Darkraft - 2

Roma 在游戏“World of Darkraft”(理论上应该是 World of darkcraft,MineCraft 的一个版本)找到一个新角色。 \(\mathrm{Roma}\) 有 k种装备,一开始每种装备各有 1个,且每种装备的初始等级均为 1。
游戏中可以靠打怪来获取新装备,总共有 n 只怪兽,每打赢 1 只怪兽后,\(\mathrm{Roma}\) 会随机获得一种装备 \(a\) (a∈[1,k]),假设原有的 a 装备的等级为 t,那么新获得的装备的等级为 [1,t+1],\(\mathrm{Roma}\) 会将新获得的装备和原来的装备中等级较高的装备留下,等级较低的装备卖出,卖出可获得的金币为该装备的等级。 问打完这 n只怪兽后,\(\mathrm{Roma}\) 获得的金币的期望。 \(1<=n<=10^5,1<=k<=10^3\)
Sample Input

input1

1 3

input2

2 1

input3

10 2

Sample Output

output1

1.0000000000

output2

2.3333333333

output3

15.9380768924

这道题我们考虑概率DP

设dp[i][j]表示打赢只要求出了前i只怪兽后某种装备被替换后价值为j的期望。

为什么是某装备?

因为所有装备的期望值是相同的,所以我们只要求出一种装备的期望,再乘上k即可。

有两种情况:

1.爆出了等级为j+1的神装。(概率为\(1/(j+1)\)

dp[i][j]+=(dp[i-1][j-1]+j)/j+1。

2.只得到等级<=j的装备。(概率为\(j/(j+1)\)

dp[i][[j]+=(dp[i-1][j]+(j+1)/2)*j/(j+1);//因为1至j是等概率出现的,所以取平均(j+1)/2。

但是这样你打出来你会发现自己MLE或TLE了,为什么?

i过于大,j也过于大。

优化:1.dp[i][j]的更新只需要他前一个的情况,所以我们可以考虑用滚动数组记录。

2.因为这个状态的转移是会收敛的,所以j越大他所产生的变化就越小,最终小到可以忽略不计,所以j只用取到很小就可以当作答案用了(有很多神仙j取600就过了,不过保险起见最好取1000)。

#include<bits/stdc++.h>
using namespace std;
int n,k,s=1;
double dp[2][1010];
int main()
{
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++,s^=1)
	{
		for(int j=min(n,1000);j;j--)
		{
			dp[s][j]=((dp[s^1][j+1]+j)/(j+1.0)+(dp[s^1][j]+(j+1)/2.0)*j/(j+1.0))/double(k)+dp[s^1][j]*(k-1.0)/double(k);
		}
	}
	printf("%.11lf\n",dp[s^1][1]*k);
	return 0;
}
posted @ 2019-08-17 16:52  ezoi_ly  阅读(286)  评论(0编辑  收藏  举报