线段树

1. // 区间增减，区间查询

2.  

3. #include <cstdio>

4. #include <algorithm>

5. using namespace std;

6.  

7. #define lson l, m, rt<<1

8. #define rson m+1, r, rt<<1|1

9. const int MAXN = 最大n;

10. const int INF = 0x3f3f3f3f;

11.  

12. int SUM[MAXN<<2], MIN[MAXN<<2], MAX[MAXN<<2];

13. int lazy[MAXN<<2];

14.  

15. void PushUp(int rt) {

16.         SUM[rt] = SUM[rt<<1] + SUM[rt<<1|1];

17.         MIN[rt] = min(MIN[rt<<1], MIN[rt<<1|1]);

18.         MAX[rt] = max(MAX[rt<<1], MAX[rt<<1|1]);

19. }

20.  

21. void PushDown(int rt, int m) {

22.         if(lazy[rt]) {

23.                 lazy[rt<<1] += lazy[rt];

24.                 lazy[rt<<1|1] += lazy[rt];

25.                 SUM[rt<<1] += lazy[rt] * (m - (m >> 1));

26.                 SUM[rt<<1|1] += lazy[rt] * (m >> 1);

27.                 MIN[rt<<1] += lazy[rt];

28.                 MIN[rt<<1|1] += lazy[rt];

29.                 MAX[rt<<1] += lazy[rt];

30.                 MAX[rt<<1|1] += lazy[rt];

31.                 lazy[rt] = 0;

32.         }

33. }

34.  

35. // 以下的 l, r, rt 带入 1, n, 1

36. void Build(int l, int r, int rt) {

37.         lazy[rt] = 0;

38.         if(l == r) {

39.                 // 初始化树为 0 的写法

40.                 SUM[rt] = MIN[rt] = MAX[rt] = 0;

41.                 // 边读入边建树的写法

42.                 // scanf("%d", &SUM[rt]);

43.                 // MIN[rt] = MAX[rt] = SUM[rt];

44.                 return;

45.         }

46.         int m = (l + r) >> 1;

47.         Build(lson);

48.         Build(rson);

49.         PushUp(rt);

50. }

51.  

52. // 将区间 L~R 的值增加 v

53. void Update(int L, int R, int v, int l, int r, int rt) {

54.         if(L<=l && r<=R) {

55.                 lazy[rt] += v;

56.                 SUM[rt] += v * (r - l + 1);

57.                 MIN[rt] += v;

58.                 MAX[rt] += v;

59.                 return;

60.         }

61.         PushDown(rt, r-l+1);

62.         int m = (l + r) >> 1;

63.         if(L <= m) Update(L, R, v, lson);

64.         if(R > m) Update(L, R, v, rson);

65.         PushUp(rt);

66. }

67.  

68. // 求区间 L~R 的和

69. int QuerySum(int L, int R, int l, int r, int rt) {

70.         if(L<=l && r<=R) return SUM[rt];

71.         PushDown(rt, r-l+1);

72.         int m = (l + r) >> 1;

73.         int ret = 0;

74.         if(L <= m) ret += QuerySum(L, R, lson);

75.         if(R > m) ret += QuerySum(L, R, rson);

76.  

77.         return ret;

78. }

79.  

80. // 求区间 L~R 的最小值

81. int QueryMin(int L, int R, int l, int r, int rt) {

82.         if(L<=l && r<=R) return MIN[rt];

83.         PushDown(rt, r-l+1);

84.         int m = (l + r) >> 1;

85.         int ret = INF;

86.         if(L <= m) ret = min(ret, QueryMin(L, R, lson));

87.         if(R > m) ret = min(ret, QueryMin(L, R, rson));

88.  

89.         return ret;

90. }

91.  

92. // 求区间 L~R 的最大值

93. int QueryMax(int L, int R, int l, int r, int rt) {

94.         if(L<=l && r<=R) return MAX[rt];

95.         PushDown(rt, r-l+1);

96.         int m = (l + r) >> 1;

97.         int ret = -INF;

98.         if(L <= m) ret = max(ret, QueryMax(L, R, lson));

99.         if(R > m) ret = max(ret, QueryMax(L, R, rson));

100.  

101.         return ret;

102. }