CF gym102992 F.Fireworks(几何分布、三分)

https://codeforces.com/gym/102992/problem/F
题意：制作一支烟花需要花费 $n$ 分钟，每支烟花有 $p\times 10^{-4}$ 的概率是完美的，每次可以花费 $m$ 分钟点燃之前制作的所有烟花，若发现至少有一支完美的，则停止。问最优策略下，最短的停下的时间期望是多少？

思路：
首先每次都是生产相同数量的烟花再点燃的，因为要求最优策略，而且每次生产的烟花是否完美的这个事件是相互独立的，所以如果第一批生产 $x$ 支是最优策略，那么第二批同样应该生产 $x$ 支再点燃。

假设最优策略是连续制作 $x$ 支后点燃
设每支烟花不完美的概率为 $p_0=1-p\times 10^{-4}$
那么 $x$ 支烟花中至少有一支完美烟花的概率为 $1-p_0^x$
则需要制作烟花的轮数的期望是$\frac{1}{1-p_0^x}$（几何分布）
总时间$f(x)=\frac{n\ast x+m}{1-p_0^x}$

分析分子分母的函数图像可以大致判断，f(x)先减后增，是一个凹函数，所以可以三分答案

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<vector>
using namespace std;
//#define int long long
#define ll long long
#define uql unsigned long long
#define pii pair<int,int>
#define mid ((l + r)>>1)
#define chl (root<<1)
#define chr (root<<1|1)
#define lowbit(x) ( x&(-x) )
const int manx = 2e5+10;
const int manx2 = 4e7 + 10;
const ll INF = 1e18;
const int mod = 1000000007;
const double eps=1e-18;

int n,m,t;
double p,p0;
double F(double x)
{
    if(p0<eps)return 1.0*n*x+m;
    return (1.0*n*x+m)/(1.0-pow(p0,x));
}
double F(int x)
{
    if(p0<eps)return 1.0*n*x+m;
    return (1.0*n*x+m)/(1.0-pow(p0,x));
}
int main()
{
    scanf("%d",&t);
    while(t--){
        scanf("%d%d%lf",&n,&m,&p);
        p0=1.0-p*1e-4;
        double l=1,r=1e9,ml,mr;
        double fl,fr;
        while(r-l>1e-9){
            double tmp=(r-l)/3;
            ml=l+tmp,mr=r-tmp;
            fl=F(ml),fr=F(mr);
            if(fl<=fr)
                r=mr;
            else l=ml;
        }
        int tmp=(int)l;
        double ans=min(F(tmp),F(tmp+1));
        printf("%.10f\n",ans);
    }
    return 0;
}