Hackerrank Random Number Generator

题目链接:
https://www.hackerrank.com/challenges/random-number-generator-1
题目大意:
$P_i>=0$
且$1=Σ_{i=1}^{n}P_i$
最大化 $Σ_{i=1}^{n}P_i*(1-P_i)*i$

一开始的感觉很对 但是发现拉格朗日出现负数就不会处理
后来问了uoj的大爷们才发现自己对拉格朗日理解不深。。。

套用拉格朗日
$f(x)=Σ_{i=1}^{n}P_i*(1-Pi)*i$
$φ(x)=Σ_{i=1}^{n}P_i -1$
$F(x,φ)=Σ_{i=1}^{n}P_i*(1-Pi)*i-λ(Σ_{i=1}^{n}P_i -1)$

$F'(x,φ)_{xi}=i-2P_i*i-λ=0$
$i-λ=2P_i*i$
$P_i=0.5-λ/(2*i)$

$P_i=0.5-λ/(2*i)$
$Σ_{i=1}^{n}P_in*0.5-λ*0.5*(Σ_{i=1}^n 1/n)=1$
$λ=(n-2)/(Σ_{i=1}^n 1/n)$
$λ≈n/ln(n)$
然后就会出现$P_i<0$
考虑为什么会出现这种情况 感性理解后发现是由于有权重 i 的存在，使越靠后的$P_i$越接近于0.5(上面推出来的对称轴) 那么我们需要使这部分小于0的$P_i$ 变成0 同时满足此时目标最小
继续套用拉格朗日 二分找到一个分界点mid 使任意$i<mid$有$P_i=0$ 并满足上述条件
然后就是$O(Tlogn)$ 了

#include<cstdio>
#include<cstring>
using namespace std;

double f[1000001];


int main()
{
    int n;
    n=1000000;
    int i;
    for(int i=1;i<=n;i++)f[i]=f[i-1]+1.0/i;
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        if(n==1){puts("0.00000000000");continue;}
        if(n==2){puts("0.50000000000");continue;}
        int l=1,mid,r=n;
        while(l<r)
        {
            mid=l+r>>1;
            double F=(n-mid-1)/(f[n]-f[mid-1]);
            if(0.5-F/2/mid>=0)r=mid;
            else l=mid+1;
        }
        r=n;
        double F=(r-l-1)/(f[r]-f[l-1]);
        double As=(r-l+1)*1ll*(r+l)/4.0-F*F/2*(f[r]-f[l-1]);
        printf("%.10f\n",As/(2*n-1));
    }
    return 0;
}