?

?

線段樹模板題：

?

1 #include<iostream> 2 #include<vector> 3 #include<string> 4 #include<cmath> 5 #include<set> 6 #include<algorithm> 7 #include<cstdio> 8 #include<map> 9 #include<cstring> 10 #include<list> 11 12 #define MAXSIZE 50010 13 14 using namespace std; 15 16 int tree[4*MAXSIZE]; 17 int T, N; 18 char order[100]; 19 20 void init() 21 { 22 memset(tree, 0, sizeof(tree)); 23 } 24 25 26 void build(int node, int l, int r) 27 { 28 if(l == r) // 到達葉子節點，賦值 29 { 30 scanf("%d", &tree[node]); 31 return; 32 } 33 34 int mid = (l+r)/2; 35 build(node*2, l, mid); 36 build(node*2+1, mid+1, r); 37 38 tree[node] = tree[node*2] + tree[node*2+1]; 39 } 40 41 // 單點更新 42 void update(int node, int l, int r, int index, int add) 43 { 44 if(l == r) 45 { 46 tree[node] += add; // 更新方式 47 return; 48 } 49 50 int mid = (l+r)/2; 51 if(index <= mid) // 進入左子樹 52 update(node*2, l, mid, index, add); 53 else // 進入右子樹 54 update(node*2+1, mid+1, r, index, add); 55 56 tree[node] = tree[node*2] + tree[node*2 + 1]; 57 } 58 59 // 區間查找 60 int query_range(int node, int l, int r, int L, int R) 61 { 62 if(l <= L && r >= R) 63 return tree[node]; 64 int mid = (L+R)/2; 65 int sum = 0; 66 if(mid >= l) 67 sum += query_range(node*2, l, r, L, mid); 68 if(mid < r) 69 sum += query_range(node*2+1, l, r, mid+1, R); 70 71 return sum; 72 } 73 74 int main() 75 { 76 scanf("%d", &T); 77 for(int t = 1; t <= T; ++t) 78 { 79 scanf("%d", &N); 80 init(); 81 build(1, 1, N); 82 printf("Case %d:\n", t); 83 while(1) 84 { 85 scanf("%s", order); 86 if(strcmp(order, "End") == 0) 87 break; 88 89 int i, j; 90 scanf("%d%d", &i, &j); 91 switch(order[0]) 92 { 93 case 'A': 94 update(1, 1, N, i, j); 95 break; 96 case 'S': 97 update(1, 1, N, i, -j); 98 break; 99 case 'Q': 100 printf("%d\n", query_range(1, i, j, 1, N)); 101 break; 102 } 103 } 104 } 105 106 return 0; 107 }

?

?

?

?

1 #include<iostream> 2 #include<vector> 3 #include<string> 4 #include<cmath> 5 #include<set> 6 #include<algorithm> 7 #include<cstdio> 8 #include<map> 9 #include<cstring> 10 #include<list> 11 12 #define MAXSIZE 50010 13 14 using namespace std; 15 16 int tree[4*MAXSIZE]; 17 int lz[4*MAXSIZE]; 18 int T, N; 19 char order[100]; 20 21 void init() 22 { 23 memset(tree, 0, sizeof(tree)); 24 memset(lz, 0, sizeof(lz)); 25 } 26 27 28 void build(int node, int l, int r) 29 { 30 if(l == r) // 到達葉子節點，賦值 31 { 32 scanf("%d", &tree[node]); 33 return; 34 } 35 36 int mid = (l+r)/2; 37 build(node*2, l, mid); 38 build(node*2+1, mid+1, r); 39 40 tree[node] = tree[node*2] + tree[node*2+1]; 41 } 42 43 44 void push_down(int node, int l, int r) 45 { 46 if(lz[node]) 47 { 48 int mid = (l+r)/2; 49 lz[node*2] += lz[node]; 50 lz[node*2+1] += lz[node]; 51 tree[node*2] += (mid-l+1)*lz[node]; 52 tree[node*2+1] += (r-mid)*lz[node]; 53 lz[node] = 0; 54 } 55 } 56 57 // 區間更新，lr為更新范圍，LR為線段樹范圍，add為更新值 58 void update_range(int node, int l, int r, int L, int R, int add) 59 { 60 if(l <= L && r >= R) 61 { 62 lz[node] += add; 63 tree[node] += (R-L+1)*add; // 更新方式 64 return; 65 } 66 67 push_down(node, L, R); 68 int mid = (L+R)/2; 69 if(mid >= l) // 進入左子樹 70 update_range(node*2, l, r, L, mid, add); 71 if(r > mid) // 進入右子樹 72 update_range(node*2+1, l, r, mid+1, R, add); 73 74 tree[node] = tree[node*2] + tree[node*2 + 1]; 75 } 76 77 // 區間查找 78 int query_range(int node, int l, int r, int L, int R) 79 { 80 if(l <= L && r >= R) 81 return tree[node]; 82 push_down(node, L, R); 83 int mid = (L+R)/2; 84 int sum = 0; 85 if(mid >= l) 86 sum += query_range(node*2, l, r, L, mid); 87 if(mid < r) 88 sum += query_range(node*2+1, l, r, mid+1, R); 89 90 return sum; 91 } 92 93 94 void Add(int i, int j) 95 { 96 update_range(1, i, i, 1, N, j); 97 } 98 99 100 void Sub(int i, int j) 101 { 102 update_range(1, i, i, 1, N, -j); 103 } 104 105 106 int Query(int i, int j) 107 { 108 int sum = query_range(1, i, j, 1, N); 109 return sum; 110 } 111 112 113 int main() 114 { 115 scanf("%d", &T); 116 for(int t = 1; t <= T; ++t) 117 { 118 scanf("%d", &N); 119 init(); 120 build(1, 1, N); 121 printf("Case %d:\n", t); 122 while(1) 123 { 124 scanf("%s", order); 125 if(strcmp(order, "End") == 0) 126 break; 127 128 int i, j; 129 scanf("%d%d", &i, &j); 130 switch(order[0]) 131 { 132 case 'A': 133 Add(i, j); 134 break; 135 case 'S': 136 Sub(i, j); 137 break; 138 case 'Q': 139 printf("%d\n", Query(i, j)); 140 break; 141 } 142 } 143 } 144 145 return 0; 146 }

?

轉載于:https://www.cnblogs.com/FengZeng666/p/11445662.html

總結

以上是生活随笔為你收集整理的敌兵布阵 HDU - 1166 (线段树)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。