饱和式救援-概率dp

题目链接：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;
}