1. 程式人生 > >【洛谷P1471】方差 分塊

【洛谷P1471】方差 分塊

#include<bits/stdc++.h>
using namespace std;
#define db double
#define N 100005
int n,m,bl[N],t;
db a[N],sum1[330],sum2[330],lz[330];
inline void add(int l,int r,db x){
    int r1=min(bl[l]*t,r);
    for(int i=l;i<=r1;i++){
        sum2[bl[l]]+=(2*a[i]+x)*x;
        sum1[bl[l]]+=x;a[i]+=x;
    }
    if
(bl[l]!=bl[r]) for(int i=(bl[r]-1)*t+1;i<=r;i++){ sum2[bl[r]]+=(2*a[i]+x)*x; a[i]+=x;sum1[bl[r]]+=x; } for(int i=bl[l]+1;i<=bl[r]-1;i++)lz[i]+=x; } inline db query1(int l,int r){ int r1=min(bl[l]*t,r); double ans=0; for(int i=l;i<=r1;i++) ans+=a[i]+lz[bl[l]]; if
(bl[l]!=bl[r]) for(int i=(bl[r]-1)*t+1;i<=r;i++) ans+=a[i]+lz[bl[r]]; for(int i=bl[l]+1;i<=bl[r]-1;i++) ans+=sum1[i]+lz[i]*t; return ans/(r-l+1); } inline db query2(int l,int r){ int r1=min(bl[l]*t,r); double ave=0,ans=0; for(int i=l;i<=r1;i++) ave+=a[i]+lz[bl[l]],ans+=(a[i]+lz[bl[l]])*(
a[i]+lz[bl[l]]); if(bl[l]!=bl[r]) for(int i=(bl[r]-1)*t+1;i<=r;i++) ave+=a[i]+lz[bl[r]],ans+=(a[i]+lz[bl[r]])*(a[i]+lz[bl[r]]); for(int i=bl[l]+1;i<=bl[r]-1;i++) ave+=sum1[i]+lz[i]*t,ans+=sum2[i]+2*sum1[i]*lz[i]+t*lz[i]*lz[i]; return -(ave/(r-l+1))*(ave/(r-l+1))+ans/(r-l+1); } int main(){ scanf("%d%d",&n,&m); t=sqrt(n); for(int i=1;i<=n;i++){ scanf("%lf",&a[i]); bl[i]=(i-1)/t+1; sum1[bl[i]]+=a[i]; sum2[bl[i]]+=a[i]*a[i]; } int flag,l,r; while(m--){ scanf("%d%d%d",&flag,&l,&r); if(flag==1){ db x; scanf("%lf",&x); add(l,r,x); } else if(flag==2)printf("%.3lf\n",query1(l,r)); else printf("%.3lf\n",query2(l,r)); } return 0; }