-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy path715+Range Module_ST_bool.cpp
104 lines (87 loc) · 3.76 KB
/
715+Range Module_ST_bool.cpp
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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
const int maxn = 1e9 + 10;
class Node { // 线段树节点
public:
Node *lNode, *rNode; // 区间的左右端点,即[l,r]
int l, r; // 区间的左右端点,即[l,r]
bool val; // 区间最大值
bool add; // 懒标记
int mid; // 构造的时候就可以捎带计算一下
Node () {}
Node (int _l, int _r) {
this->l = _l;
this->r = _r;
this->mid = _l + _r >> 1;
this->val = false; // 表示是否全部被跟踪
this->add = false;
this->lNode = nullptr;
this->rNode = nullptr;
}
};
class SegTree { // 线段树不一定满二叉树,也不一定是完全二叉树,但一定是平衡二叉树
public:
Node* root;
SegTree() {
root = new Node(0, maxn);
}
void push_up(Node* node) { // 回溯时更新父结点
node->val = node->lNode->val && node->rNode->val; // 递归往回走的时候 一路上去更新, 更新祖先节点
}
void push_down(Node* node) { // 下传懒标记
if (!node->lNode) node->lNode = new Node(node->l, node->mid);
if (!node->rNode) node->rNode = new Node(node->mid + 1, node->r);
if (node->add == 0) return; // 欠的钱不为 0
node->lNode->val = node->val; // +=
node->rNode->val = node->val;
node->lNode->add = node->add;
node->rNode->add = node->add;
node->add = 0;
}
// 区间修改 [l,r] 的和,也可以用于单点修改l=y即可
void modify(Node* node, int l, int r, bool v) { // u为根节点索引 l左边界 y右边界 v为修改的值
if (l <= node->l && node->r <= r) { // 修改区间 [l,r] 完全覆盖当前节点区
node->val = v; // 进来修改, +=
node->add = true; // 懒标记 v +=
return;
}
// 不覆盖裂开
push_down(node); // “下次需要”,下传懒标记
if (l <= node->mid) modify(node->lNode, l, r, v);
if (r > node->mid) modify(node->rNode, l, r, v);
push_up(node); // 向上回溯更新祖先节点
}
// 区间查询 [l,r] 的和,利用拆分与拼凑的思想,把大区间变为多个小区间的和
int query(Node* node, int l, int r) { // u为根节点索引 l左边界 r右边界
if (l <= node->l && node->r <= r) // 查询区间 [l,r] 完全覆盖当前节点区
return node->val; // 间,立即回溯,返回该区间的sum值,让上一层累加
// 非叶子节点,裂开
push_down(node); // “下次需要”,下传懒标记
int res = 1;
if (l <= node->mid) res &= query(node->lNode, l, r); // 查询左边界比mid小,左子节点与区间 [l,r] 有重叠,递归访问左子树
if (r > node->mid) res &= query(node->rNode, l, r); // 查询右边界比mid小,右子节点与区间 [l,r] 有重叠,递归访问右子树
return res;
}
};
class RangeModule {
public:
RangeModule() {
ST = SegTree();
}
void addRange(int left, int right) {
ST.modify(ST.root, left, right - 1, true);
}
bool queryRange(int left, int right) {
return ST.query(ST.root, left, right - 1);
}
void removeRange(int left, int right) {
ST.modify(ST.root, left, right - 1, false);
}
private:
SegTree ST;
};
/**
* Your RangeModule object will be instantiated and called as such:
* RangeModule* obj = new RangeModule();
* obj->addRange(left,right);
* bool param_2 = obj->queryRange(left,right);
* obj->removeRange(left,right);
*/