Divide by Zero 2017 and Codeforces Round #399 (Div. 1 + Div. 2, combined) D. Jon and Orbs

不吃土司边

于 2022-06-04 20:36:21 发布

阅读量79

点赞数

分类专栏： acm 文章标签：动态规划算法

本文链接：https://blog.csdn.net/weixin_45714303/article/details/125125339

版权

acm 专栏收录该内容

302 篇文章 1 订阅

订阅专栏

这段代码实现了一个概率动态规划算法，用于解决在k个类别中选择元素的问题。算法计算了在前i个元素中选择j个不同类别的元素的概率，并在给定概率阈值下找出满足条件的最小元素数量。程序读取k和查询次数q，然后进行多次查询，输出满足概率条件的元素数量。

摘要由CSDN通过智能技术生成

#include<bits/stdc++.h>

using namespace std;

#define X first
#define Y second
#define eps  1e-2
#define gcd __gcd
#define pb push_back
#define PI acos(-1.0)
#define lowbit(x) (x)&(-x)
#define bug printf("!!!!!\n");
#define mem(x,y) memset(x,y,sizeof(x))

typedef long long LL;
typedef long double LD;
typedef pair<int,int> pii;
typedef unsigned long long uLL;

const int maxn = 1e5+2;
const int INF  = 1<<30;
const int mod  = 1e9+7;
const int N=1e4+10;
const int M=1e3+10;
int k,q;
double dp[N][M];
void solve(){
    cin>>k>>q;
    dp[0][0]=1;
    for(int i=1;i<N;i++){
        for(int j=1;j<=k;j++){
            dp[i][j]=dp[i-1][j]*j*1.0/k+dp[i-1][j-1]*(k-j+1)*1.0/k;
        }
    }

    while(q--){
        int x;cin>>x;
        int j = 1;
        for(j=1;j<N;j++){
            if(dp[j][k]>=x*0.0005) break;
        }
        cout<<j<<endl;
    }
    
    return;
}

int main()
{
//    freopen("in.txt","r",stdin);
//    freopen("out.txt","w",stdout);
//    ios::sync_with_stdio(false);
    int t = 1;
    //scanf("%d",&t);
    while(t--){
    //    printf("Case %d: ",cas++);
        solve();
    }
    return 0;
}