首页 > 代码库 > PKU A Simple Problem with Integers (线段树区间更新求和)
PKU A Simple Problem with Integers (线段树区间更新求和)
题意:典型的线段树C,Q问题,有n个数a[i] (1~n),C, a, b,c在[a,b]区间增加c
Q a b 求[a,b]的和。
#include<cstdio> #include<stdlib.h> #include<string.h> #include<string> #include<map> #include<cmath> #include<iostream> #include <queue> #include <stack> #include<algorithm> #include<set> using namespace std; #define INF 1e8 #define eps 1e-8 #define ll __int64 #define maxn 100005 #define mod 1000000009 struct node { ll l,r,sum; ll lazy; }tree[maxn*10]; ll a[maxn]; void Pushup(ll rt) { tree[rt].sum=tree[rt<<1].sum+tree[rt<<1|1].sum; } void Pushdown(ll rt) { if(tree[rt].lazy!=0) { tree[rt<<1].lazy+=tree[rt].lazy;//注意是+=,不是=; tree[rt<<1|1].lazy+=tree[rt].lazy; tree[rt<<1].sum+=(tree[rt<<1].r-tree[rt<<1].l+1)*tree[rt].lazy; tree[rt<<1|1].sum+=(tree[rt<<1|1].r-tree[rt<<1|1].l+1)*tree[rt].lazy; tree[rt].lazy=0; } } void build(ll l,ll r,ll rt) { tree[rt].l=l;tree[rt].r=r; tree[rt].sum=0;tree[rt].lazy=0; if(l==r) { tree[rt].sum=a[l]; return; } ll mid=(l+r)/2; build(l,mid,rt<<1); build(mid+1,r,rt<<1|1); Pushup(rt); } void update(ll rt,ll l,ll r,ll v) { Pushdown(rt); if(tree[rt].l==l&&tree[rt].r==r) { tree[rt].lazy=v; tree[rt].sum+=(tree[rt].r-tree[rt].l+1)*v; return ; } ll mid=(tree[rt].l+tree[rt].r)>>1; if(mid<l) update(rt<<1|1,l,r,v); else if(mid>=r) update(rt<<1,l,r,v); else { update(rt<<1,l,mid,v); update(rt<<1|1,mid+1,r,v); } Pushup(rt); } ll query(ll rt,ll l,ll r) { if(tree[rt].l==l&&tree[rt].r==r) return tree[rt].sum; Pushdown(rt); ll mid=(tree[rt].l+tree[rt].r)>>1,ret=0; if(mid<l) ret+=query(rt<<1|1,l,r); else if(mid>=r) ret+=query(rt<<1,l,r); else { ret+=query(rt<<1,l,mid); ret+=query(rt<<1|1,mid+1,r); } Pushup(rt); return ret; } int main() { ll n,m; while(~scanf("%I64d%I64d",&n,&m)) { for(int i=1;i<=n;i++) scanf("%I64d",&a[i]); build(1,n,1); char s[2]; ll u,v,c; while(m--) { scanf("%s",s); if(s[0]=='Q') { scanf("%I64d%I64d",&u,&v); printf("%I64d\n",query(1,u,v)); } else { scanf("%I64d%I64d%I64d",&u,&v,&c); update(1,u,v,c); } } } return 0; } /* 10 5 1 2 3 4 5 6 7 8 9 10 Q 4 4 Q 1 10 Q 2 4 C 3 6 3 Q 2 4 10 22 1 2 3 4 5 6 7 8 9 10 Q 4 4 C 1 10 3 C 6 10 3 C 6 9 3 C 8 9 -100 C 7 9 3 C 7 10 3 C 1 10 3 Q 6 10 Q 6 9 Q 8 9 Q 7 9 Q 7 10 Q 1 10 Q 2 4 C 3 6 3 Q 9 9 Q 1 1 Q 5 5 Q 6 6 Q 7 7 Q 6 8 */
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。