首页 > 代码库 > 线段树练习3

线段树练习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

分类标签 Tags 点此展开 

 

#include<cstdio>#include<cstdlib>#include<cstring>#include<iostream>#include<algorithm>using namespace std;int n,m,a,b,c,d;const int maxn=100000;struct nn{    int r,l,flag,dis;}tree[maxn<<2];void tree_up(int now){    tree[now].dis=tree[now<<1].dis+tree[now<<1|1].dis;}void tree_down(int now){    tree[now<<1].flag+=tree[now].flag;    tree[now<<1|1].flag+=tree[now].flag;    tree[now<<1].dis+=tree[now<<1].flag*(tree[now<<1].r-tree[now<<1].l+1);    tree[now<<1|1].dis+=tree[now<<1|1].flag*(tree[now<<1|1].r-tree[now<<1].l+1);    tree[now].flag=0;    return;} inline void tree_build(int now,int l,int r){    tree[now].l=l;    tree[now].r=r;    if(l==r)    {        scanf("%d",&tree[now].dis);        return ;    }    int mid=(tree[now].l+tree[now].r)>>1;    tree_build(now<<1,l,mid);    tree_build(now<<1|1,mid+1,r);    tree_up(now); } inline void tree_change(int now,int l,int r,int x){    if(tree[now].l==l&&tree[now].r==r)    {        tree[now].dis+=(tree[now].r-tree[now].l+1)*x;        tree[now].flag+=x;        return ;    }    if(tree[now].flag)      tree_down(now);    int mid=(tree[now].l+tree[now].r)>>1;    if(l>mid)      tree_change(now<<1|1,l,r,x);    else if(r<=mid)            tree_change(now<<1,l,r,x);         else          {             tree_change(now<<1,l,mid,x);             tree_change(now<<1|1,mid+1,r,x);         }    tree_up(now);}inline int tree_chaxun(int now,int l,int r){    if(tree[now].l==l&&tree[now].r==r)      return tree[now].dis;    if(tree[now].flag)       tree_down(now);    int mid=(tree[now].l+tree[now].r)>>1;    if(l>mid)        return tree_chaxun(now<<1|1,l,r);    else if(r<=mid)           return tree_chaxun(now<<1,l,r);         else                 return tree_chaxun(now<<1,l,mid)+tree_chaxun(now<<1|1,mid+1,r);}int main(){    scanf("%d",&n);    tree_build(1,1,n);    scanf("%d",&m);    for(int i=1;i<=m;i++)     {         scanf("%d",&a);         if(a==1)          {              scanf("%d%d%d",&b,&c,&d);              tree_change(1,b,c,d);          }        else           {              scanf("%d%d",&b,&c);              cout<<tree_chaxun(1,b,c);          }     }     return 0;}

 

线段树练习3