【线段树】[LUOGU 方差] 线段树+数学技巧

题目:

题目链接:LUOGU 方差
题解:
(其实最近写线段树的题还有一部分原因是因为这个题,,,码了半个多小时,调了一两天,,,/(ㄒoㄒ)/~~,因为这个题,三个人一起改,,,还是没改出来,,,之后我还去学了Dev的单步跟踪,,,结果,,单步跟踪挂掉了,,一些玄学错误不好调了,完了,,,就只好重新码一遍,结果还好重构之后就是过了,,,不容易啊(〃>_<;〃))
还是 进入正题吧,,,这个题其实在用数学公式化简之后就可以看到是维护两颗线段树就行,一个是区间和,一个就是区间平方和,这里就不展开了,比较显然,,,

代码:

#include<bits/stdc++.h>
#define D double
#define lk (k<<1)
#define rk (k<<1|1)
using namespace std;
inline int read()
{
	int s=0,w=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
	while(ch<='9'&&ch>='0')s=s*10+ch-'0',ch=getchar();
	return s*w;
}
const int sea=1e6+7;
struct hit{int l,r;D w,s,lazy;}tr[sea*4];
int n,m;
void build(int l,int r,int k)
{
    tr[k].lazy=0; tr[k].l=l;tr[k].r=r;
	if(l==r){scanf("%lf",&tr[k].w),tr[k].s=tr[k].w*tr[k].w;return ;}
	int mid=(l+r)/2; 
	build(l,mid,lk);build(mid+1,r,rk);
	tr[k].w=tr[lk].w+tr[rk].w; tr[k].s=tr[lk].s+tr[rk].s;
}
void down(int k)
{
	if(!tr[k].lazy)return;
	int l=tr[k].l,r=tr[k].r,mid=(l+r)>>1;
    tr[lk].lazy+=tr[k].lazy; tr[rk].lazy+=tr[k].lazy;
	tr[lk].s+=2.0*tr[k].lazy*tr[lk].w+tr[k].lazy*tr[k].lazy*1.0*(mid-l+1),
	tr[rk].s+=2.0*tr[k].lazy*tr[rk].w+tr[k].lazy*tr[k].lazy*1.0*(r-mid);
	tr[lk].w+=tr[k].lazy*(mid-l+1)*1.0,tr[rk].w+=tr[k].lazy*1.0*(r-mid);
    tr[k].lazy=0;
}
void alter(int k,int x,int y,D z)
{
	int l=tr[k].l,r=tr[k].r;
	if(x<=l&&r<=y)
	{
		tr[k].lazy+=z; 
		tr[k].s+=2.0*z*tr[k].w+z*z*(r-l+1);
		tr[k].w+=z*(r-l+1);
		return ;
	}
	down(k); int mid=(l+r)/2;
	if(x<=mid) alter(lk,x,y,z);if(y>mid) alter(rk,x,y,z);
	tr[k].w=tr[lk].w+tr[rk].w,tr[k].s=tr[lk].s+tr[rk].s; 
}
D ask1(int k,int x,int y)
{
	int l=tr[k].l,r=tr[k].r;
    if(x<=l&&r<=y) return tr[k].w;
    down(k); int mid=l+r>>1; D ans=0;
    if(x<=mid) ans=ask1(lk,x,y); if(mid<y) ans+=ask1(rk,x,y);
    return ans;
}
D ask2(int k,int x,int y)
{
	int l=tr[k].l,r=tr[k].r;
    if(x<=l&&r<=y) return tr[k].s; 
	down(k); int mid=l+r>>1; D ans=0;
    if(x<=mid) ans=ask2(lk,x,y); if(mid<y) ans+=ask2(rk,x,y);
    return ans;
}
int main()
{
	n=read(); m=read();
	build(1,n,1);
	for(int i=1;i<=m;i++)
	{
		int s=read(); int x,y;D z,ans1,ans2,ss; x=read(),y=read();
		if(s==1) scanf("%lf",&z),alter(1,x,y,z);
		else if(s==2) ans1=ask1(1,x,y),printf("%.4lf\n",ans1/(y-x+1)*1.0);
		else ss=(y-x+1),ans1=ask1(1,x,y),ans2=ask2(1,x,y),printf("%.4lf\n",ans2/ss-ans1*ans1/(ss*ss));
	}
	return 0;
}

Continue……

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值