[BZOJ4821][SDOI2017]相关分析-线段树

最新推荐文章于 2021-08-11 11:43:06 发布

zlttttt

最新推荐文章于 2021-08-11 11:43:06 发布

阅读量189

点赞数

分类专栏：线段树【Segment Tree】

本文链接：https://blog.csdn.net/zlttttt/article/details/79562252

版权

线段树【Segment Tree】专栏收录该内容

14 篇文章 0 订阅

订阅专栏

(╯‵□′)╯︵┻━┻

思路:
纯粹的线段树。

首先拆式子。具体方式为，把平均数全部换成总和除以个数，然后整理一下。
可以发现会拆出来这样一个东西:

a = \sum x i * y i - \sum x * \sum y r - l + 1 \sum x 2 i - ( \sum x ) 2 r - l + 1

$a=\frac{\sum x_i*y_i-\frac{\sum x*\sum y}{r-l+1}}{\sum x_i^2-\frac{(\sum x)^2}{r-l+1}}$

那么维护 $8$ 个值即可:
$\sum x,\sum y,\sum x^2,\sum x*y$ ，以及两种操作对应的 $S$ 和 $T$ 的tag。

然后就是推 $push\_down()$ 函数了。
显然 $\sum x$ 和 $\sum y$ 是非常简单的。

考虑2操作，此时有:

$\sum (x+s)^2=\sum x^2+2sx+s^2$
于是可以利用之前的 $\sum x$ 和 $\sum x^2$ ，加上 $s^2$ 求出。

$\sum (x+s)*(y+t)=\sum xy + \sum sy + \sum tx + \sum st$
对于二、三两项可以利用之前的 $\sum x$ 和 $\sum y$ 求出。

然后是魔幻的3操作:

$\sum\limits_{i=l}^r (s+i)^2=\sum\limits_{i=l}^r s^2+2si+i^2 =\sum s^2+2s*\sum i +\sum i^2$
于是套用前缀和公式和平方和公式即可。

$\sum\limits_{i=l}^r (s+i)(t+i)=\sum\limits_{i=l}^r st+(s+t)i+i^2=\sum st + \sum (s+t)i +\sum i^2$
于是同样套用前缀和公式和平方和公式即可。

最后，把线段树的 $long long$ 改成 $double$ 。
然后就可以过了~

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef double db;

const int N=1e5+9;
const int hath=19260817;
int n,m;
db x[N],y[N];

inline db read()
{
    db x=0,f=1;char ch=getchar();
    while(ch<'0' || '9'<ch){if(ch=='-')f=-1;ch=getchar();}
    while('0'<=ch && ch<='9')x=x*10+(ch^48),ch=getchar();
    return x*f;
}


inline db sum(db l,db r){return (l+r)*(r-l+1)/2;}
inline db sqr(db n){return n*(n+1)*(2*n+1)/6;}


namespace segt
{
    db tsumx[N<<2],tsumy[N<<2];
    db tsqr[N<<2],tyx[N<<2];
    db tagax[N<<2],tagay[N<<2];
    db tagsx[N<<2],tagsy[N<<2];

    inline void marks(int x,db l,db r,db s,db t)
    {
        tagsx[x]=s;tagsy[x]=t;
        tagax[x]=tagay[x]=hath;
        tsumx[x]=(s+s+r-l)*(r-l+1)/2;
        tsumy[x]=(t+t+r-l)*(r-l+1)/2;
        tsqr[x]=sqr(s+r-l)-sqr(s-1);
        tyx[x]=(r-l+1)*s*t+sum(1,r-l)*(s+t)+sqr(r-l);
    }

    inline void marka(int x,db l,db r,db s,db t)
    {
        if(tagax[x]==hath)tagax[x]=s,tagay[x]=t;
        else tagax[x]+=s,tagay[x]+=t;
        tsqr[x]+=tsumx[x]*2*s+s*s*(r-l+1);
        tyx[x]+=s*tsumy[x]+t*tsumx[x]+s*t*(r-l+1);
        tsumx[x]+=s*(r-l+1);
        tsumy[x]+=t*(r-l+1);
    }

