bzoj3029 守卫者的挑战

题目：

题目描述
打开了黑魔法师 Vani 的大门，队员们在迷宫般的路上漫无目的地搜寻着关押 applepi 的
监狱的所在地。突然，眼前一道亮光闪过。“我， Nizem，是黑魔法圣殿的守卫者。如果你
能通过我的挑战，那么你可以带走黑魔法圣殿的地图„„”瞬间，队员们被传送到了一个擂
台上，最初身边有一个容量为 K 的包包。
擂台赛一共有 N 项挑战，各项挑战依次进行。第i 项挑战有一个属性 ai ，如果ai   ，
表示这次挑战成功后可以再获得一个容量为 ai 的包包；如果 ai  ，则表示这次挑战成功
后可以得到一个大小为 1 的地图残片。地图残片必须装在包包里才能带出擂台，包包没有
必要全部装满，但是队员们必须把获得的所有的地图残片都带走（没有得到的不用考虑，只
需要完成所有 N 项挑战后背包容量足够容纳地图残片即可），才能拼出完整的地图。并且他
们至少要挑战成功 L 次才能离开擂台。
队员们一筹莫展之时，善良的守卫者 Nizem 帮忙预估出了每项挑战成功的概率，其中
第 i 项挑战成功的概率为

pi 。现在，请你帮忙预测一下，队员们能够带上他们获得的地图
残片离开擂台的概率。
输入格式
第一行三个整数 N , L , K 。
第二行 N 个实数，第i 个实数 pi 表示第i 项挑战成功的百分比。
第三行 N 个整数，第i 个整数 ai 表示第i 项挑战的属性值.
输出格式
一个整数，表示所求概率，四舍五入保留 6 位小数。
样例输入 1
3 1 0
10 20 30
-1 -1 2
样例输出 1
0.300000样例输入 2
5 1 2
36 44 13 83 63
-1 2 -1 2 1
样例输出 2
0.980387
样例说明
在第一个样例中， 若第三项挑战成功，如果前两场中某场胜利，队员们就有空间来容纳
得到的地图残片，如果挑战失败，根本就没有获得地图残片，不用考虑是否能装下；若第三
项挑战失败，如果前两场有胜利，没有包来装地图残片，如果前两场都失败，不满足至少挑
战成功 L 次（L 1）的要求。因此所求概率就是第三场挑战获胜的概率。
数据范围与约定
对于 100% 的数据，保证  K  ，   N  ，  ai  ，   L  N ，
  
pi 。

思路：
dp。
令f[i][j][k]表示前i个挑战，成功了j项，背包剩余k容量的概率。
其中，如果k大于n，则令k等于n。
且k可以为负，处理方法是把k加上n。
转移方程：

f[i+1][j+1][k+w[i+1]<2*n?k+w[i+1]:2*n]+=f[i][j][k]*P[i+1]/100.0;
f[i+1][j][k]+=f[i][j][k]*(100-P[i+1])/100.0;

代码：

#include<bits/stdc++.h>
using namespace std;

#define read(x) scanf("%d",&x)
#define maxn 200
#define maxK 2000
#define db double

int n,m,K;
int P[maxn+5],w[maxn+5];

db f[maxn+5][maxn+5][2*maxn+5];

int main() {
	freopen("guard.in","r",stdin);
	freopen("guard.out","w",stdout);
	read(n),read(m),read(K);
	for(int i=1;i<=n;i++) read(P[i]);
	for(int i=1;i<=n;i++) read(w[i]);
	
	f[0][0][K+n<2*n?K+n:2*n]=1;
	
	for(int i=0;i<n;i++) {
		for(int j=0;j<=i;j++) {
			for(int k=0;k<=2*n;k++) {
				if(w[i+1]==-1) {
					f[i+1][j+1][k-1]+=f[i][j][k]*P[i+1]/100.0;
					f[i+1][j][k]+=f[i][j][k]*(100-P[i+1])/100.0;
				} else {
					f[i+1][j+1][k+w[i+1]<2*n?k+w[i+1]:2*n]+=f[i][j][k]*P[i+1]/100.0;
					f[i+1][j][k]+=f[i][j][k]*(100-P[i+1])/100.0;
				}
			}
		}
	}
	
	db ans=0;
	for(int i=m;i<=n;i++) for(int j=n;j<=2*n;j++) ans+=f[n][i][j];
	printf("%.6lf",ans); 
	
	return 0;
}