CF768D,概率DP

问题：http://codeforces.com/problemset/problem/768/D

 题意：琼恩打败白色的步行者需要k中能量水晶，但是这k种水晶并不是按照一定顺序出现的，而是随机出现，

并且每种水晶出现的概率相同，琼恩想知道让每种水晶都至少有一件的概率大于P/2000至少需要多少天，共有

q次查询

题解：因为每种水晶出现是按照相同的概率的，很容易想到需要DP一下，而且是概率DP。DP[i][j]表示i天获得j种水晶的概率，则DP[i][j] = DP[i-1][j] * j / k + DP[i-1][j - 1] * (k - j + 1) / k. 二维可用1维滚动数组优化。然后又因为dp[i][k]是递增的,且天数我们是从低到高的， (p-eps)/2000也是递增的，利用单调性可以边计算dp[i][j]边给满足dp[i][j] >= (p-eps)/2000的p数组赋值.

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
typedef long double ld;

#define x0 x0___
#define y0 y0___
#define pb push_back
#define SZ(X) ((int)X.size())
#define mp make_pair
#define fi first
#define se second
#define pii pair<int,int>
#define pll pair<ll,ll>
#define pli pair<ll,int>
#define pil pair<int,ll>
#define ALL(X) X.begin(),X.end()
#define RALL(X) X.rbegin(),X.rend()
#define rep(i,j,k) for(int i = j;i <= k;i ++)
#define per(i,j,k) for(int i = j;i >= k;i --)
#define mem(a,p) memset(a,p,sizeof(a))


const ll MOD = 1E9 + 7;
ll qmod(ll a,ll b,ll c) {ll res=1;a%=c; assert(b>=0); for(;b;b>>=1){if(b&1)res=res*a%c;a=a*a%c;}return res;}
ll gcd(ll a,ll b) { return b?gcd(b,a%b):a;}

template<typename T, typename S>
void upmax(T& a,S b){if(a<b) a=b;}
template<typename T, typename S>
void upmin(T& a,S b){if(a>b) a=b;}
template<typename T>
void W(T b){cout << b << endl;}
void gettle() {while(1);}
void getre() {int t=0;t/=t;}


/
/
/
/
const int N = 1024;

double dp[N];
int ans[N];
const double eps = 1e-7;

int main()
{
    int k, q;
    scanf("%d %d", &k, &q);
    dp[0] = 1.0; //第0天获得0个的概率为1
    int num = 1;
    for(int i = 1;num <= 1000;i ++) {
        for(int j = k;j >= 1;j --) dp[j] = ((k - j + 1.0) * dp[j - 1] + dp[j] * j) / k;
        while(num <= 1000 && dp[k] * 2000 >= (num - eps)) {
            ans[num++] = i; //天数递增,num也是,两个都是单调增的,所以可以利用单调性优化.
        }
        dp[0] = 0.0; //之后i天获得0个概率为0.
    }
    while(q --) {
        int d;  scanf("%d",&d);
        printf("%d\n",ans[d]);
    }
    return 0;
}