POJ 3468 A Simple Problem with Integers(线段树区间修改)

来源:互联网 发布:淘宝网男鞋休闲鞋2016 编辑:程序博客网 时间:2024/06/10 21:42

题意就是说一定区间的数会一起修改,然后询问区间求区间和。

要用延迟更新,意思就是说,如果当前这个区间已经包含在要求更新的区间范围内,那么就在这里停止,不需要更新下去了,因为这个区间下面的所有的都需要加一个数,标记一下,下次访问下面的数的时候再进行更新。

AC代码:

#include<cstdio>#include<ctype.h>#include<algorithm>#include<iostream>#include<cstring>#include<vector>#include<stack>#include<cmath>#include<queue>#include<set>#include<ctime>using namespace std;#define pb push_back#define ll long long#define lson l, m, rt<<1#define rson m+1, r, rt<<1|1ll T[1<<18],a[1<<18];void pushup(int rt){    T[rt] = T[rt<<1]+T[rt<<1|1];}void build(int l, int r, int rt){    a[rt] = 0;    if(l == r)    {        scanf("%lld",&T[rt]);//        cout<<" "<<rt<<endl;        return;    }    int m = (l+r)>>1;    build(lson);    build(rson);    pushup(rt);}void pushdown(int rt, int m){    if(a[rt])    {       a[rt<<1] += a[rt];       a[rt<<1|1] += a[rt];       T[rt<<1] += a[rt]*(ll)(m - (m>>1));       T[rt<<1|1] += a[rt]*(ll)(m>>1);       a[rt] = 0;    }}void update(int L, int R, int k, int l, int r, int rt){    if(L <= l && R >= r)    {        a[rt] += (ll)k;        T[rt] += (ll)k*(ll)(r-l+1);        return;    }    pushdown(rt, r-l+1);    int m = (r+l)>>1;    if(L <= m) update(L,R,k,lson);    if(R > m) update(L,R,k,rson);    pushup(rt);}ll query(int L, int R, int l, int r, int rt){    if(L <= l && R >= r)        return T[rt];    pushdown(rt,r-l+1);    int m = (r+l)>>1;    ll ret = 0;    if(L <= m) ret += query(L,R,lson);    if(R > m) ret += query(L,R,rson);    return ret;}int main(){    //freopen("input.txt","r",stdin);    //freopen("o1.txt","w",stdout);    int i,j,n,q;    while(~scanf("%d%d",&n,&q))    {        build(1,n,1);        char temp;        while(q--)        {            getchar();            scanf("%c",&temp);            int a,b,c;            if(temp == 'Q')            {                scanf("%d%d",&a,&b);                printf("%lld\n",query(a,b,1,n,1));            }            else            {                scanf("%d%d%d",&a,&b,&c);                update(a,b,c,1,n,1);//                cout<<T[]            }        }    }    return 0;}


0 0
原创粉丝点击