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; }