[题解]P1471

题目背景:

见2021-11-03发表的我又回来了
记得支持我!

题目来源:

P1471

思路:

方差公式还有另一种形式,就是所有数的平方和的算术平均数减去这些数的算数平均数的平方。
写出来就是:
方差公式推导
所以用线段树维护区间和和区间平方和就可以了。代码也非常简单。
但是,在更新数据时,必须先更新区间平方和,再更新区间和,不然会出错。
原因是:
修改区间和
这就是原因所在。
综上,这真是一道优秀的模板题。

代码:

#include<bits/stdc++.h>
using namespace std;
const int MAXN=100000+10;
int k;//类型符,1修改,2查询sum,3 sumsq
int l,r;
double d;
int n,m;
double a[MAXN];
struct SegmentTree{
	int l,r;//左右端点 
	double sum;//附加信息,如区间和
	double sumsq;//区间平方和 
	double add;//懒标记 
}tree[MAXN*4];
void build(int p,int l,int r){
	tree[p].l=l;tree[p].r=r;//确定树的左右端点 
	if(l==r){tree[p].sum=a[l];tree[p].sumsq=a[l]*a[l];return ;}//如果左右相等,那么确定附加信息并返回
	int mid=(l+r)/2;
	build(p*2,l,mid);//递归构造左子树 
	build(p*2+1,mid+1,r);//递归构造右子树 
	tree[p].sum=tree[p*2].sum+tree[p*2+1].sum;//向上更新信息,pushup 
	tree[p].sumsq=tree[p*2].sumsq+tree[p*2+1].sumsq;
}
void pushdown(int p){
	if(tree[p].add){//节点p有懒标记 
		tree[p*2].sumsq+=((tree[p*2].r-tree[p*2].l+1)*tree[p].add*tree[p].add+2.0*tree[p].add*tree[p*2].sum);
		tree[p*2+1].sumsq+=((tree[p*2+1].r-tree[p*2+1].l+1)*tree[p].add*tree[p].add+2.0*tree[p].add*tree[p*2+1].sum);
		tree[p*2].sum+=tree[p].add*(tree[p*2].r-tree[p*2].l+1);
		tree[p*2+1].sum+=tree[p].add*(tree[p*2+1].r-tree[p*2+1].l+1);
		tree[p*2].add+=tree[p].add;
		tree[p*2+1].add+=tree[p].add;
		tree[p].add=0;//清除懒标记 
	}
} 
void update(int p,int l,int r,double d){//区间修改,将a[l]~a[r]都加上d 
	if(l<=tree[p].l&&tree[p].r<=r) {
		tree[p].sumsq+=((tree[p].r-tree[p].l+1)*d*d+2.0*d*tree[p].sum);
		tree[p].sum+=d*(tree[p].r-tree[p].l+1);
		tree[p].add+=d;//增加懒标记 
		return;
	}//找到目标,直接更新 
	//折半查找(修改点x不在左半就在右半)
	pushdown(p);//下传懒标记 
	int mid=(tree[p].l+tree[p].r)/2;//求区间中点 
	if(l<=mid)update(p*2,l,r,d);//递归更新左半
	if(r>mid)update(p*2+1,l,r,d);//递归更新右半
	tree[p].sum=tree[p*2].sum+tree[p*2+1].sum;//pushup
	tree[p].sumsq=tree[p*2].sumsq+tree[p*2+1].sumsq;
}
double querysum(int p,int l,int r){//返回区间[l,r]的区间sum
	//若目标区间完全包含节点p代表的区间,直接返回区间值
	if(l<=tree[p].l&&tree[p].r<=r)return tree[p].sum;
	pushdown(p);//下传懒标记 
	int mid=(tree[p].l+tree[p].r)/2;//求区间中点
	double ans=0;
	if(l<=mid)ans+=querysum(p*2,l,r);
	if(r>mid)ans+=querysum(p*2+1,l,r);
	return ans;
}
double querysumsq(int p,int l,int r){//返回区间[l,r]的区间sumsq
	//若目标区间完全包含节点p代表的区间,直接返回区间值
	if(l<=tree[p].l&&tree[p].r<=r)return tree[p].sumsq;
	pushdown(p);//下传懒标记 
	int mid=(tree[p].l+tree[p].r)/2;//求区间中点
	double ans=0;
	if(l<=mid)ans+=querysumsq(p*2,l,r);
	if(r>mid)ans+=querysumsq(p*2+1,l,r);
	return ans;
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)scanf("%lf",&a[i]);
	build(1,1,n);
	for(int i=1;i<=m;i++){
		scanf("%d",&k);
		if(k==1){
			scanf("%d",&l);
			scanf("%d",&r);
			scanf("%lf",&d);
			update(1,l,r,d);
		}
		else if(k==2){
			scanf("%d",&l);
			scanf("%d",&r);
			printf("%.4lf\n",querysum(1,l,r)*1.0/(r-l+1));
		}
		else{
			scanf("%d",&l);
			scanf("%d",&r);
			double avg=querysum(1,l,r)*1.0/(r-l+1);
			double p=querysumsq(1,l,r);
			printf("%.4lf\n",(p*1.0/(r-l+1)-avg*avg));
		}
	}
	return 0;
}

在这里插入图片描述

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值