codevs 1082 线段树练习 3
1082 线段树练习 3
时间限制: 3 s
空间限制: 128000 KB
题目等级 : 大师 Master
题目描述 Description
给你N个数，有两种操作：
1：给区间[a,b]的所有数增加X
2：询问区间[a,b]的数的和。
输入描述
Input Description
第一行一个正整数n，接下来n行n个整数，
再接下来一个正整数Q，每行表示操作的个数，
如果第一个数是1，后接3个正整数，
表示在区间[a,b]内每个数增加X,如果是2，
表示操作2询问区间[a,b]的和是多少。
pascal选手请不要使用readln读入
输出描述
Output Description
对于每个询问输出一行一个答案
样例输入
Sample Input
3
1
2
3
2
1 2 3 2
2 2 3
样例输出
Sample Output
9
数据范围及提示
Data Size & Hint
数据范围
1<=n<=200000
1<=q<=200000
这题需要开long long，函数也要开！坑
1 //s d s 2 #include<cstdio> 3 #include<iostream> 4 #include<cstdlib> 5 using namespace std; 6 const int N=5000006; 7 long long a[N],sum[N];int miku[N]; 8 long long b,c,d,e; 9 10 void update(int rt) 11 { 12 sum[rt]=sum[rt<<1]+sum[rt<<1|1]; 13 } 14 15 void build(int l,int r,int rt) 16 { 17 if(l==r) 18 { 19 sum[rt]=a[l]; 20 return ; 21 } 22 int m=(l+r)>>1; 23 build(l,m,rt<<1); 24 build(m+1,r,rt<<1|1); 25 update(rt); 26 } 27 28 void midify_interval(int l,int r,int rt,int nowl,int nowr,int neww) 29 { 30 if(nowl==l&&nowr==r) 31 { 32 miku[rt]+=neww; 33 sum[rt]+=neww*(r-l+1); 34 return ; 35 } 36 int m=(l+r)>>1; 37 sum[rt<<1]+=miku[rt]*(m-l+1); 38 sum[rt<<1|1]+=miku[rt]*(r-m); 39 miku[rt<<1]+=miku[rt]; 40 miku[rt<<1|1]+=miku[rt]; 41 miku[rt]=0; 42 if(nowr<=m) midify_interval(l,m,rt<<1,nowl,nowr,neww); 43 else if(nowl>m) midify_interval(m+1,r,rt<<1|1,nowl,nowr,neww); 44 else 45 { 46 midify_interval(l,m,rt<<1,nowl,m,neww); 47 midify_interval(m+1,r,rt<<1|1,m+1,nowr,neww); 48 } 49 update(rt); 50 } 51 52 long long query(int l,int r,int rt,int nowl,int nowr) 53 { 54 if(nowl<=l&&nowr>=r) 55 { 56 return sum[rt]; 57 } 58 int m=(r+l)>>1; 59 sum[rt<<1]+=miku[rt]*(m-l+1); 60 sum[rt<<1|1]+=miku[rt]*(r-m); 61 miku[rt<<1]+=miku[rt]; 62 miku[rt<<1|1]+=miku[rt]; 63 miku[rt]=0; 64 long long ans=0; 65 if(nowl<=m) ans+=query(l,m,rt<<1,nowl,nowr); 66 if(nowr>m)ans+=query(m+1,r,rt<<1|1,nowl,nowr); 67 return ans; 68 } 69 70 71 72 int main() 73 { 74 int n; 75 scanf("%lld",&n); 76 for(int i=1;i<=n;i++)scanf("%lld",a+i); 77 build(1,n,1); 78 int m; 79 scanf("%lld",&m); 80 81 for(int i=1;i<=m;i++) 82 { 83 scanf("%lld",&b); 84 if(b==1) 85 { 86 scanf("%lld%lld%lld",&c,&d,&e); 87 midify_interval(1,n,1,c,d,e); 88 } 89 if(b==2) 90 { 91 scanf("%lld%lld",&c,&d); 92 printf("%lld\n",query(1,n,1,c,d)); 93 } 94 } 95 return 0; 96 97 }
