BZOJ 4821 [Sdoi2017]相关分析 ——线段树

打开题面，看到许多$\sum$

woc，好神啊，SDOI好强啊

然后展开之后，woc，SDOI好弱啊，怎么T3出个线段树裸题啊。

最后写代码的时候，woc，SDOI怎么出个这么码农的题啊，怎么调啊。

想想自己加上考场$Debuff$是肯定写不出来的。

#include <map>
#include <cmath>
#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define F(i,j,k) for (int i=j;i<=k;++i)
#define D(i,j,k) for (int i=j;i>=k;--i)
//#define double long double
#define ll long long
#define mp make_pair
#define maxn 400005

#define sum(x) (1LL*(x)*((x)+1)/2)
#define sum2(x) (1LL*(x)*((x)+1)*(2*(x)+1)/6)

double sum_x[maxn],sum_y[maxn],sum_xy[maxn],sum_xx[maxn];
double tag_x[maxn],tag_y[maxn];
bool tag_c[maxn];
int x[maxn],y[maxn];
double sumx,sumy,sumxy,sumxx;

int n,m;

void update(int o,int l,int r)
{
sum_x[o]=sum_x[o<<1]+sum_x[o<<1|1];
sum_y[o]=sum_y[o<<1]+sum_y[o<<1|1];
sum_xy[o]=sum_xy[o<<1]+sum_xy[o<<1|1];
sum_xx[o]=sum_xx[o<<1]+sum_xx[o<<1|1];
}

void add_tag_x(int o,int l,int r,double s)
{
tag_x[o]+=s;
sum_xx[o]+=1LL*s*s*(r-l+1)+2*s*sum_x[o];
sum_xy[o]+=1LL*sum_y[o]*s;
sum_x[o]+=1LL*(r-l+1)*s;
}

void add_tag_y(int o,int l,int r,double t)
{
tag_y[o]+=t;
sum_xy[o]+=1LL*sum_x[o]*t;
sum_y[o]+=1LL*(r-l+1)*t;
}

void add_cov(int o,int l,int r)
{
sum_y[o]=sum_x[o]=sum(r)-sum(l-1);
sum_xy[o]=sum_xx[o]=sum2(r)-sum2(l-1);
tag_x[o]=tag_y[o]=0; tag_c[o]=true;
}

void pushdown(int o,int l,int r)
{
if (o==0||l==r) return;
int mid=l+r>>1;
if (tag_c[o])
{
add_cov(o<<1,l,mid);
add_cov(o<<1|1,mid+1,r);
tag_c[o]=false;
}
if (tag_x[o])
{
add_tag_x(o<<1,l,mid,tag_x[o]);
add_tag_x(o<<1|1,mid+1,r,tag_x[o]);
tag_x[o]=0;
}
if (tag_y[o])
{
add_tag_y(o<<1,l,mid,tag_y[o]);
add_tag_y(o<<1|1,mid+1,r,tag_y[o]);
tag_y[o]=0;
}
}

void st_modify(int o,int l,int r,int L,int R,double s,double t)
{
if (L<=l&&r<=R){add_tag_x(o,l,r,s);add_tag_y(o,l,r,t);return;}
int mid=l+r>>1; pushdown(o,l,r);
if (L<=mid) st_modify(o<<1,l,mid,L,R,s,t);
if (R>mid) st_modify(o<<1|1,mid+1,r,L,R,s,t);
update(o,l,r);
}

void modify(int l,int r,double s,double t)
{st_modify(1,1,n,l,r,s,t);}

void st_cov(int o,int l,int r,int L,int R)
{
if (L<=l&&r<=R){add_cov(o,l,r);return;}
int mid=l+r>>1; pushdown(o,l,r);
if (L<=mid) st_cov(o<<1,l,mid,L,R);
if (R>mid) st_cov(o<<1|1,mid+1,r,L,R);
update(o,l,r);
}

void cover(int l,int r)
{st_cov(1,1,n,l,r);}

void st_query(int o,int l,int r,int L,int R)
{
if (L<=l&&r<=R){sumx+=sum_x[o],sumy+=sum_y[o],sumxx+=sum_xx[o];sumxy+=sum_xy[o];return;}
int mid=l+r>>1; pushdown(o,l,r);
if (R<=mid) st_query(o<<1,l,mid,L,R);
else if (L>mid) st_query(o<<1|1,mid+1,r,L,R);
else st_query(o<<1,l,mid,L,R),st_query(o<<1|1,mid+1,r,L,R);
}

double query(int l,int r)
{
sumx=0; sumy=0;sumxx=0; sumxy=0;
double len=r-l+1; st_query(1,1,n,l,r);
printf("%.10lf\n",(len*sumxy-sumx*sumy)/(len*sumxx-sumx*sumx));
}

void build(int o,int l,int r)
{
if (l==r)
{
sum_x[o]=x[l]; sum_y[o]=y[l];
sum_xy[o]=1LL*x[l]*y[l]; sum_xx[o]=1LL*x[l]*x[l];
return ;
}
int mid=l+r>>1;
build(o<<1,l,mid);
build(o<<1|1,mid+1,r);
update(o,l,r);
}

int main()
{
scanf("%d%d",&n,&m);
F(i,1,n) scanf("%d",&x[i]);
F(i,1,n) scanf("%d",&y[i]);
build(1,1,n);
F(i,1,m)
{
int opt,l,r,s,t;
scanf("%d",&opt);
switch(opt)
{
case 1:
scanf("%d%d",&l,&r);
query(l,r);
break;
case 2:
scanf("%d%d%d%d",&l,&r,&s,&t);
modify(l,r,(double)s,(double)t);
break;
case 3:
scanf("%d%d%d%d",&l,&r,&s,&t);
cover(l,r);
modify(l,r,(double)s,(double)t);
}
}
}