POJ 3468 A Simple Problem with Integers（线段树）

点我看题目

 

题意  ：N个数Q条命令，Q a b代表输入A_a, A_a+1, ... , A_b的和。C a b c代表让你把c加给A_a, A_a+1, ... , A_b的每一个数。

思路 ：这个题分类好像不是线段树。。不过我硬是按线段树的方法做了，总觉得和HDU1166和1754差不多，反正要注意数据范围很大不能用int，我还因为用了intWA了一次。反正比平时那些模板的更新啊加减什么的都比较麻烦。

#include <iostream>

#include <stdio.h>

#include <string.h>



using namespace std;



const int maxn = 5000004 ;

typedef long long  LL;

struct node

{

    int l,r ;

    LL sum,add ;

    int boundary ;

} Node[maxn] ;

int a[maxn] ;



void build(int v,int  l,int r)

{

    Node[v].l = l ;

    Node[v].r = r ;

    Node[v].boundary = (l+r)/2 ;

    Node[v].add = 0 ;

    if(l == r)

    {

        Node[v].sum = a[l] ;

        return ;

    }

    int mid = (l+r)>>1 ;

    build(v*2,l,mid) ;

    build(v*2+1,mid+1,r) ;

    Node[v].sum = Node[v*2].sum + Node[v*2+1].sum ;

}



void update(int v,int value,int l,int r)

{

    if(Node[v].l == l && Node[v].r == r)

    {

        Node[v].add += value ;

        return ;

    }

    Node[v].sum += (LL)(value*(1+r-l)) ;

    if(Node[v].boundary >= r)

        update(v*2,value,l,r) ;

    else if(Node[v].boundary < l)

        update(v*2+1,value,l,r) ;

    else

    {

        update(v*2,value,l,Node[v].boundary) ;

        update(v*2+1,value,Node[v].boundary+1,r) ;

    }

}

LL query(int v,int l,int r)

{

    if(Node[v].l == l && Node[v].r == r)

    {

        return Node[v].sum + Node[v].add*(LL)(r-l+1) ;

    }

    if(Node[v].add != 0)

    {

        Node[v*2].add += Node[v].add ;

        Node[v*2+1].add += Node[v].add ;

        Node[v].sum += (LL)(Node[v].r-Node[v].l+1)*Node[v].add ;

        Node[v].add = 0 ;

    }

    if(Node[v].boundary >= r)

        return query(v*2,l,r) ;

    else if(Node[v].boundary < l)

        return query(v*2+1,l,r) ;

    else return query(v*2,l,Node[v].boundary)+query(v*2+1,Node[v].boundary+1,r) ;

}



int main()

{

    int m,n ;

    while(~scanf("%d %d",&n,&m))

    {

        for(int i = 1 ; i <= n ; i++ )

        {

            scanf("%d",&a[i]) ;

        }

        int x,y,z ;

        build(1,1,n) ;

        char str[54] ;

        for(int i = 1 ; i <= m ; i++)

        {

            scanf("%s",str) ;

            if(str[0] == 'Q')

            {

                scanf("%d %d",&x,&y) ;

                printf("%lld\n",query(1,x,y)) ;

            }

            if(str[0] == 'C')

            {

                scanf("%d %d %d",&x,&y,&z) ;

                update(1,z,x,y) ;

            }

        }

    }

    return 0;

}

View Code