CodeForces 518D---概率dp

题意：

有 n 个人上电梯，每个人是否上电梯的概率是 p ,每个人不管是否上电梯都需要花费一秒，问 t 秒内，电梯上有多少人？（上了电梯

的人要一直在电梯上呆着）。

分析：

因为上了电梯的人要一直在电梯上呆着，因此用 dp[i][j] 表示第 i 秒第 j 个人上电梯的概率，初始dp[0][0]=1,当 n 个人全部上电梯以后，

此时时间已经不能影响概率：dp[i][j]=dp[i-1][j];当没全部上完电梯时，第 i 秒第 j 个人上电梯的概率等于上一秒他不上电梯的概率：

dp[i][j]=(1-p)*dp[i-1][j];所以第 i 秒第 j 个人上电梯的概率dp[i][j]+=p*dp[i-1][j-1],他上电梯的概率加上他之前上电梯的人的概率。

代码：

#include <bits/stdc++.h>
using namespace std;
//题意：上了电梯就不能下来
const int maxn=2000+10;
double dp[maxn][maxn];  ///dp[i][j]表示第i秒第j个人上电梯的概率
int main()
{
    int n,t;
    double p;
    cin >>n>>p>>t;
    dp[0][0]=1;
    for(int i=1;i<=t;i++)
    {
        for(int j=0;j<=n;j++)
        {
            if(j==n)
                dp[i][j]=dp[i-1][j]; //n个人上完了，概率不随时间改变
            else
                dp[i][j]=(1-p)*dp[i-1][j]; //后一秒上的概率等于前一秒他没上的概率
            if(j)
                dp[i][j]+=p*dp[i-1][j-1]; //第i秒第j个人上电梯的概率等于他上电梯的概率加上他之前的人上电梯的概率
        }
    }
    double ans=0;
    for(int i=1;i<=n;i++)
    {
        ans+=dp[t][i]*i; //第i个人上电梯的概率*i人
    }
    printf("%.7g\n",ans);
    return 0;
}