    inline void push(int x,int l,int r)
    {
        int mid=l+r>>1;
        if(tagsx[x]!=hath)
        {
            marks(x<<1,l,mid,tagsx[x],tagsy[x]);
            marks(x<<1|1,mid+1,r,tagsx[x]+(mid-l+1),tagsy[x]+(mid-l+1));
            tagsx[x]=tagsy[x]=hath;
        }
        if(tagax[x]!=hath)
        {
            marka(x<<1,l,mid,tagax[x],tagay[x]);
            marka(x<<1|1,mid+1,r,tagax[x],tagay[x]);
            tagax[x]=tagay[x]=hath;
        }
    }

    inline void update(int x)
    {
        tsumx[x]=tsumx[x<<1]+tsumx[x<<1|1];
        tsumy[x]=tsumy[x<<1]+tsumy[x<<1|1];
        tsqr[x]=tsqr[x<<1]+tsqr[x<<1|1];
        tyx[x]=tyx[x<<1]+tyx[x<<1|1];
    }

    inline void build(int rt,int l,int r)
    {
        tagax[rt]=tagay[rt]=hath;
        tagsx[rt]=tagsy[rt]=hath;
        if(l==r)
        {
            tsumx[rt]=x[l];tyx[rt]=x[l]*y[l];
            tsumy[rt]=y[l];tsqr[rt]=x[l]*x[l];
            return;
        }
        int mid=l+r>>1;
        build(rt<<1,l,mid);
        build(rt<<1|1,mid+1,r);
        update(rt);
    }

    inline void modify(int x,int l,int r,int dl,int dr,int s,int t)
    {
        if(l!=r)push(x,l,r);
        if(dl==l && r==dr)
        {
            marks(x,l,r,s+l,t+l);
            return;
        }
        int mid=l+r>>1;
        if(dr<=mid)
            modify(x<<1,l,mid,dl,dr,s,t);
        else if(mid<dl)
            modify(x<<1|1,mid+1,r,dl,dr,s,t);
        else
        {
            modify(x<<1,l,mid,dl,mid,s,t);
            modify(x<<1|1,mid+1,r,mid+1,dr,s,t);
        }
        update(x);
    }

    inline void add(int x,int l,int r,int dl,int dr,int s,int t)
    {
        if(l!=r)push(x,l,r);int mid=l+r>>1;
        if(dl==l && r==dr){marka(x,l,r,s,t);return;}
        if(dr<=mid)add(x<<1,l,mid,dl,dr,s,t);
        else if(mid<dl)add(x<<1|1,mid+1,r,dl,dr,s,t);
        else add(x<<1,l,mid,dl,mid,s,t),add(x<<1|1,mid+1,r,mid+1,dr,s,t);
        update(x);
    }

    inline db query(int x,int l,int r,int dl,int dr,db *t)
    {
        if(l!=r)push(x,l,r);int mid=l+r>>1;
        if(dl==l && r==dr)return t[x];
        if(dr<=mid)return query(x<<1,l,mid,dl,dr,t);
        if(mid<dl)return query(x<<1|1,mid+1,r,dl,dr,t);
        return query(x<<1,l,mid,dl,mid,t)+query(x<<1|1,mid+1,r,mid+1,dr,t);
    }

    inline db check(int x,int l,int r,int p,db *t)
    {
        if(l!=r)push(x,l,r);
        if(l==r)return t[x];int mid=l+r>>1;
        if(p<=mid)return check(x<<1,l,mid,p,t);
        else return check(x<<1|1,mid+1,r,p,t);
    }
}

int main()
{
    n=read();m=read();
    for(int i=1;i<=n;i++)
        x[i]=read();
    for(int i=1;i<=n;i++)
        y[i]=read();

    segt::build(1,1,n);
    while(m--)
    {
        int ty=read();ll l=read(),r=read();
        if(ty==1)
        {
            db sumx=segt::query(1,1,n,l,r,segt::tsumx);
            db sumy=segt::query(1,1,n,l,r,segt::tsumy);
            db up=0,down=0;

            up+=segt::query(1,1,n,l,r,segt::tyx);
            up-=(db)sumx*sumy/((db)r-l+1.0);
            down+=segt::query(1,1,n,l,r,segt::tsqr);
            down-=(db)sumx*sumx/((db)r-l+1.0);

            printf("%.8f\n",up/down);
        }
        else
        {
            ll s=read(),t=read();
            if(ty==2)
                segt::add(1,1,n,l,r,s,t);
            else
                segt::modify(1,1,n,l,r,s,t);
        }
    }

    return 0;
}