线段树模板

转载自 http://blog.csdn.net/wjw0130/article/details/38498481

HH神犇的板子写的太漂亮了...

来自NOTONLYSUCCESS博客，汇总了一下

写法十分飘逸，简洁。膜拜。。

单点替换、单点增减、区间求和、区间最值

[html]  view plain  copy

1. #include <cstdio>  
2. #include <algorithm>  
3. using namespace std;  
4. #define lson l , m , rt << 1  
5. #define rson m + 1 , r , rt << 1 | 1  
6. const int maxn = 222222;  
7. int MAX[maxn<<2];  
8. int MIN[maxn<<2];  
9. int SUM[maxn<<2];  
10. int max(int a,int b){if(a>b)return a;else return b;}  
11. int min(int a,int b){if(a<b)return a;else return b;}  
12. void PushUP(int rt)  
13. {  
14.     MAX[rt] = max(MAX[rt<<1] , MAX[rt<<1|1]);  
15.     MIN[rt] = min(MIN[rt<<1] , MIN[rt<<1|1]);  
16.     SUM[rt] = SUM[rt<<1] + SUM[rt<<1|1];  
17. }  
18. void build(int l,int r,int rt) {  
19.     if (l == r)  
20.         {  
21.         scanf("%d",&MAX[rt]);  
22.         MIN[rt] = MAX[rt];  
23.         SUM[rt] = MAX[rt];  
24.         //printf("mi = %d\n",MIN[rt]);  
25.     //  printf("ma = %d\n",MAX[rt]);  
26.         return ;  
27.     }  
28.     int m = (l + r) >> 1;  
29.     build(lson);  
30.     build(rson);  
31.     PushUP(rt);  
32. }  
33. void update(int p,int tihuan,int l,int r,int rt)  
34. {  
35.     if (l == r) {  
36.         MAX[rt] = tihuan;  
37.         MIN[rt] = tihuan;  
38.         SUM[rt] = tihuan;  
39.         return ;  
40.     }  
41.     int m = (l + r) >> 1;  
42.     if (p <= m) update(p , tihuan ,lson);  
43.     else update(p , tihuan , rson);  
44.     PushUP(rt);  
45. }  
46. void update1(int p,int add,int l,int r,int rt)  
47. {  
48.     if (l == r) {  
49.         SUM[rt] = SUM[rt] + add;  
50.         return ;  
51.     }  
52.     int m = (l + r) >> 1;  
53.     if (p <= m) update1(p , add ,lson);  
54.     else update1(p , add , rson);  
55.     PushUP(rt);  
56. }  
57. int query(int L,int R,int l,int r,int rt)  
58. {  
59.     if (L <= l && r <= R)  
60.     {  
61.         return MAX[rt];  
62.     }  
63.     int m = (l + r) >> 1;  
64.     int ret = -1;  
65.     if (L <= m) ret = max(ret , query(L , R , lson));  
66.     if (R > m)  ret =  max(ret , query(L , R , rson));  
67.     return ret;  
68. }  
69. int query1(int L,int R,int l,int r,int rt)  
70. {  
71.     if (L <= l && r <= R)  
72.     {  
73.         return MIN[rt];  
74.     }  
75.     int m = (l + r) >> 1;  
76.     int ret = 99999;  
77.     if (L <= m) ret = min(ret , query1(L , R , lson));  
78.     if (R > m)  ret =  min(ret , query1(L , R , rson));  
79.     return ret;  
80. }  
81. int queryhe(int L,int R,int l,int r,int rt)  
82. {  
83.     if (L <= l && r <= R)  
84.     {  
85.         return SUM[rt];  
86.     }  
87.     int m = (l + r) >> 1;  
88.     int ret = 0;  
89.     if (L <= m) ret += queryhe(L , R , lson);  
90.     if (R > m)  ret +=  queryhe(L , R , rson);  
91.     return ret;  
92. }  
93. int main()  
94. {  
95.     int n , m;  
96.     while (~scanf("%d%d",&n,&m))  
97.     {  
98.         build(1 , n , 1);  
99.         while (m --) {  
100.             char op[2];  
101.             int a , b;  
102.             scanf("%s%d%d",op,&a,&b);  
103.             if (op[0] == 'Q') //区间求最大  
104.             {  
105.                /* for(int i = 1;i<=10;i++)  
106.                     printf("%d ",MAX[i]);  
107.                 puts("");*/  
108.                 printf("%d\n",query(a , b , 1 , n , 1));  
109.             }  
110.             else if(op[0]=='U') //单点替换  
111.                 update(a , b , 1 , n , 1);  
112.             else if(op[0]=='M')//区间求最小  
113.             {  
114.                 /*for(int i = 1;i<=10;i++)  
115.                     printf("%d ",MIN[i]);  
116.                 puts("");*/  
117.                 printf("%d\n",query1(a , b , 1 , n , 1));  
118.             }  
119.             else if(op[0]=='H')//区间求和  
120.             {  
121.                 printf("%d\n",queryhe(a , b , 1 , n , 1));  
122.             }  
123.             else if(op[0]=='S')//单点增加  
124.             {  
125.                 scanf("%d%d",&a,&b);  
126.                 update1(a , b , 1 , n , 1);  
127.             }  
128.              else if(op[0]=='E')//单点减少  
129.             {  
130.                 scanf("%d%d",&a,&b);  
131.                 update1(a , -b , 1 , n , 1);  
132.             }  
133.         }  
134.     }  
135.     return 0;  
136. }  

