题意:给你n个连续的不是0就是1的数串,要找出一个与它等长的bi使得对应方差和最小。
题解:比赛的时候,想岔了,先是想了左边第一个不是0右边第一个不是1的然后中间的bi其实都是一样的,但这样绝壁是错的,后面又想了分组,但是没想到要合并,最后wa了7发= =
实际上是先分组,然后在合并就可以了= =
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<map>
#define n 100005
using namespace std;
double num[n],l[n];
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
int h;
int k;
int cnt=0;
scanf("%d",&k);
for(int i=0;i<k;i++)
{
scanf("%d",&h);
num[cnt]=h;
l[cnt++]=1;
while(cnt>=2)
{
if(num[cnt-1]/l[cnt-1]>num[cnt-2]/l[cnt-2])
break;
num[cnt-2]+=num[cnt-1];
l[cnt-2]+=l[cnt-1];
cnt--;
}
}
double sum=0.0;
for(int i=0;i<cnt;i++)
{
double cur=num[i]/l[i];
sum+=cur*cur*(l[i]-num[i])+(1.0-cur)*(1.0-cur)*num[i];
}
printf("%.6lf\n",sum);
}
return 0;
}