我自己的小小线段树模板

#include<iostream>
#include<cstdio>
using namespace std;
#define maxn 100005
#define mid m=(l+r)>>1
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
int c[maxn << 2];
int mark[maxn << 2];
void pushup(int rt)
{
	c[rt] = c[rt << 1] + c[rt << 1 | 1];
}
void pushdown(int rt, int m)
{
	if (mark[rt])
	{
		mark[rt << 1] += mark[rt];
		mark[rt << 1 | 1] += mark[rt];
		c[rt << 1] += (m - (m >> 1))*mark[rt];
		c[rt << 1 | 1] += (m >> 1)*mark[rt];
		mark[rt] = 0;
	}
}
void build(int l, int r, int rt)
{
	mark[rt] = 0;
	if (l == r)
	{
		scanf("%intd", &c[rt]);
		return;
	}
	int mid;
	build(lson);
	build(rson);
	pushup(rt);
}
void see(int l, int r, int rt)
{
	cout << l << " " << r << " " << c[rt] << " " << mark[rt] << endl;
	if (l == r)
	{
		return;
	}
	int mid;
	see(lson);
	see(rson);
}
void updata(int L, int R, int d, int l, int r, int rt)
{
	if (L <= l && r <= R)
	{
		c[rt] += (r - l + 1)*d;
		mark[rt] += d;
		return;
	}
	pushdown(rt, r - l + 1);
	int mid;
	if (L <= m)
	{
		updata(L, R, d, lson);
	}
	if (R > m)
	{
		updata(L, R, d, rson);
	}
	pushup(rt);
}
int query(int L, int R, int l, int r, int rt)
{
	if (L <= l&&r <= R)
	{
		return c[rt];
	}
	int ans = 0, mid;
	pushdown(rt, r - l + 1);
	if (L <= m)
	{
		ans += query(L, R, lson);
	}
	if (R > m)
	{
		ans += query(L, R, rson);
	}
	return ans;
}
int main()
{
	int n, m;
	scanf("%intd%intd", &n, &m);
	build(1, n, 1);
	while (m--)
	{
		char q;
		cin >> q;
		if (q == 'Q')
		{
			int a, b;
			scanf("%d%d", &a, &b);
			cout << query(a, b, 1, n, 1) << endl;
		}
		else
		{
			int a, b, c;
			scanf("%d%d%d", &a, &b, &c);
			updata(a, b, c, 1, n, 1);
			//see(1,n,1);
		}
	}
}

 

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值