T1081 线段树练习 2 codevs

http://codevs.cn/problem/1081/

 时间限制: 1 s

 空间限制: 128000 KB

 题目等级 : 大师 Master

题目描述 Description

给你N个数，有两种操作

1：给区间[a,b]的所有数都增加X

2：询问第i个数是什么？

输入描述 Input Description

第一行一个正整数n，接下来n行n个整数，再接下来一个正整数Q，表示操作的个数. 接下来Q行每行若干个整数。如果第一个数是1，后接3个正整数a,b,X，表示在区间[a,b]内每个数增加X,如果是2，后面跟1个整数i, 表示询问第i个位置的数是多少。

输出描述 Output Description

对于每个询问输出一行一个答案

样例输入 Sample Input

3

1

2

3

2

1 2 3 2

2 3

样例输出 Sample Output

5

数据范围及提示 Data Size & Hint

数据范围

1<=n<=100000

1<=q<=100000

分类标签 Tags 点此展开

 

树状数组 线段树 树结构

 

 #include <iostream>
 #define maxn 100005

 using namespace std;

 int n,m,x,a,b,X;
 struct node
 {
     int now,dis,l,r,mid,flag;
 }tree[maxn*];

 void tree_up(int now)
 {
     tree[now].dis=tree[now*].dis+tree[now*+].dis;
     return ;
 }

 void tree_build(int now,int l,int r)
 {
     tree[now].l=l,tree[now].r=r;
     if(tree[now].l==tree[now].r)
     {
         cin>>tree[now].dis;
         return ;
     }
     tree[now].mid=(tree[now].l+tree[now].r)/;
     tree_build(now*,l,tree[now].mid);
     tree_build(now*+,tree[now].mid+,r);
     tree_up(now);
 }

 void tree_down(int now)
 {
     tree[now*].flag+=tree[now].flag;
     tree[now*].dis+=tree[now].flag*(tree[now*].r-tree[now*].l+);
     tree[now*+].flag+=tree[now].flag;
     tree[now*+].dis+=tree[now].flag*(tree[now*+].r-tree[now*+].l+);
     tree[now].flag=;    return ;
 }

 void tree_change(int now,int l,int r,int x)
 {
     if(tree[now].l==l&&tree[now].r==r)
     {
         tree[now].dis+=x*(r-l+);
         tree[now].flag+=x;
         return ;
     }
     if(tree[now].flag) tree_down(now);
     if(tree[now].mid>=r)    tree_change(now*,l,r,x);
     else if(tree[now].mid<l)    tree_change(now*+,l,r,x);
     else
     {
         tree_change(now*,l,tree[now].mid,x);
         tree_change(now*+,tree[now].mid+,r,x);
     }
     tree_up(now);
 }

 int tree_query(int now,int to)
 {
     if(tree[now].l==tree[now].r)    return tree[now].dis;
     if(tree[now].flag) tree_down(now);
     if(tree[now].mid>=to)    return tree_query(now*,to);
     else if(tree[now].mid<to)    return tree_query(now*+,to);
 }

 int main()
 {
     cin>>n;
     tree_build(,,n);
     cin>>m;
     while(m--)
     {
         cin>>x;
         if(x==)
         {
             cin>>a>>b>>X;
             tree_change(,a,b,X);
         }
         else
         {
             cin>>X;
             cout<<tree_query(,X)<<endl;
         }
     }
     return ;
 }

区间改变 单点查询