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;

}

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