-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathsegmentTree.cpp
More file actions
60 lines (58 loc) · 1.35 KB
/
segmentTree.cpp
File metadata and controls
60 lines (58 loc) · 1.35 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
#include<bits/stdc++.h>
using namespace std;
#define MAX_SIZE 1000000
int n;
int arr[MAX_SIZE+1];
int tree[MAX_SIZE*4+1];
void buildtree(int node,int b,int e)
{
if(b==e){
tree[node]=arr[b];
return;
}
buildtree(node*2,b,(b+e)/2);
buildtree(node*2+1,(b+e)/2+1,e);
tree[node] = tree[node*2] + tree[node*2+1];
}
int query(int node,int b,int e,int i,int j)
{
if(i>e || j<b)return 0;
if(i<=b && j>=e)return tree[node];
int query1 = query(node*2,b,(b+e)/2,i,j);
int query2 = query(node*2+1,(b+e)/2+1,e,i,j);
return query1+query2;
}
void update(int node, int b, int e, int index, int val) {
if(index<b || e<index)return;
if (index == b && b == e){
arr[b] = val;
tree[node] = val;
}
else {
if (index <= (b+e)/2)update(node*2, b, (b+e)/2, index, val);
else update(node*2+1, (b+e)/2+1, e, index, val);
tree[node] = tree[node*2] + tree[node*2+1];
}
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
cin>>arr[i];
buildtree(1,1,n);
for(int i=1;i<=n*4;i++)
cout<<i<<" "<<tree[i]<<endl;
int k,i,j,p,mark;
cin>>p;
while(p--){
cin>>i>>j;
//cout<<"Update:";
update(1,1,n,i,j);
//else cout<<query(1,1,n,i,j)<<endl;
}
cin>>k;
while(k--){
cin>>i>>j;
cout<<query(1,1,n,i,j)<<endl;
}
}