数据结构--线段树--成段更新(poj 3468 裸题)

鉴于在网上找的代码和我心中的好代码相差较大，贴个自己写的，不好莫怪哈~~

题目链接：http://poj.org/problem?id=3468

代码实现：

#include "stdio.h"  //线段树成段更新 poj 3468
#include "string.h"
#include "stdlib.h"
#define N 100005

typedef struct node {
    int x;
    int y;
    __int64 sum;
    __int64 add;
}point;

point a[3*N];

void PushDown(int t);
__int64 query(int t,int x,int y);
void update(int t,int x,int y,int k);
void CreatTree(int t,int x,int y);

int main()
{
    int Q,n;
    int x,y,z;
    char ch[2];
    while(scanf("%d%d",&n,&Q)!=-1)
    {
        CreatTree(1,1,n);
        while(Q--)
        {
            scanf("%s",ch);
            if(ch[0]=='Q')
            {
                scanf("%d %d",&x,&y);
                printf("%I64d
",query(1,x,y));
            }
            else
            {
                scanf("%d %d %d",&x,&y,&z);
                update(1,x,y,z);
            }

        }
    
    }
    return 0;
}

void update(int t,int x,int y,int k)
{
    if(a[t].x==x && a[t].y==y)
    {
        a[t].add+=(__int64)k;
        a[t].sum+=(__int64)k*(a[t].y-a[t].x+1);
        return ;
    }
    
    PushDown(t);
    
    int temp=2*t;
    int mid=(a[t].x + a[t].y)/2;

    if(y<=mid)
        update(temp,x,y,k);
    else if(x>mid)
        update(temp+1,x,y,k);
    else
    {
        update(temp,x,mid,k);
        update(temp+1,mid+1,y,k);
    }
    a[t].sum=a[temp].sum+a[temp+1].sum;
}


void PushDown(int t)  //主要是这儿延迟一步更新。
{
    if(a[t].add)
    {
        int temp=2*t;
        a[temp].add  += a[t].add;
        a[temp+1].add+= a[t].add;
        
        a[temp].sum  += a[t].add*(a[temp].y - a[temp].x+1);
        a[temp+1].sum+= a[t].add*(a[temp+1].y - a[temp+1].x+1);
        a[t].add=0;
    }
}


__int64 query(int t,int x,int y)
{
    if(a[t].x==x && a[t].y==y)
        return a[t].sum;
    
    int mid=(a[t].x + a[t].y)/2;
    int temp=2*t;
    PushDown(t);
    if(y<=mid)
        return query(temp,x,y);
    else if(x>mid)
        return query(temp+1,x,y);
    else
        return query(temp,x,mid) + query(temp+1,mid+1,y);

}

void CreatTree(int t,int x,int y)  //还是建树
{
    a[t].x=x;
    a[t].y=y;
    a[t].add=a[t].sum=0;
    if(a[t].x==a[t].y)
    {
        scanf("%I64d",&a[t].sum);
        return ;
    }
    int mid=(a[t].x + a[t].y)/2;
    int temp=2*t;
    CreatTree(temp,x,mid);
    CreatTree(temp+1,mid+1,y);
    a[t].sum+=a[temp].sum + a[temp+1].sum;
    return ;
}