具体实现
当然最开始学的时候理解比较困难,哎,难就是难在遗忘,我前不久学的线段树,现在又忘了。
class Segment_Tree {//维护价格区间,实现菜品价格的筛选
private:
// int l, r, w, f;//f为懒惰标记, l, r为区间左右端点, w为权值
struct Tree {
int l, r, w, f;//f为懒惰标记, l, r为区间左右端点, w为权值
}tree[4 * maxn];
int ans;
public:
Segment_Tree();
~Segment_Tree();
void build(int k, int l, int r, int ww);//建树
void down(int k);//懒惰标记下传
void ask_point(int k, int x);//单点查询
void change_point(int k, int x);//单点修改
void ask_interval(int k, int a, int b);//区间查询
void change_interval(int k, int a, int b, int x);//区间修改
};
Segment_Tree::Segment_Tree() {//构造函数
for (int i = 0; i < 4 * maxn; i++) {
tree[i].l = 0;
tree[i].r = 0;
tree[i].f = 0;
tree[i].w = 0;
}
ans = 0;
}
Segment_Tree::~Segment_Tree() {//析构函数
}
void Segment_Tree::build(int k, int ll, int rr, int ww) {//建树
tree[k].l = ll;
tree[k].r = rr;
if (tree[k].l == tree[k].r) {//到了叶子节点
tree[k].w = ww;
return;
}
int mid = (ll + rr) >> 1;
build(2*k, ll, mid, ww);//左子树
build(2*k + 1, mid + 1, rr, ww);//右子树
tree[k].w = tree[2*k].w + tree[2*k + 1].w;//合并权值
}
void Segment_Tree::down(int k) {//懒惰标记向下传递
tree[2*k].f += tree[k].f;
tree[2*k + 1].f += tree[k].f;
tree[2*k].w += tree[k].f * (tree[2*k].r - tree[2*k].l + 1);
tree[2*k + 1].w += tree[k].f * (tree[2*k + 1].r - tree[2*k + 1].l + 1);
tree[k].f = 0;
}
void Segment_Tree::ask_point(int k, int x) {//单点查询
if (tree[k].l == tree[k].r) {//遍历到叶子节点
ans = tree[k].w;///
return;
}
if (tree[k].f) down(k);
int mid = (tree[k].l + tree[k].r) >> 1;
if (x <= mid) /
ask_point(2*k, x);
else
ask_point(2*k + 1, x);
}
void Segment_Tree::ask_interval(int k,int a, int b) {//区间查询
if (tree[k].l >= a && tree[k].r <= b) {
ans += tree[k].w;///
return;
}
if (tree[k].f) down(k);
int mid = (tree[k].l + tree[k].r) >> 1;
if (a <= mid)
ask_interval(2*k, a, mid);
if (b > mid)
ask_interval(2*k + 1, b, mid);
}
void Segment_Tree::change_point(int k, int x) {//单点修改
if (tree[k].l == tree[k].r) {
tree[k].w += x;/
return;
}
if (tree[k].f) down(k);
int mid = (tree[k].l + tree[k].r) >> 1;
if (x <= mid)///
change_point(2*k, x);
else
change_point(2*k + 1, x);
tree[k].w += tree[2*k].w + tree[2*k + 1].w;
}
void Segment_Tree::change_interval(int k,int a, int b, int x) {//区间修改
if (tree[k].l >= a && tree[k].r <= b) {
tree[k].w += x * (tree[k].r - tree[k].l + 1);
tree[k].f += x;
return;
}
if (tree[k].f) down(k);
int mid = (tree[k].l + tree[k].r) >> 1;
if (a <= mid)
change_interval(2*k, a, mid, x);
if (b > mid)
change_interval(2*k + 1, mid, b, x);
tree[k].w += tree[2*k].w + tree[2*k + 1].w;
}