Codeforces768 D. Jon and Orbs（概率dp）

题意：

有n种物品，每种无限个，每天你可以随机等概率取出一个物品，可能抽到已有种类的物品。
q组询问，每组询问给出一个p，要求找到一个最小的x，
满足在第x天抽到n种物品的概率>=p/2000。

数据范围：n,q<=1e3，p<=1e3

解法：

令d[i][j]表示i天选出j种的概率.
O(1e4*n)预处理出前1e4天的,
为什么是前1e4天呢,因为p<=1e3,那么p/2000<=0.5,
而1e4足够大,对于所有p都能计算出答案,d[1e4][1e3]已经大于0.9了.

对于每组询问,从1开始枚举x,找到最小的满足d[x][n]>p/2000的x,输出即可.

code：

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxm=2e6+5;
double d[11111][1111];
int n,q;
void solve(){
    cin>>n>>q;
    d[0][0]=1;
    for(int i=0;i<1e4;i++){
        for(int j=0;j<=n;j++){
            if(d[i][j]>0){
                d[i+1][j]+=d[i][j]*j/n;
                if(j+1<=n)d[i+1][j+1]+=d[i][j]*(n-j)/n;
            }
        }
    }
    while(q--){
        int p;cin>>p;
        double t=p*1.0/2000;
        for(int i=1;i<=10000;i++){
            if(d[i][n]>=t){
                cout<<i<<endl;
                break;
            }
        }
    }
}
signed main(){
    ios::sync_with_stdio(0);
    solve();
    return 0;
}

---