A Dangerous Maze (II) （概率DP）

题目链接：https://cn.vjudge.net/contest/208908#problem/P

题目大意：和http://blog.csdn.net/haut_ykc/article/details/79143158这道题大致规则一样，只不过多加一个条件，就是主人公记得之前选错的k扇门是那几个，因此他接下来是不会再选择之前的k扇门（但是别忘了）在选择第k+1扇错门后，最前边呢个门就已经忘记是哪个了。

题目解答：我们令dp[i]表示记得之前i扇门是那几扇当前能逃出去的时间期望，然后递推即可。

#include<stdio.h>
#include<algorithm>
using namespace std;
double dp[105];
int main(void)
{
	int T,i,n,k,x,cases=0;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d%d",&n,&k);
		double sm1=0,sm2=0;
		int a=0,b=0;
		for(i=1;i<=n;i++)
		{
			scanf("%d",&x);
			if(x>0)
				sm1+=x,a++;
			else
				sm2-=x;
		}
		printf("Case %d: ",++cases);
		if(a==0)
		{
			printf("-1\n");
			continue;
		}
		b=n-a;k=min(k,b);
		if(a) sm1/=a;
		if(b) sm2/=b;
		dp[k]=sm1+sm2*(b-k)/a;
		for(i=k-1;i>=0;i--)
		{
			dp[i]=(sm2+dp[i+1])*(b-i);
			dp[i]+=sm1*a;
			dp[i]/=(n-i);
		}
		printf("%.7f\n",dp[0]);
	}
	return 0;
}