poj 3468 A Simple Problem with Integers

題目鏈接:http://poj.org/problem?id=3468

題目描述:有一個長度爲n 的序列,有兩個操作

1. C l r c 給[l ,r ]增加c (-10000<=c<=10000)

2. Q l r    查詢[l ,r ]的和

解題思路:線段樹直接秒!每個節點維護 區間整體增加的值,以及區間和

但是線段樹常數有點大。。。。沒辦法

另外,有負數的猥瑣輸入並不能加速。。。sad

//#pragma comment(linker,"/STACK:102400000,102400000")
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<string>
#define ll long long
#define db double
#define PB push_back
#define lson k<<1
#define rson k<<1|1
using namespace std;

const int N = 100005;
const ll INF = 1000000000;

struct node
{
    int l,r;
    int val;
    ll sum;
} tr[N*4];
int a[N];

void push_up(int k)
{
    tr[k].sum=tr[lson].sum+tr[rson].sum;
}

void push_down(int k)
{
    if(tr[k].val)
    {
        tr[lson].val+=tr[k].val;
        tr[lson].sum+=(ll)tr[k].val*(ll)(tr[lson].r-tr[lson].l+1);
        tr[rson].val+=tr[k].val;
        tr[rson].sum+=(ll)tr[k].val*(ll)(tr[rson].r-tr[rson].l+1);
        tr[k].val=0;
    }
}

void build(int k,int l,int r)
{
    tr[k].l=l,tr[k].r=r;
    tr[k].val=0;
    if(l<r)
    {
        int mid=(l+r)>>1;
        build(lson,l,mid);
        build(rson,mid+1,r);
        push_up(k);
    }else tr[k].sum=a[l];
}

void add(int k,int l,int r,int val)
{
    if(tr[k].l==l&&tr[k].r==r)
    {
        tr[k].val+=val;
        tr[k].sum+=(ll)val*(ll)(tr[k].r-tr[k].l+1);
        return;
    }
    push_down(k);
    int mid=(tr[k].l+tr[k].r)>>1;
    if(r<=mid) add(lson,l,r,val);
    else if(l>mid) add(rson,l,r,val);
    else add(lson,l,mid,val),add(rson,mid+1,r,val);
    push_up(k);
}

ll query(int k,int l,int r)
{
    if(tr[k].l==l&&tr[k].r==r)
    {
        return tr[k].sum;
    }
    push_down(k);
    int mid=(tr[k].l+tr[k].r)>>1;
    if(r<=mid) return query(lson,l,r);
    else if(l>mid) return query(rson,l,r);
    else return query(lson,l,mid)+query(rson,mid+1,r);
}

int main()
{
#ifdef PKWV
    freopen("in.in","r",stdin);
#endif // PKWV
    int n,Q;
    scanf("%d%d",&n,&Q);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    build(1,1,n);
    while(Q--)
    {
        char ch[10];
        scanf("%s",ch);
        if(ch[0]=='Q')
        {
            int l,r;
            scanf("%d%d",&l,&r);
            printf("%I64d\n",query(1,l,r));
        }else
        {
            int l,r,c;
            scanf("%d%d%d",&l,&r,&c);
            add(1,l,r,c);
        }
    }
    return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章