CodeVS 1081 线段树练习 2

1081 线段树练习 2

时间限制: 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

1 2 3 2

2 3

样例输出 Sample Output

数据范围及提示 Data Size & Hint

数据范围

1<=n<=100000

1<=q<=100000

解题：线段树的基本操作。。。

 1 #include <iostream>
 2 #include <cstdio>
 3 #include <cstring>
 4 #include <cmath>
 5 #include <algorithm>
 6 #include <climits>
 7 #include <vector>
 8 #include <queue>
 9 #include <cstdlib>
10 #include <string>
11 #include <set>
12 #include <stack>
13 #define LL long long
14 #define pii pair<int,int>
15 #define INF 0x3f3f3f3f
16 using namespace std;
17 const int maxn = 100010;
18 struct node {
19     int lt,rt,val;
20 };
21 node tree[maxn<<2];
22 void build(int lt,int rt,int v) {
23     tree[v].lt = lt;
24     tree[v].rt = rt;
25     if(lt == rt) {
26         scanf("%d",&tree[v].val);
27         return;
28     }
29     tree[v].val = 0;
30     int mid = (lt + rt)>>1;
31     build(lt,mid,v<<1);
32     build(mid+1,rt,v<<1|1);
33 }
34 void update(int lt,int rt,int v,int val) {
35     if(tree[v].lt >= lt && tree[v].rt <= rt) {
36         tree[v].val += val;
37         return;
38     }
39     if(tree[v].val) {
40         tree[v<<1].val += tree[v].val;
41         tree[v<<1|1].val += tree[v].val;
42         tree[v].val = 0;
43         return;
44     }
45     if(rt >= tree[v<<1|1].lt) update(lt,rt,v<<1|1,val);
46     if(lt <= tree[v<<1].lt) update(lt,rt,v<<1,val);
47 }
48 int query(int lt,int rt,int v) {
49     if(tree[v].lt >= lt && tree[v].rt <= rt) {
50         return tree[v].val;
51     }
52     if(tree[v].val) {
53         tree[v<<1].val += tree[v].val;
54         tree[v<<1|1].val += tree[v].val;
55         tree[v].val = 0;
56     }
57     if(lt <= tree[v<<1].rt) return query(lt,rt,v<<1);
58     if(rt >= tree[v<<1|1].lt) return query(lt,rt,v<<1|1);
59 }
60 int main() {
61     int n,m,op,a,b,x;
62     while(~scanf("%d",&n)) {
63         build(1,n,1);
64         scanf("%d",&m);
65         while(m--) {
66             scanf("%d",&op);
67             if(op == 1) {
68                 scanf("%d %d %d",&a,&b,&x);
69                 update(a,b,1,x);
70             } else if(op == 2) {
71                 scanf("%d",&a);
72                 printf("%d\n",query(a,a,1));
73             }
74         }
75     }
76     return 0;
77 }