区间替换

[html]  view plain  copy

1. #include <iostream>  
2. #include <cstdio>  
3. #include <cstdlib>  
4. #include <cstring>  
5. #include <algorithm>  
6.   
7. #define max(a,b) (a>b)?a:b  
8. #define min(a,b) (a>b)?b:a  
9. #define lson l , m , rt << 1  
10. #define rson m + 1 , r , rt << 1 | 1  
11. #define LL long long  
12. const int maxn = 100100;  
13. using namespace std;  
14.   
15. int lazy[maxn<<2];  
16. int sum[maxn<<2];  
17. void PushUp(int rt)//由左孩子、右孩子向上更新父节点  
18. {  
19.     sum[rt] = sum[rt<<1] + sum[rt<<1|1];  
20. }  
21. void PushDown(int rt,int m) //向下更新  
22. {  
23.     if (lazy[rt]) //懒惰标记  
24.     {  
25.         lazy[rt<<1] = lazy[rt<<1|1] = lazy[rt];  
26.         sum[rt<<1] = (m - (m >> 1)) * lazy[rt];  
27.         sum[rt<<1|1] = ((m >> 1)) * lazy[rt];  
28.         lazy[rt] = 0;  
29.     }  
30. }  
31. void build(int l,int r,int rt)//建树  
32. {  
33.     lazy[rt] = 0;  
34.   
35.     if (l== r)  
36.     {  
37.         scanf("%d",&sum[rt]);  
38.         return ;  
39.     }  
40.     int m = (l + r) >> 1;  
41.     build(lson);  
42.     build(rson);  
43.     PushUp(rt);  
44. }  
45. void update(int L,int R,int c,int l,int r,int rt)//更新  
46. {  
47.     //if(L>l||R>r) return;  
48.     if (L <= l && r <= R)  
49.     {  
50.         lazy[rt] = c;  
51.         sum[rt] = c * (r - l + 1);  
52.         //printf("%d %d %d %d %d\n", rt, sum[rt], c, l, r);  
53.         return ;  
54.     }  
55.     PushDown(rt , r - l + 1);  
56.     int m = (l + r) >> 1;  
57.     if (L <= m) update(L , R , c , lson);  
58.     if (R > m) update(L , R , c , rson);  
59.     PushUp(rt);  
60. }  
61.   
62. LL query(int L,int R,int l,int r,int rt)   
63. {  
64.     if (L <= l && r <= R)  
65.     {  
66.         //printf("%d\n", sum[rt]);  
67.         return sum[rt];  
68.     }  
69.     PushDown(rt , r - l + 1);  
70.     int m = (l + r) >> 1;  
71.     LL ret = 0;  
72.     if (L <= m) ret += query(L , R , lson);  
73.     if (m < R) ret += query(L , R , rson);  
74.     return ret;  
75. }  
76. int main()   
77. {  
78.     int  n , m;  
79.     char str[5];  
80.     while(scanf("%d%d",&n,&m))  
81.     {  
82.         build(1 , n , 1);  
83.         while (m--)  
84.         {  
85.             scanf("%s",str);  
86.             int a , b , c;  
87.             if(str[0]=='T')  
88.             {  
89.                 scanf("%d%d%d",&a,&b,&c);  
90.                 update(a , b , c , 1 , n , 1);  
91.             }  
92.             else if(str[0]=='Q')  
93.             {  
94.                 scanf("%d%d",&a,&b);  
95.                 cout<<query(a,b,1,n,1)<<endl;  
96.             }  
97.         }  
98.     }  
99.     return 0;  
100. }  

