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

题目大意

有$n$个人，排成一列，现在他们将要进入自动扶梯。
已知在$1$秒内，排在对头的人有$p$的概率进入自动扶梯，有$1-p$的概率保持不动。
$n$个人必须依次进入自动扶梯，进入了自动扶梯的人不会再次离开扶梯。
问在$t$秒后，自动扶梯里人数的期望是多少。

时间限制

2s

数据范围

$n,t\le 2000$

题解

一个非常好想到的dp，
设$f_{i,j}$表示在第$i$秒，自动扶梯里有$j$个人的概率。
转移显然
$$f_{i,j}=fi-1,j\times (1-p)+f_{i-1,j-1}\times p$$
值得注意的是，$f_{i,n}=fi-1,n+f_{i-1,n-1}\times p$，因为当$n$个人已经全部进入自动扶梯之后，整个过程就结束了，不再需要接着计算概率。

Code

//#pragma GCC optimize (2)
//#pragma G++ optimize (2)
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <iostream>
#include <vector>
#include <queue>
#define G getchar
#define ll long long
using namespace std;
 
ll read()
{
    char ch;
    for(ch = G();(ch < '0' || ch > '9') && ch != '-';ch = G());
    ll n = 0 , w;
    if (ch == '-')
    {
        w = -1;
        ch = G();
    } else w = 1;
    for(;'0' <= ch && ch <= '9';ch = G())n = (n<<1)+(n<<3)+ch-48;
    return n * w;
}

const int N = 2003;
const int mo = 998244353;

int n , k;
double p , p0 , f[N][N] , ans;

int main()
{
    //freopen("b.in","r",stdin);
    //freopen("2.txt","w",stdout);

    n = read();
    scanf("%lf", &p);
    p0 = 1 - p;
    k = read();
    f[0][0] = 1;
    for (int i = 1 ; i <= k ; i++)
    {
        f[i][0] = f[i - 1][0] * p0;
        for (int j = 1 ; j < n ; j++)
            f[i][j] = f[i - 1][j] * p0 + f[i - 1][j - 1] * p;
        f[i][n] = f[i - 1][n] + f[i - 1][n - 1] * p;
    }

    for (int i = 1 ; i <= n ; i++)
        ans = ans + f[k][i] * i;
    printf("%.8lf\n", ans);
}