[洛谷1471] 方差 {线段树}

题目

https://www.luogu.org/problemnew/show/P1471

---

解题思路

这道题跟[jzoj 4231] 寻找神格{线段树+分解因式}是孪生兄弟。
不过需要注意一下change的是实数，ask的是平均数，不是总和。

---

代码

#include<cstdio>
#define rr register 
#define db double 
#define ll long long 
using namespace std;
const ll N=100010;
ll t,n,m,q,w; 
db x,y,z,e,ans,sum[N*4],num[N*4],add[N*4];
void good(ll l,ll r,ll mid,ll p){
	if (add[p]){
    	sum[p*2]=sum[p*2]+add[p]*add[p]*(mid-l+1)+2*num[p*2]*add[p];
        num[p*2]=num[p*2]+add[p]*(mid-l+1);
        add[p*2]=add[p*2]+add[p];
        sum[p*2+1]=sum[p*2+1]+add[p]*add[p]*(r-mid)+2*num[p*2+1]*add[p];
        num[p*2+1]=num[p*2+1]+add[p]*(r-mid);
        add[p*2+1]=add[p*2+1]+add[p];
        add[p]=0;
    }
}
void change(ll p,ll l,ll r,ll a,ll b,db x){
    if (l==a&&r==b){
        sum[p]=sum[p]+x*x*(r-l+1)+2*num[p]*x;
        num[p]=num[p]+x*(r-l+1);
        add[p]=add[p]+x;
        return;
    }
    ll mid=(l+r)>>1; good(l,r,mid,p); 
    if (b<=mid) change(p*2,l,mid,a,b,x); 
	else if (a>mid) change(p*2+1,mid+1,r,a,b,x);
	else change(p*2,l,mid,a,mid,x),change(p*2+1,mid+1,r,mid+1,b,x);
    sum[p]=sum[p*2]+sum[p*2+1];
    num[p]=num[p*2]+num[p*2+1];
}
db getsum(ll p,ll l,ll r,ll a,ll b){
    if (l==a&&r==b) return sum[p];
    ll mid=(l+r)>>1; good(l,r,mid,p); 
    if (b<=mid) return getsum(p*2,l,mid,a,b);
    else if (a>mid) return getsum(p*2+1,mid+1,r,a,b);
    else return getsum(p*2,l,mid,a,mid)+getsum(p*2+1,mid+1,r,mid+1,b);
}
db getnum(ll p,ll l,ll r,ll a,ll b){
    if (l==a&&r==b) return num[p];
    ll mid=(l+r)>>1; good(l,r,mid,p); 
    if (b<=mid) return getnum(p*2,l,mid,a,b);
    else if (a>mid) return getnum(p*2+1,mid+1,r,a,b);
    else return getnum(p*2,l,mid,a,mid)+getnum(p*2+1,mid+1,r,mid+1,b);
}
int main(){
    scanf("%lld%lld",&n,&m);
    for(rr int i=1;i<=n;++i)
        scanf("%lf",&e),change(1,1,n,i,i,e);
    for(rr int i=1;i<=m;i++){
        scanf("%lld",&t);
		if (t==1){
            scanf("%lld%lld%lf",&q,&w,&e);
            change(1,1,n,q,w,e);
        } else if (t==2){
            scanf("%lld%lld",&q,&w);
            printf("%.4lf\n",getnum(1,1,n,q,w)/(w-q+1));
        } else{
            scanf("%lld%lld",&q,&w);
            x=getnum(1,1,n,q,w);y=getsum(1,1,n,q,w);z=(w-q+1);
            ans=y*z; ans-=x*x; ans/=(z*z);
            printf("%.4lf\n",ans);
        }
    }
}