HDU 4923 （杭电多校 #6 1003题）Room and Moor(公式+栈）

题目地址：HDU 4923

比赛的时候脑残了。。思路完全想出来了。。只不过想出了个根本不可能存在的极端数据，然后一看输入数据是100组，就把自己给否决了。。。sad。。当时就应该大胆试一试的。。。

这个题首先可以把最前面的0和最后面的1去掉，因为这两块总可以用0和1抵消掉。然后中间就分成了10相间的情况，然后根据10相间，可以分成若干组，每一组都是由几个1和几个0组成的。比如说1101101110,就可以分成110,110,1110这样的三组。

然后这时候可以可以对每一组内只取一个数来使得这组的和最小。那应该怎么找这个数呢。假设这组有a个1，b个0,取的数为X,就可以列出式子(a+b)x^2-2*a*x+a,对他求导得知当x为a/(a+b)的时候最小。所以这一段应该取a/(a+b).

然后剩下的只有一个任务了，怎么让某些组合在一块保证所取得这个数是递增的。为了方便，可以弄一个栈。然后当栈为空的时候就放进去。遍历这些组。当该组的x比栈顶的x小，这组就跟前面的那组合并，并把栈顶的弹出。合并完后的值继续与栈顶的比较，直到栈为空或者栈顶的x比该组的x小，再放入栈。如果一上来就比栈顶的大，那就直接放入栈。

最后，由于每组取的值都是a/(a+b)，可以代入列出的式子。求得最终的值是(a*b)/(a+b)。可以直接利用这个值来计算。

注意中间过程可能会爆int，要用int64。代码如下：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <math.h>
#include <stack>
using namespace std;
__int64 a[110000], b[110000];
struct node
{
__int64 l, r, sum, a;
double z;
}q[110000];
int main()
{
__int64 t, n, i, j, pos1, pos2, cnt, flag;
double ans;
scanf("%I64d",&t);
while(t--)
{
stack<node>Q;
scanf("%I64d",&n);
for(i=0;i<n;i++)
{
scanf("%I64d",&a[i]);
}
for(i=0;i<n;i++)
{
if(a[i])
{
pos1=i;
break;
}
}
for(i=n-1;i>=0;i--)
{
if(a[i]==0)
{
pos2=i+1;
break;
}
}
flag=0;
cnt=0;
int b=0, c=0;
for(i=pos1;i!=pos2;)
{
b=0;
c=0;
q[cnt].l=i;
while(a[i]==1)
{
i++;
b++;
}
while(a[i]==0&&i<n)
{
i++;
c++;
}
q[cnt].a=b;
q[cnt].r=i-1;
q[cnt].sum=q[cnt].r-q[cnt].l+1;
q[cnt++].z=b*1.0/(b+c);
}
for(i=0;i<cnt;i++)
{
if(Q.empty())
{
Q.push(q[i]);
continue ;
}
while(!Q.empty()&&q[i].z<=Q.top().z)
{
node f1=Q.top();
q[i].l-=f1.sum;
q[i].sum+=f1.sum;
q[i].a+=f1.a;
q[i].z=q[i].a*1.0/q[i].sum;
Q.pop();
}
Q.push(q[i]);
}
ans=0;
while(!Q.empty())
{
node f1=Q.top();
ans+=f1.a*(f1.sum-f1.a)*1.0/f1.sum;
Q.pop();
}
printf("%.6lf\n",ans);
}
return 0;
}