poj A Simple Problem with Integers

这道题，做了好几次，彻头彻尾的经典线段树。

#include<stdio.h>
#define HH 1
struct st
{
    __int64 l;
    __int64 r;
    __int64 color;
    __int64 num;
    __int64 sum;
}f[100001*4];
__int64 date[100002];
void build(__int64 l,__int64 r,__int64 n)
{
    __int64 mid=(l+r)/2;
    f[n].l=l;
    f[n].r=r;
    f[n].color=0;
    f[n].num=0;
    if(l==r)
    {
        f[n].sum=date[l];
        return ;
    }
    build(l,mid,n*2);
    build(mid+1,r,n*2+1);
    f[n].sum=f[n*2].sum+f[n*2+1].sum;
}
void down(__int64 n)
{
    f[n].color=0;
    if(f[n*2].color==HH)
        f[n*2].num+=f[n].num;
    else
        f[n*2].num=f[n].num;
    if(f[n*2+1].color==HH)
        f[n*2+1].num+=f[n].num;
    else 
        f[n*2+1].num=f[n].num;
    f[n*2].color=HH;
    f[n*2+1].color=HH;
    f[n*2].sum+=(f[n*2].r-f[n].l+1)*f[n].num;
    f[n*2+1].sum+=(f[n*2+1].r-f[n*2+1].l+1)*f[n].num;
    f[n].num=0;
}
void up(__int64 n)
{
    f[n].sum=f[n*2].sum+f[n*2+1].sum;
}
void update(__int64 l,__int64 r,__int64 num,__int64 n)
{
    __int64 mid=(f[n].l+f[n].r)/2;
    if(f[n].l==l&&f[n].r==r)
    {
        if(f[n].color==HH)
            f[n].num+=num;
        else
            f[n].num=num;
        f[n].color=HH;
        f[n].sum=f[n].sum+(f[n].r-f[n].l+1)*num;
        return ;
    }
    if(f[n].color==HH)
        down(n);
    if(mid>=r)
        update(l,r,num,n*2);
    else if(mid<l)
        update(l,r,num,n*2+1);
    else{
        update(l,mid,num,n*2);
        update(mid+1,r,num,n*2+1);
    }
    up(n);

}
__int64 query(__int64 l,__int64 r,__int64 n)
{
    __int64 mid=(f[n].l+f[n].r)/2,cos=0;
    if(f[n].l==l&&f[n].r==r)
    {
    //    printf("%d %d %d %d..\n",f[n].l,f[n].r,f[n].num,f[n].sum);
        return f[n].sum;
    }
    if(f[n].color==HH)
        down(n);
    if(mid>=r)
       cos+=query(l,r,n*2);
    else if(mid<l)
       cos+=query(l,r,n*2+1);
    else
    {
        cos+=query(l,mid,n*2);
        cos+=query(mid+1,r,n*2+1);
    }
    return cos;
}
int main()
{
    __int64 i,j,n,t,q,l,r,num;
    __int64 k;
    char c[10];
    while(scanf("%I64d%I64d",&n,&q)>0)
    {
        for(i=1;i<=n;i++)
            scanf("%I64d",&date[i]);
        build(1,n,1);
        getchar();
        while(q--)
        {
            scanf("%s",c);
            if(c[0]=='Q')
            {
                scanf("%I64d%I64d",&l,&r);
                k=query(l,r,1);
                printf("%I64d\n",k);
            }
            else if(c[0]=='C')
            {
                scanf("%I64d%I64d%I64d",&l,&r,&num);
                update(l,r,num,1);
            }
        }
    }
    return 0;
}

过的时间还是比较多，要改进才行。