线段树维护方差

我们把方差公式展开
在这里插入图片描述
所以只需要维护一个区间平方和和区间和

当我们更新一个区间加时
在这里插入图片描述

#include <cstdio>
#include <iostream>
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
#define Maxn 300010
using namespace std;
double tree1[Maxn],tree2[Maxn],a[Maxn],lazy[Maxn];
void pushup(int x)
{
    tree1[x]=tree1[x<<1]+tree1[x<<1|1];
    tree2[x]=tree2[x<<1]+tree2[x<<1|1];
}
void pushdown(int rt,int x)
{
    if (lazy[rt])
    {
        tree2[rt<<1]+=2*lazy[rt]*tree1[rt<<1]+(x-x>>1)*lazy[rt]*lazy[rt];
        tree2[rt<<1|1]+=2*lazy[rt]*tree1[rt<<1|1]+(x>>1)*lazy[rt]*lazy[rt];
        tree1[rt<<1]+=(x-x>>1)*lazy[rt];
        tree1[rt<<1|1]+=(x>>1)*lazy[rt];
        lazy[rt<<1]+=lazy[rt];
        lazy[rt<<1|1]+=lazy[rt];
        lazy[rt]=0;
    }
}
void build(int rt,int l,int r)
{
    if (l==r)
        tree1[rt]=a[l],tree2[rt]=tree1[rt]*tree1[rt];
    else
    {
        int mid=(l+r)>>1;
        build(lson);
        build(rson);
        pushup(rt);
    }
}
double query_a(int rt,int l,int r,int L,int R)
{
    if (l>=L && r<=R)
        return tree1[rt];
    else
    {
        pushdown(rt,r-l+1);
        int mid=(r+l)>>1;
        double ret=0;
        if (mid>=L)
            ret+=query_a(lson,L,R);
        if (mid<R)
            ret+=query_a(rson,L,R);
        return ret;
    }
}
double query_b(int rt,int l,int r,int L,int R)
{
    if (l>=L && r<=R)
        return tree2[rt];
    else
    {
        pushdown(rt,r-l+1);
        int mid=(r+l)>>1;
        double ret=0;
        if (mid>=L)
            ret+=query_b(lson,L,R);
        if (mid<R)
            ret+=query_b(rson,L,R);
        return ret;
    }
}
void update(int rt,int l,int r,int L,int R,double x)
{
    if (l>=L && r<=R)
        lazy[rt]+=x,tree2[rt]+=2*x*tree1[rt]+x*x*(r-l+1),tree1[rt]+=(r-l+1)*x;
    else
    {
        pushdown(rt,r-l+1);
        int mid=(r+l)>>1;
        if (mid>=L)
            update(lson,L,R,x);
        if (mid<R)
            update(rson,L,R,x);
        pushup(rt);
    }
}

main()
{
    int n,m,x,y,c;
    double z;
    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",&c);
        if (c==2)
            scanf("%d%d",&x,&y),printf("%.4lf\n",query_a(1,1,n,x,y)/(y-x+1));
        if (c==1)
            scanf("%d%d",&x,&y),cin>>z,update(1,1,n,x,y,z);
        if (c==3)
        {
            scanf("%d%d",&x,&y);
            double sum1=query_b(1,1,n,x,y)/(y-x+1),sum2=query_a(1,1,n,x,y)/(y-x+1);
            double ans=sum1-sum2*sum2;
            printf("%.4lf\n",ans);
        }
    }
}

普通线段树(区间和,最值)

#define LL long long      
const int INF = 0x3f3f3f3f;
int n, q;
LL x[300009], ma[300009 << 2], sum[300009 << 2];
int p, l, r;
int ans[1000009];
void init(){
	for (int i = 1; i <= 1000000; i++)
		for (int j = i; j <= 1000000; j += i)
			ans[j]++;
}
void build(int k, int l, int r){
	if (l == r){
		scanf("%lld", &x[l]);
		ma[k] = sum[k] = x[l];
		return;
	}
	int mid = (l + r) >> 1;
	build(k << 1, l, mid);
	build(k << 1 | 1, mid + 1, r);
	ma[k] = max(ma[k << 1], ma[k << 1 | 1]);
	sum[k] = sum[k << 1] + sum[k << 1 | 1];
}
void update(int k, int l, int r, int ll, int rr){
	if (ma[k] <= 2LL) return;
	if (l == r) { ma[k] = sum[k] = x[l] = ans[x[l]]; return; }
	int mid = (l + r) >> 1;
	if (ll <= mid) update(k << 1, l, mid, ll, rr);
	if (rr > mid) update(k << 1 | 1, mid + 1, r, ll, rr);
	ma[k] = max(ma[k << 1], ma[k << 1 | 1]);
	sum[k] = sum[k << 1] + sum[k << 1 | 1];
}
LL query(int k, int l, int r, int ll, int rr){
	if (l >= ll && r <= rr) return sum[k];
	int mid = (l + r) >> 1;
	LL ans = 0;
	if (ll <= mid) ans += query(k << 1, l, mid, ll, rr);
	if (rr > mid) ans += query(k << 1 | 1, mid + 1, r, ll, rr);
	return ans;
}
int main(){
	init();
	while (~scanf("%d %d", &n, &q)){
		build(1, 1, n);
		while (q--){
			scanf("%d %d %d", &p, &l, &r);
			if (p == 1) update(1, 1, n, l, r);
			else printf("%lld\n", query(1, 1, n, l, r));
		}
	}
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值