[bzoj4821][Sdoi2017]相关分析

Posted FallDream

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了[bzoj4821][Sdoi2017]相关分析相关的知识,希望对你有一定的参考价值。

来自FallDream的博客,未经允许,请勿转载,谢谢。


Frank对天文学非常感兴趣,他经常用望远镜看星星,同时记录下它们的信息,比如亮度、颜色等等,进而估算出星星的距离,半径等等。Frank不仅喜欢观测,还喜欢分析观测到的数据。他经常分析两个参数之间(比如亮度和半径)是否存在某种关系。现在Frank要分析参数X与Y之间的关系。他有n组观测数据,第i组观测数据记录了x_i和y_i。他需要一下几种操作1 L,R:用直线拟合第L组到底R组观测数据。用xx表示这些观测数据中x的平均数,用yy表示这些观测数据中y的平均数,即
xx=Σx_i/(R-L+1)(L<=i<=R)
yy=Σy_i/(R-L+1)(L<=i<=R)
如果直线方程是y=ax+b,那么a应当这样计算:
a=(Σ(x_i-xx)(y_i-yy))/(Σ(x_i-xx)(x_i-xx)) (L<=i<=R)
你需要帮助Frank计算a。
2 L,R,S,T:Frank发现测量数据第L组到底R组数据有误差,对每个i满足L <= i <= R,x_i需要加上S,y_i需要加上T。
3 L,R,S,T:Frank发现第L组到第R组数据需要修改,对于每个i满足L <= i <= R,x_i需要修改为(S+i),y_i需要修改为(T+i)。
 
展开式子,答案是
$\frac{\sum(xiyi -xyi - yxi + xy)}{\sum (xi^{2} -2xxi + x^{2})}$
所以维护区间的x,y,xy的和就行了。
x^2维不维护比较随意。
计算记得用longdouble  不然会爆掉
#include<iostream>
#include<cstdio>
#define MN 100000
#define ld long double
using namespace std;
inline int read()
{
    int x = 0 , f = 1; char ch = getchar();
    while(ch < 0 || ch > 9){ if(ch == -) f = -1;  ch = getchar();}
    while(ch >= 0 && ch <= 9){x = x * 10 + ch - 0;ch = getchar();}
    return x * f;
}

struct data{ld x,y,sqx,xy;
    friend data operator + (data a,data b)
    {
        return (data){a.x+b.x,a.y+b.y,a.sqx+b.sqx,a.xy+b.xy};    
    }
}res;
struct Mark{int op,s,t;}M[MN+5];
struct Tree{int l,r,tag;ld s,t;data x;}T[MN*4+5]; 
int n,m,X[MN+5],Y[MN+5],tms=0;
void build(int x,int l,int r)
{
    if((T[x].l=l)==(T[x].r=r))
    {
        T[x].x=(data){X[l],Y[l],(ld)X[l]*X[l],(ld)X[l]*Y[l]};
        return;
    }
    int mid=l+r>>1;
    build(x<<1,l,mid);build(x<<1|1,mid+1,r);
    T[x].x=T[x<<1].x+T[x<<1|1].x;
}
inline ld Sum(int x){return (ld)x*(x+1)/2;}
inline ld Sqr(int x){return (ld)x*(x+1)*(2*x+1)/6;}
void _Mark(int x,int op,int s,int t)
{
    if(op==3){T[x].tag=3;T[x].s=s;T[x].t=t;}
    else if(T[x].tag) T[x].s+=s,T[x].t+=t;
    else T[x].tag=2,T[x].s=s,T[x].t=t;
    if(op==2)
    {
        T[x].x.sqx+=(ld)2*s*T[x].x.x+(ld)(T[x].r-T[x].l+1)*s*s;
        T[x].x.xy+=(ld)s*T[x].x.y+(ld)t*T[x].x.x+(ld)(T[x].r-T[x].l+1)*s*t;
        T[x].x.x+=(ld)(T[x].r-T[x].l+1)*s;
        T[x].x.y+=(ld)(T[x].r-T[x].l+1)*t;
    }
    else
    {
        T[x].x.x=Sum(T[x].r)-Sum(T[x].l-1)+(ld)(T[x].r-T[x].l+1)*s;
        T[x].x.y=Sum(T[x].r)-Sum(T[x].l-1)+(ld)(T[x].r-T[x].l+1)*t;
        T[x].x.sqx=(ld)(T[x].r-T[x].l+1)*s*s+Sqr(T[x].r)-Sqr(T[x].l-1)+(ld)2*s*(Sum(T[x].r)-Sum(T[x].l-1));
        T[x].x.xy=(ld)(T[x].r-T[x].l+1)*s*t+(ld)(s+t)*(Sum(T[x].r)-Sum(T[x].l-1))+Sqr(T[x].r)-Sqr(T[x].l-1); 
    }
}

void pushdown(int x)
{
    if(T[x].tag)
    {
        int l=x<<1,r=l|1; 
        _Mark(l,T[x].tag,T[x].s,T[x].t);
        _Mark(r,T[x].tag,T[x].s,T[x].t);    
        T[x].tag=0;
    }
}

void Modify(int x,int l,int r,int op)
{
    if(T[x].l==l&&T[x].r==r)
    {
        _Mark(x,M[op].op,M[op].s,M[op].t);
        return;    
    }
    pushdown(x);
    int mid=T[x].l+T[x].r>>1;
    if(r<=mid) Modify(x<<1,l,r,op);
    else if(l>mid) Modify(x<<1|1,l,r,op);
    else Modify(x<<1,l,mid,op),Modify(x<<1|1,mid+1,r,op);
    T[x].x=T[x<<1].x+T[x<<1|1].x;
}    

void Query(int x,int l,int r)
{
    if(T[x].l==l&&T[x].r==r){res=res+T[x].x;return;}    
    pushdown(x);
    int mid=T[x].l+T[x].r>>1;
    if(r<=mid) Query(x<<1,l,r);
    else if(l>mid) Query(x<<1|1,l,r);
    else Query(x<<1,l,mid),Query(x<<1|1,mid+1,r);
}

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();
    build(1,1,n);
    for(int i=1;i<=m;++i)
    {
        int op=read(),l=read(),r=read();        
        if(op==1)
        {
            res=(data){0,0,0,0};Query(1,l,r); 
            ld _x=(ld)res.x/(r-l+1),_y=(ld)res.y/(r-l+1);
            ld u=res.xy-_x*res.y-_y*res.x+_x*_y*(r-l+1);
            ld d=res.sqx-2*_x*res.x+_x*_x*(r-l+1);
            printf("%.8lf\n",(double)u/(double)d);
        }
        else
        {
            int s=read(),t=read();
            M[i]=(Mark){op,s,t};
            Modify(1,l,r,i);
        }
    }
    return 0;
}

以上是关于[bzoj4821][Sdoi2017]相关分析的主要内容,如果未能解决你的问题,请参考以下文章

bzoj4821 && luogu3707 SDOI2017相关分析(线段树,数学)

BZOJ4817 SDOI2017 相关分析

[LOJ#2270][BZOJ4912][SDOI2017]天才黑客

[Luogu 3707] SDOI2017 相关分析

SDOI2017 相关分析

洛谷P3707 [SDOI2017]相关分析