768D Jon and Orbs[概率dp]

D. Jon and Orbs
time limit per test
2 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output

Jon Snow is on the lookout for some orbs required to defeat the white walkers. There are k different types of orbs and he needs at least one of each. One orb spawns daily at the base of a Weirwood tree north of the wall. The probability of this orb being of any kind is equal. As the north of wall is full of dangers, he wants to know the minimum number of days he should wait before sending a ranger to collect the orbs such that the probability of him getting at least one of each kind of orb is at least , where ε < 10 - 7.

To better prepare himself, he wants to know the answer for q different values of pi. Since he is busy designing the battle strategy with Sam, he asks you for your help.

Input

First line consists of two space separated integers kq (1 ≤ k, q ≤ 1000) — number of different kinds of orbs and number of queries respectively.

Each of the next q lines contain a single integer pi (1 ≤ pi ≤ 1000) — i-th query.

Output

Output q lines. On i-th of them output single integer — answer for i-th query.

Examples
input
1 1
1
output
1
input
2 2
1
2
output
2
2


题意:有k种不同的水晶,每天等概率出现一个,q次询问至少多少天可以满足出现k种的概率大于pi / 2000

思路:

对于k种水晶,至少k天才能使所有水晶出现的概率不为0,对于当天出现的水晶,分为两种情况,第一种:这个水晶以前出现过了。 第二种:这个水晶以前没出现过。

则用dp[i][j]表示第i天出现了j种水晶的概率,则dp[i][j] = dp[i-1][j]*(j/k) + dp[i-1][j-1]*(j-k+1)/k;

式中第一项,表示第i-1天已经出现了j种水晶,那么这一天仍然出现j种水晶的概率就是再出现这j种中的一种,第二项则表示出现了新的一种。

则dp[i][k]则表示第i天出现了k种水晶的概率

由于需要的天数未定,但询问q小于1e3,k也小于1e3。所以可以先求出所有可能询问的结果,然后O(1)查询。

代码:

#include<iostream>
#include<algorithm>
using namespace std;
long long ans[1005];
double dp[10005][1005];
int k, q;
int main()
{
    cin >> k >> q;
    dp[0][0] = 1.0;
    int p = 1;
    for(int i = 1; p <= 1000; ++i)
    {
        for(int j = 1; j <= k; ++j)
            dp[i][j] = (dp[i-1][j]*j + dp[i-1][j-1]*(k-j+1))/k;
        while(p <= 1000 && dp[i][k]*2000 > p)
            ans[p++] = i;
    }
    for(int i = 0; i < q; ++i)
    {
        int n;
        cin >> n;
        cout << ans[n] << endl;
    }

}


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值