线段树基操(入门基本操作)

线段树:

线段树是一种二叉搜索树,与区间树相似,它将一个区间划分成一些单元区间,每个单元区间对应线段树中的一个叶结点。
对于线段树中的每一个非叶子节点[a,b],它的左儿子表示的区间为[a,(a+b)/2],右儿子表示的区间为[(a+b)/2+1,b]。因此线段树是平衡二叉树,最后的子节点数目为N,即整个线段区间的长度。
使用线段树可以快速的查找某一个节点在若干条线段中出现的次数,时间复杂度为O(logN)。而未优化的空间复杂度为2N,因此有时需要离散化让空间压缩。

首先是建树:

#define MAXN 100005*8
int tree[MAXN];
void build(int l, int r, int node) {
	if (l == r) {
		scanf("%d", &tree[node]);
		return;
	}
	int mid = (l + r) >> 1;
	build(l, mid, node << 1);
	build(mid + 1, r, node << 1 | 1);
	tree[node] = max(tree[node << 1], tree[node << 1 | 1]);
}

{                       //在 main 函数中的语句
        build(1, n, 1);
}

单点修改以及区间查询操作:

void update(int p, int add, int l, int r, int node) {
	if (l == r) {
		tree[node] = add;		若当前更新点的左右端点相等即到叶子结点时,直接更新信息并返回
		return;
	}
	int m = (l + r) >> 1;
	if (p <= m) update(p, add, l, m, node << 1);
	else update(p, add, m + 1, r, node << 1 | 1);
	tree[node] = max(tree[node << 1], tree[node << 1 | 1]);	//递归回之后用儿子结点更新父节点(此处是区间最大值)
}

{									 //在 main 函数中的语句
	update(a, b, 1, n, 1);
}
对应单点修改的区间查询:
int query(int L, int R, int l, int r, int node) {
	if (L <= l && r <= R) return tree[node];
	int m = (l + r) >> 1, ret = 0;
	if (L <= m) ret = max(ret, query(L, R, l, m, node << 1));
	if (R > m) ret = max(ret, query(L, R, m + 1, r, node << 1 | 1));
	return ret;					//区间最值
}
{
		 printf("%d\n", query(a, b, 1, n, 1));
}

单点增减值:

void update(int p, int add, int l, int r, int node) {
	if (l == r) {
		tree[node] += add;		//根结点+add
		return;
	}
	int m = (l + r) >> 1;
	if (p <= m) update(p, add, l, m, node << 1);
	else update(p, add, m+1, r, node << 1 | 1);
	tree[node] = tree[node << 1] + tree[node << 1 | 1];
}
{
		update(a, b, 1, n, 1);
}
对应单点增减值的区间查询:
int query(int L, int R, int l, int r, int node) {
	if (L <= l && r <= R) return tree[node];
	int m = (l + r) >> 1, ret=0;
	if (L <= m) ret += query(L, R, l, m, node << 1);
	if (R > m) ret += query(L, R, m+1, r, node << 1 | 1);
	return ret;
}

{
	printf("%d\n", query(a, b, 1, n, 1));
}
线段数的区间修改(增减值)以及相应的查询:
ll tree[MAXN];
ll add[MAXN];
void pushup(int node) {
	tree[node] = tree[node << 1] + tree[node << 1 | 1];
}

void build(int l, int r, int node) {
	if (l == r) {
		scanf("%lld", &tree[node]);
		return;
	}
	int mid = (l + r) >> 1;
	build(l, mid, node << 1);
	build(mid + 1, r, node << 1 | 1);
	pushup(node);
}


void pushdown(int l, int r, int node) {	//pushdown函数,将node结点的信息传递到左右子节点上
	if (add[node]) {	
		add[node << 1] += add[node];		//当父节点有更新信息时才向下传递信息
		add[node << 1 | 1] += add[node];	//左右儿子结点均加上父节点的更新值
		int m = (l + r) >> 1;
		tree[node << 1] += add[node] * (m - l + 1);	//左右儿子结点均按照需要加的值总和更新结点信息
		tree[node << 1 | 1] += add[node] * (r - m);
		add[node] = 0;								 //信息传递完之后就可以将父节点的更新信息删除
	}	
}

void update_c(int a, int b, int addc, int l, int r, int node) {//a、b为需要更新的区间左右端点,addc为需要增加的值
	if (a <= l && b >= r) {							//与单点更新一样,当当前结点被需要更新的区间覆盖时
		add[node] += addc;							//更新该结点的所需更新信息
		tree[node] += addc * (r - l + 1);			//更新该结点信息
		return;	 //根据lazy思想,由于不需要遍历到下层结点,因此不需要继续向下更新,直接返回
	}

	pushdown(l, r, node);						//将当前结点的所需更新信息传递到下一层(其左右儿子结点)
	int m = (l + r) >> 1;
	if (a <= m) update_c(a, b, addc, l, m, node << 1);
	if (b > m ) update_c(a, b, addc, m + 1, r, node << 1 | 1);
	pushup(node);							 //递归回上层时一步一步更新回父节点
}

