【Luogu1471】方差（线段树）

【Luogu1471】方差（线段树）

题面

这种傻逼题。。。自己去看把。。

题解

这题太傻比了
把方差公式拆开
维护平方和和区间和
修改就把平方和的公式拆开
简直傻逼的题目

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define MAX 150000
#define lson (now<<1)
#define rson (now<<1|1)
struct Node
{
	double s,ps;
	double ly;
}t[MAX<<4];
double a[MAX];
int n,m;
void Build(int now,int l,int r)
{
	if(l==r)
	{
		t[now].s=a[l];
		t[now].ps=a[l]*a[l];
		return;
	}
	int mid=(l+r)>>1;
	Build(lson,l,mid);Build(rson,mid+1,r);
	t[now].s=t[lson].s+t[rson].s;
	t[now].ps=t[lson].ps+t[rson].ps;
}
void pushdown(int now,int l,int r)
{
	double k=t[now].ly;
	int mid=(l+r)>>1;
	t[lson].ly+=k;
	t[rson].ly+=k;
	t[lson].ps+=2*t[lson].s*k+(mid-l+1)*k*k;
	t[rson].ps+=2*t[rson].s*k+(r-mid)*k*k;
	t[lson].s+=(mid-l+1)*k;
	t[rson].s+=(r-mid)*k;
	t[now].ly=0;
}
void putlazy(int now,int l,int r,double k)
{
	t[now].ly+=k;
	t[now].ps+=2*t[now].s*k+(r-l+1)*k*k;
	t[now].s+=(r-l+1)*k;
}
void Modify(int now,int l,int r,int L,int R,double w)
{
	if(L<=l&&r<=R){putlazy(now,l,r,w);return;}
	pushdown(now,l,r);
	int mid=(l+r)>>1;
	if(L<=mid)Modify(lson,l,mid,L,R,w);
	if(R>mid)Modify(rson,mid+1,r,L,R,w);
	t[now].ps=t[lson].ps+t[rson].ps;
	t[now].s=t[lson].s+t[rson].s;
}
double Query1(int now,int l,int r,int L,int R)
{
	if(L<=l&&r<=R)return t[now].s;
	pushdown(now,l,r);
	double ret=0;
	int mid=(l+r)>>1;
	if(L<=mid)ret+=Query1(lson,l,mid,L,R);
	if(R>mid)ret+=Query1(rson,mid+1,r,L,R);
	return ret;
}
double Query2(int now,int l,int r,int L,int R)
{
	if(L<=l&&r<=R)return t[now].ps;
	pushdown(now,l,r);
	double ret=0;
	int mid=(l+r)>>1;
	if(L<=mid)ret+=Query2(lson,l,mid,L,R);
	if(R>mid)ret+=Query2(rson,mid+1,r,L,R);
	return ret;
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;++i)scanf("%lf",&a[i]);
	Build(1,1,n);
	int opt,ll,rr;
	double kk;
	while(m--)
	{
		scanf("%d%d%d",&opt,&ll,&rr);
		if(opt==1)
		{
			scanf("%lf",&kk);
			Modify(1,1,n,ll,rr,kk);
		}
		else if(opt==2)
		{
			double ret=Query1(1,1,n,ll,rr);
			printf("%.4lf\n",ret/(rr-ll+1));
		}
		else
		{
			double c1=Query1(1,1,n,ll,rr);
			double c2=Query2(1,1,n,ll,rr);
			double c3=c1/(rr-ll+1);
			double ans=c2-2*c1*c3+(rr-ll+1)*c3*c3;
			printf("%.4lf\n",ans/(rr-ll+1));
		}
	}
	return 0;
}