Codeforces Round #293 (Div. 2) D. Ilya and Escalator (概率DP)

dp[i][j]表示第i秒电梯进去的人数为j时的概率。由于概率比较好求，而且这里的样本是有限个。所以可以先求出概率，然后用公式转化成期望。

#include <iostream>
#include <string.h>
#include <math.h>
#include <queue>
#include <algorithm>
#include <stdlib.h>
#include <map>
#include <set>
#include <stdio.h>
using namespace std;
#define LL __int64
#define pi acos(-1.0)
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
const double eqs=1e-9;
double dp[2015][2015];
int main()
{
int n, t, i, j, k;
double p, ans;
while(scanf("%d%lf%d",&n,&p,&t)!=EOF){
memset(dp,0,sizeof(dp));
dp[0][0]=1;
for(i=1;i<=t;i++){
for(j=0;j<=n;j++){
dp[i][j]=dp[i-1][j-1]*p;
if(j!=n) dp[i][j]+=dp[i-1][j]*(1-p);
else dp[i][j]+=dp[i-1][j];
}
}
ans=0;
for(i=1;i<=min(n,t);i++){
ans+=i*dp[t][i];
}
printf("%.7f\n",ans);
}
return 0;
}