POJ 3468 A Simple Problem with Integers

一般说简单的题目不简单,但是有了线段树之后这些题都是简单题~~~


题目大意:

给定N个数的序列,有Q个操作。操作分两种:1、将某个区间内的所有数都加上某个数。2、计算某个区间的所有数之和并输出。


解题思路:

简单的线段树。单点更新会TLE,只能区间更新。


下面是代码:

#include <stdio.h>
#include <string.h>
#include <iostream>
#define BigInteger long long
using namespace std;
const int Max=100000;
struct nodenode
{
    BigInteger sum,val;
} node[Max<<2];
void PushUp(int tr)
{
    node[tr].sum=node[tr<<1].sum+node[tr<<1|1].sum;
}
void PushDown(int tr,int m)
{
    if(node[tr].val)
    {
        node[tr<<1].val+=node[tr].val;
        node[tr<<1|1].val+=node[tr].val;
        node[tr<<1].sum+=(BigInteger)(m-(m>>1))*node[tr].val;
        node[tr<<1|1].sum+=(BigInteger)(m>>1)*node[tr].val;
        node[tr].val=0;
    }
}
void Build(int l,int r,int tr)
{
    node[tr].val=0;
    if(l==r)
    {
        scanf("%I64d",&node[tr].sum);
        return ;
    }
    int m=(l+r)>>1;
    Build(l,m,tr<<1);
    Build(m+1,r,tr<<1|1);
    PushUp(tr);
}
BigInteger query(int L,int R,int l,int r,int tr)
{
    if(L<=l&&r<=R)
    {
        return node[tr].sum;
    }
    int m=(l+r)>>1;
    PushDown(tr,r-l+1);
    BigInteger ans=0;
    if(L<=m)ans+=query(L,R,l,m,tr<<1);
    if(m<R)ans+=query(L,R,m+1,r,tr<<1|1);
    PushUp(tr);
    return ans;
}
void Update(int L,int R,int add,int l,int r,int tr)
{
    if(L<=l&&r<=R)
    {
        node[tr].sum+=(BigInteger)add*(r-l+1);
        node[tr].val+=add;
        return ;
    }
    PushDown(tr,r-l+1);
    int m=(l+r)>>1;
    if(L<=m)Update(L,R,add,l,m,tr<<1);
    if(m<R)Update(L,R,add,m+1,r,tr<<1|1);
    PushUp(tr);
}
int main()
{
    int n,q,a,b;
    BigInteger c;
    while(scanf("%d%d",&n,&q)!=EOF)
    {
        Build(1,n,1);
        char s[3];
        while(q--)
        {
            scanf("%s",s);
            if(s[0]=='Q')
            {
                cin >>a>>b;
                cout <<query(a,b,1,n,1)<<endl;
            }
            else if(s[0]=='C')
            {
                cin >>a>>b>>c;
                Update(a,b,c,1,n,1);
            }
        }
    }
    return 0;
}


你可能感兴趣的:(线段树,poj,刷题)