BZOJ4821 SDOI2017相关分析（线段树）

　　纯粹的码农题。维护x的和、y的和、xy的和、x²的和即可。可能会炸long long。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
#define ll long double
#define N 100010
char getc(){char c=getchar();while ((c<'A'||c>'Z')&&(c<'a'||c>'z')&&(c<''||c>'')) c=getchar();return c;}
int gcd(int n,int m){return m==?n:gcd(m,n%m);}
int read()
{
    int x=,f=;char c=getchar();
    while (c<''||c>'') {if (c=='-') f=-;c=getchar();}
    while (c>=''&&c<='') x=(x<<)+(x<<)+(c^),c=getchar();
    return x*f;
}
int n,m,a[N],b[N],L[N<<],R[N<<];
ll lazyaddx[N<<],lazyaddy[N<<],s[N],t[N];
bool lazyres[N<<];
struct data
{
    ll sumx,sumy,sumxy,sqr;
    data operator +(const data&a) const
    {
        data c;
        c.sumx=sumx+a.sumx;
        c.sumy=sumy+a.sumy;
        c.sumxy=sumxy+a.sumxy;
        c.sqr=sqr+a.sqr;
        return c;
    }
}tree[N<<];
void up(int k){tree[k]=tree[k<<]+tree[k<<|];}
void update(int k,ll x,ll y,bool p)
{
    if (p)
    {
        tree[k].sumx=tree[k].sumy=s[R[k]]-s[L[k]-];
        tree[k].sumxy=tree[k].sqr=t[R[k]]-t[L[k]-];
        lazyaddx[k]=lazyaddy[k]=,lazyres[k]=;
    }
    tree[k].sqr+=x*x*(R[k]-L[k]+)+*x*tree[k].sumx;
    tree[k].sumxy+=x*tree[k].sumy;
    tree[k].sumx+=(R[k]-L[k]+)*x;
    tree[k].sumxy+=y*tree[k].sumx;
    tree[k].sumy+=(R[k]-L[k]+)*y;
    lazyaddx[k]+=x,lazyaddy[k]+=y;
}
void down(int k)
{
    update(k<<,lazyaddx[k],lazyaddy[k],lazyres[k]);
    update(k<<|,lazyaddx[k],lazyaddy[k],lazyres[k]);
    lazyaddx[k]=lazyaddy[k]=;lazyres[k]=;
}
void build(int k,int l,int r)
{
    L[k]=l,R[k]=r;
    if (l==r) {tree[k].sumx=a[l],tree[k].sumy=b[l],tree[k].sumxy=1ll*a[l]*b[l],tree[k].sqr=1ll*a[l]*a[l];return;}
    int mid=l+r>>;
    build(k<<,l,mid);
    build(k<<|,mid+,r);
    up(k);
}
data query(int k,int l,int r)
{
    if (L[k]==l&&R[k]==r) return tree[k];
    if (lazyaddx[k]||lazyaddy[k]||lazyres[k]) down(k);
    int mid=L[k]+R[k]>>;
    if (r<=mid) return query(k<<,l,r);
    else if (l>mid) return query(k<<|,l,r);
    else return query(k<<,l,mid)+query(k<<|,mid+,r);
}
void res(int k,int l,int r)
{
    if (L[k]==l&&R[k]==r) {update(k,,,);return;}
    if (lazyaddx[k]||lazyaddy[k]||lazyres[k]) down(k);
    int mid=L[k]+R[k]>>;
    if (r<=mid) res(k<<,l,r);
    else if (l>mid) res(k<<|,l,r);
    else res(k<<,l,mid),res(k<<|,mid+,r);
    up(k);
}
void add(int k,int l,int r,int x,int y)
{
    if (L[k]==l&&R[k]==r) {update(k,x,y,);return;}
    if (lazyaddx[k]||lazyaddy[k]||lazyres[k]) down(k);
    int mid=L[k]+R[k]>>;
    if (r<=mid) add(k<<,l,r,x,y);
    else if (l>mid) add(k<<|,l,r,x,y);
    else add(k<<,l,mid,x,y),add(k<<|,mid+,r,x,y);
    up(k);
}
int main()
{
#ifndef ONLINE_JUDGE
    freopen("bzoj4821.in","r",stdin);
    freopen("bzoj4821.out","w",stdout);
    const char LL[]="%I64d\n";
#else
    const char LL[]="%lld\n";
#endif
    n=read(),m=read();
    for (int i=;i<=n;i++) a[i]=read();
    for (int i=;i<=n;i++) b[i]=read();
    for (int i=;i<=n;i++) s[i]=s[i-]+i,t[i]=t[i-]+1ll*i*i;
    build(,,n);
    while (m--)
    {
        int op=read();
        if (op==)
        {
            int l=read(),r=read();
            data k=query(,l,r);
            ll x=(ll)k.sumx/(r-l+),y=(ll)k.sumy/(r-l+);
            printf("%.6lf\n",(double)((k.sumxy-k.sumx*y-k.sumy*x+x*y*(r-l+))/(k.sqr-*k.sumx*x+x*x*(r-l+))));
        }
        else
        {
            int l=read(),r=read(),x=read(),y=read();
            if (op==) res(,l,r);add(,l,r,x,y);
        }
    }
    return ;
}