您的位置:首页 > 其它

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

2014-07-30 20:50 423 查看
题意就是说一定区间的数会一起修改,然后询问区间求区间和。

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

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|1
ll 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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: