JZYZOJ1527 [haoi2012]高速公路 线段树 期望

http://172.20.6.3/Problem_Show.asp?id=1527

日常线段树的pushdown写挂，果然每次写都想得不全面，以后要注意啊……
求期望部分也不熟练，和平均数搞混也是orz，我已经是个期望都求不出来的废人了。
这道题显然（大概）每个段的贡献是val[i]*(y-i+1)*(i-x+1);
整体来说算是一看就是线段树的题。

代码

#include<iostream>  
#include<cstdio>  
#include<cstring>  
#include<algorithm>  
#include<cmath>
#include<queue>
using namespace std;
#define lc x*2
#define rc x*2+1
const int maxn=1<<18;
long long n,m;char ch[2]={}; 
struct seg{
    long long l,r,v,w,v1,v2,w1,w2;
    seg(){l=r=v1=v2=w1=w2=v=w=0;}
}e[maxn];
long long v1,v2,v3;
void pushdown(long long x,long long v){
    if(v!=0){
        long long siz=e[x].r-e[x].l+1;
        e[x].v+=siz*v;
        e[x].w1+=e[x].v1*v;
        e[x].w2+=e[x].v2*v;
        e[x].w+=v;
    }
}
void doit(long long x){
    pushdown(lc,e[x].w);
    pushdown(rc,e[x].w);
    e[x].w=0;
}
void pushup(long long x){
    if(e[x].l<e[x].r){
        e[x].v=e[lc].v+e[rc].v;
        e[x].w1=e[lc].w1+e[rc].w1;
        e[x].w2=e[lc].w2+e[rc].w2;
    }
}
void cha(long long x,long long l,long long r,long long w){
    if(l<=e[x].l&&e[x].r<=r){
        pushdown(x,w);
        return;
    }doit(x);
    long long mid=(e[x].l+e[x].r)/2;
    if(l<=mid)cha(lc,l,r,w);
    if(r>mid)cha(rc,l,r,w);
    pushup(x);
}
void sum(long long x,long long l,long long r){
    if(l<=e[x].l&&e[x].r<=r){
        v1+=e[x].v;v2+=e[x].w1;v3+=e[x].w2;
        return;
    }doit(x);
    long long mid=(e[x].l+e[x].r)/2;
    if(l<=mid)sum(lc,l,r);
    if(r>mid)sum(rc,l,r);
    pushup(x);
}
void build(long long x,long long l,long long r){
    e[x].l=l;e[x].r=r;
    if(l==r){
        e[x].v1=l;e[x].v2=l*l;return;
    }
    long long mid=(l+r)/2;
    build(lc,l,mid);
    build(rc,mid+1,r);
    e[x].v1=e[lc].v1+e[rc].v1;e[x].v2=e[lc].v2+e[rc].v2;
}
long long gcd(long long x,long long y){
    while(y){
        long long w=y;y=x%y;x=w;
    }
    return x;
}
int main(){
    build(1,1,1<<17);
    scanf("%I64d%I64d",&n,&m);
    long long x,y,v;
    for(int i=1;i<=m;i++){
        scanf("%s",&ch);
        if(ch[0]=='C'){
            scanf("%I64d%I64d%I64d",&x,&y,&v);
            cha(1,x,y-1,v);
        }
        else{
            scanf("%I64d%I64d",&x,&y);
            v1=v2=v3=0;sum(1,x,y-1);
            long long ans=v1*(y-x*y)+v2*(x+y-1)-v3;
            long long z=y-x;long long zong=z*(z+1)/2;
            long long w=gcd(ans,zong);
            printf("%I64d/%I64d
",ans/w,zong/w);
        }
    }
    return 0;
}