ll query_c(int a, int b, int l, int r, int node) {	//a, b为需要查询的区间
	if (a <= l && b >= r) return tree[node];
	pushdown(l, r, node);							 //将当前结点的更新信息传递给其左右子节点
	int m = (l + r) >> 1;
	ll ans = 0;
	if (a <= m) ans += query_c(a, b, l, m, node << 1);
	if (b > m) ans += query_c(a, b, m + 1, r, node << 1 | 1);
	return ans;
}
区间改值(其实只有pushdow函数和update中修改部分与区间加值不同):

修改如下:

void pushdown(int l, int r, int node) {	//pushdown函数,将node结点的信息传递到左右子节点上
	if (add[node]) {	
		add[node << 1] = add[node];		//当父节点有更新信息时才向下传递信息
		add[node << 1 | 1] = add[node];	//左右儿子结点均加上父节点的更新值
		int m = (l + r) >> 1;
		tree[node << 1] = add[node] * (m - l + 1);	//左右儿子结点均按照需要加的值总和更新结点信息
		tree[node << 1 | 1] = add[node] * (r - m);
		add[node] = 0;								 //信息传递完之后就可以将父节点的更新信息删除
	}	
}
void update_c(int a, int b, int addc, int l, int r, int node) {//a、b为需要更新的区间左右端点,addc为需要增加的值
	if (a <= l && b >= r) {							//与单点更新一样,当当前结点被需要更新的区间覆盖时
		add[node] = addc;							//更新该结点的所需更新信息
		tree[node] = addc * (r - l + 1);			//更新该结点信息
		return;	 //根据lazy思想,由于不需要遍历到下层结点,因此不需要继续向下更新,直接返回
	}

	pushdown(l, r, node);						//将当前结点的所需更新信息传递到下一层(其左右儿子结点)
	int m = (l + r) >> 1;
	if (a <= m) update_c(a, b, addc, l, m, node << 1);
	if (b > m ) update_c(a, b, addc, m + 1, r, node << 1 | 1);
	pushup(node);							 //递归回上层时一步一步更新回父节点
}

其实就是把 “+=” 的地方全部改成 “=” ;

附上我丑不拉叽的代码:
题目连接:POJ3468——A Simple Problem with Integers

#include<stdio.h>
#include<iostream>
#include<map>
#include<algorithm>
#include<cstring>
#include<string.h>
#include<string>
#include<math.h>
#include<vector>
#include<map>
using namespace std;
typedef long long ll;
#define MAXN 100005*8
#define INF 0x3f3f3f3f//将近int类型最大数的一半,而且乘2不会爆int
#define MOD 1000000007

ll tree[MAXN];
ll add[MAXN];
void pushup(int node) {
	tree[node] = tree[node << 1] + tree[node << 1 | 1];
}

void build(int l, int r, int node) {
	if (l == r) {
		scanf("%lld", &tree[node]);
		return;
	}
	int mid = (l + r) >> 1;
	build(l, mid, node << 1);
	build(mid + 1, r, node << 1 | 1);
	pushup(node);
}


void pushdown(int l, int r, int node) {
	if (add[node]) {
		add[node << 1] += add[node];
		add[node << 1 | 1] += add[node];
		int m = (l + r) >> 1;
		tree[node << 1] += add[node] * (m - l + 1);
		tree[node << 1 | 1] += add[node] * (r - m);
		add[node] = 0;
	}
}

void update_c(int a, int b, int addc, int l, int r, int node) { //a、b为需要更新的区间左右端点,addc为需要增加的值
	if (a <= l && b >= r) {
		add[node] += addc;
		tree[node] += addc * (r - l + 1);
		return;
	}

	pushdown(l, r, node);
	int m = (l + r) >> 1;
	if (a <= m) update_c(a, b, addc, l, m, node << 1);
	if (b > m ) update_c(a, b, addc, m + 1, r, node << 1 | 1);
	pushup(node);
}

ll query_c(int a, int b, int l, int r, int node) {
	if (a <= l && b >= r) return tree[node];
	pushdown(l, r, node);
	int m = (l + r) >> 1;
	ll ans = 0;
	if (a <= m) ans += query_c(a, b, l, m, node << 1);
	if (b > m) ans += query_c(a, b, m + 1, r, node << 1 | 1);
	return ans;
}

//void update(int p, int add, int l, int r, int node) {
//	if (l == r) {
//		tree[node] = add;		//根结点+add
//		return;
//	}
//	int m = (l + r) >> 1;
//	if (p <= m) update(p, add, l, m, node << 1);
//	else update(p, add, m + 1, r, node << 1 | 1);
//	tree[node] = max(tree[node << 1], tree[node << 1 | 1]);
//}
//
//int query(int L, int R, int l, int r, int node) {
//	if (L <= l && r <= R) return tree[node];
//	int m = (l + r) >> 1, ret = 0;
//	if (L <= m) ret = max(ret, query(L, R, l, m, node << 1));
//	if (R > m) ret = max(ret, query(L, R, m + 1, r, node << 1 | 1));
//	return ret;
//}

int main()
{
	int n, m;
	cin >> n >> m;
	build(1, n, 1);
	//getchar();
	for (int i = 0; i < m; ++i) {
		char ch[2];
		int a, b, c;
		scanf("%s", ch);
		if (ch[0] == 'Q') {
			scanf("%d%d", &a, &b);
			printf("%lld\n", query_c(a, b, 1, n, 1));
		}
		else {
			scanf("%d%d%d", &a, &b, &c);
			update_c(a, b, c, 1, n, 1);
		}
	}
	system("pause");
	return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值