题目链接:https://ac.nowcoder.com/acm/problem/25546
Problem
n个救援队,第i个救援队有pi的概率重启指定的行星发动机。m个被摧毁的发动机,问至少成功重启k个发动机的概率。
Solution
先求出每个发动机能够重新启动的概率,因为不同救援队指定重启的发动机可能相同,而这些救援队是相互独立的事件,,
表示A发生或B发生或AB都发生的概率。这样求比较麻烦,正难则反,求出都不能发生的概率P,然后1-P就是可以发生的概率。
求出概率后,找状态:dp[i][j] 表示前面i个发动机重启j个的概率。
状态转移: ,第i个不重启,那前i-1重启j个,第i个重启,那前i个重启j-1个,算上第i个就是j个了。
赋初值:容易想到dp[0][0] = 1,而dp[i][0] = dp[i-1][0]*(1-a[i]),是乘上第i个不能重启的概率豁。
Code
#include <bits/stdc++.h>
#define ll long long
#define pir pair<int,int>
#define debug(x) cout << #x << ":" << x << '\n'
const int N = 5e5+7;
const int mod = 1e9+7;
const ll ds = 1e15;
const double eps = 1e-8;
using namespace std;
int n,m,k,x;
double dp[3005][3005],a[N],p;
void solve(){
scanf("%d%d%d",&n,&m,&k);
for(int i = 1; i <= m; i++){
a[i] = 1;
}
for(int i = 1; i <= n; i++){
scanf("%d%lf",&x,&p);
a[x] = a[x]*(1-p);
}
for(int i = 1; i <= m; i++){
a[i] = 1-a[i];
}
dp[0][0] = 1.0;
for(int i = 1; i <= m; i++){
dp[i][0] = dp[i-1][0]*(1-a[i]);
for(int j = 1; j <= i; j++){
if(j == i) dp[i][j] = dp[i-1][j-1]*a[i];
else dp[i][j] = (dp[i-1][j]*(1-a[i])+dp[i-1][j-1]*a[i]);
}
}
double ans = 0.0;
for(int i = k; i <= m; i++){
ans += dp[m][i];
}
printf("%.3f\n",ans);
}
int main(){
// int t;
// cin >> t;
// while(t--)
solve();
system("pause");
return 0;
}