西北大学集训队选拔赛 B 饱和式救援(概率dp)

https://ac.nowcoder.com/acm/contest/892/B

n个选出至少k个的概率

 

#include<bits/stdc++.h>
using namespace std;
int n,m,k;
double dp[2005][2005];
double p[2005];
int l;
double r;
double sum=0;
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	cin>>n>>m>>k;
	for(int i=1;i<=2000;i++)
		p[i] = 1.0;
	while(n--){
		cin>>l>>r;
		p[l] = p[l]*(1-r);
	}
	dp[0][0] = 1;
	for(int i=1;i<=m;i++)
		dp[i][0] = dp[i-1][0]*p[i];
	for(int i=1;i<=m;i++){
		for(int j=1;j<=i;j++){
			if(i==j)
				dp[i][j] = dp[i-1][j-1]*(1.0-p[i]);
			else
				dp[i][j] = dp[i-1][j]*p[i] + dp[i-1][j-1]*(1.0-p[i]);
		}
	}
	for(int i=k;i<=m;i++)
		sum+=dp[m][i];
	printf("%.3lf\n",sum);
	return 0;
}