区间增减

[html]  view plain  copy

1. #include <iostream>  
2. #include <cstdio>  
3. #include <cstdlib>  
4. #include <cstring>  
5. #include <algorithm>  
6.   
7. #define max(a,b) (a>b)?a:b  
8. #define min(a,b) (a>b)?b:a  
9. #define lson l , m , rt << 1  
10. #define rson m + 1 , r , rt << 1 | 1  
11. #define LL __int64  
12. const int maxn = 100100;  
13. using namespace std;  
14. LL lazy[maxn<<2];  
15. LL sum[maxn<<2];  
16.   
17. void putup(int rt)  
18. {  
19.     sum[rt] = sum[rt<<1] + sum[rt<<1|1];  
20. }  
21. void putdown(int rt,int m)  
22. {  
23.     if (lazy[rt])  
24.     {  
25.         lazy[rt<<1] += lazy[rt];  
26.         lazy[rt<<1|1] += lazy[rt];  
27.         sum[rt<<1] += lazy[rt] * (m - (m >> 1));  
28.         sum[rt<<1|1] += lazy[rt] * (m >> 1);  
29.         lazy[rt] = 0;  
30.     }  
31. }  
32. void build(int l,int r,int rt) {  
33.     lazy[rt] = 0;  
34.     if (l == r)  
35.     {  
36.         scanf("%I64d",&sum[rt]);  
37.         return ;  
38.     }  
39.     int m = (l + r) >> 1;  
40.     build(lson);  
41.     build(rson);  
42.     putup(rt);  
43. }  
44. void update(int L,int R,int c,int l,int r,int rt)  
45. {  
46.     if (L <= l && r <= R)  
47.     {  
48.         lazy[rt] += c;  
49.         sum[rt] += (LL)c * (r - l + 1);  
50.         return ;  
51.     }  
52.     putdown(rt , r - l + 1);  
53.     int m = (l + r) >> 1;  
54.     if (L <= m) update(L , R , c , lson);  
55.     if (m < R) update(L , R , c , rson);  
56.     putup(rt);  
57. }  
58. LL query(int L,int R,int l,int r,int rt)  
59. {  
60.     if (L <= l && r <= R)  
61.     {  
62.         return sum[rt];  
63.     }  
64.     putdown(rt , r - l + 1);  
65.     int m = (l + r) >> 1;  
66.     LL ret = 0;  
67.     if (L <= m) ret += query(L , R , lson);  
68.     if (m < R) ret += query(L , R , rson);  
69.     return ret;  
70. }  
71. int main()  
72. {  
73.     int n , m;int a , b , c;  
74.     char str[5];  
75.     scanf("%d%d",&n,&m);  
76.     build(1 , n , 1);  
77.     while (m--)  
78.     {  
79.   
80.   
81.         scanf("%s",str);  
82.         if (str[0] == 'Q')  
83.         {  
84.             scanf("%d%d",&a,&b);  
85.             printf("%I64d\n",query(a , b , 1 , n , 1));  
86.         }  
87.         else if(str[0]=='C')  
88.         {  
89.             scanf("%d%d%d",&a,&b,&c);  
90.             update(a , b , c , 1 , n , 1);  
91.         }  
92.     }  
93.     return 0;  
94. }