题目链接:
https://www.hackerrank.com/challenges/random-number-generator-1
题目大意:
Pi>=0
且
1=Σni=1Pi
最大化
Σni=1Pi∗(1−Pi)∗i
一开始的感觉很对 但是发现拉格朗日出现负数就不会处理
后来问了uoj的大爷们才发现自己对拉格朗日理解不深。。。
套用拉格朗日
f(x)=Σni=1Pi∗(1−Pi)∗i
φ(x)=Σni=1Pi−1
F(x,φ)=Σni=1Pi∗(1−Pi)∗i−λ(Σni=1Pi−1)
F′(x,φ)xi=i−2Pi∗i−λ=0
i−λ=2Pi∗i
Pi=0.5−λ/(2∗i)
Pi=0.5−λ/(2∗i)
Σni=1Pin∗0.5−λ∗0.5∗(Σni=11/n)=1
λ=(n−2)/(Σni=11/n)
λ≈n/ln(n)
然后就会出现
Pi<0
考虑为什么会出现这种情况 感性理解后发现是由于有权重 i 的存在,使越靠后的
Pi
越接近于0.5(上面推出来的对称轴) 那么我们需要使这部分小于0的
Pi
变成0 同时满足此时目标最小
继续套用拉格朗日 二分找到一个分界点mid 使任意
i<mid
有
Pi=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;
}