线段树常用模板

CTGU-Yoghurt

已于 2023-07-13 10:09:45 修改

阅读量119

点赞数 1

分类专栏：板子文章标签：数据结构线段树

于 2022-08-14 17:35:48 首次发布

本文链接：https://blog.csdn.net/weixin_60536621/article/details/126333757

版权

板子专栏收录该内容

12 篇文章 0 订阅

订阅专栏

1）.题目操作：查询：某个区间数值的和操作：对单个位置进行操作，对某个区间取模

2）.题目操作：查询：单个区间的值操作：对区间的值进行操作

3）附录：

1.时间复杂度：O (logn)

1）.题目操作：查询：某个区间数值的和操作：对单个位置进行操作，对某个区间取模

struct node{
	int l, r, s ,m;
//构建的树状数组的 s记录区间和，m记录区间最值
}tr[4*N];

void build(int l, int r, int p) {//建树
	tr[p] = { l,r,0 ,0};
	if (l == r) {
		tr[p].s = a[l];
		tr[p].m = a[l];
		return;
	}
	int mid = l + r >> 1;
	build(l, mid, p << 1);
	build(mid+1,r, p << 1|1);
	tr[p].s = tr[p << 1].s + tr[p << 1 | 1].s;
	tr[p].m = max(tr[p << 1].m, tr[p << 1 | 1].m);
}

int IntervalAdd(int l,int r,int p) {//求某个区间和
	if (l <= tr[p].l && r >= tr[p].r) {
		return tr[p].s;
	}
	int s = 0;
	int mid = tr[p].r + tr[p].l >> 1;
	if (mid >= l) s += IntervalAdd(l, r, p << 1);
	if (mid < r) s += IntervalAdd(l, r, p << 1|1);
	return s;
}

void ReplaceX(int p,int x,int k) {//对某个位置进行操作
	if (tr[p].l == tr[p].r) {
		tr[p].s = k;
		tr[p].m = k;
		return;
	}
	int mid = tr[p].l + tr[p].r >> 1;
	if (mid >= x) ReplaceX(p << 1, x, k);
	else ReplaceX(p << 1 | 1, x, k);
	tr[p].s = tr[p<<1].s + tr[p<<1|1].s;
	tr[p].m = max(tr[p << 1].m, tr[p << 1 | 1].m);
}

void IntervaMod(int l,int r,int M,int p) {//对某个区间进行取模
	if (tr[p].m < M) return;
	if (tr[p].l == tr[p].r) {
		tr[p].m %= M;
		tr[p].s %= M;
		return;
	}
	int mid = tr[p].l + tr[p].r >> 1;
	if (mid >= l)IntervaMod(l,r,M,p<<1);
	if(mid<r)IntervaMod(l, r, M, p << 1|1);
	tr[p].s = tr[p << 1].s + tr[p << 1 | 1].s;
	tr[p].m = max(tr[p << 1].m, tr[p << 1 | 1].m);
}

2）.题目操作：查询：单个区间的值操作：对区间的值进行操作

struct node {
	int l, r, v;
//构建的树状数组v为懒标记方便对区间进行加法
}tr[N * 4];

void build(int l, int r, int p) {
	tr[p] = { l,r,0 };
	if (l == r) {
		tr[p].v = a[l];
		return;
	}
	int mid = tr[p].l + tr[p].r >> 1;
	build(l, mid, p << 1);
	build(mid + 1, r, p << 1 | 1);
}

void IntervalAdd(int l, int r, int p, int k) {//对区间进行操作
	if (tr[p].l >= l && tr[p].r <= r) {
		tr[p].v += k;
		return;
	}
	int mid = tr[p].r + tr[p].l >> 1;
	if (mid >= l) IntervalAdd(l, r, p << 1, k);
	if (mid < r) IntervalAdd(l, r, p << 1 | 1, k);
}

void AllAdd(int p, int x) {//求某个位置的值，单点查询时加上记录的区间和
	sum += tr[p].v;
	if (tr[p].l == tr[p].r) {
		return;
	}
	int mid = tr[p].l + tr[p].r>>1;
	if (mid >= x)  AllAdd(p << 1, x);
	else  AllAdd(p << 1 | 1, x);
}