P1276 校门外的树（增强版）(线段树)

题目链接：link

题目大意：

校门外马路上本来从编号 0 到 L，每一编号的位置都有一棵树。有砍树者每次从编号 A 到 B 处连续砍掉每一棵树，就连树苗也不放过（记 0 A B，含 A 和 B）；幸运的是还有植树者每次从编号 C 到 D 中凡是空穴（树被砍且还没种上树苗或树苗又被砍掉）的地方都补种上树苗（记 1 C D，含 C 和 D）；问最终校门外留下的树苗多少棵？植树者种上又被砍掉的树苗有多少棵？

解题思路：

对于第一问，留下的树苗数 = 留下的（树与树苗）总数 - 留下的（树）的总数。
对于第二问，种上又被拔掉的树苗数 = 被砍掉的（树与树苗）总数 - 被砍掉的（树）的总数。
因此，我们可以建立两棵线段树，第一棵线段树维护（树和树苗）剩余总数和（树和树苗）砍掉总数，第二棵线段树维护（树）剩余总数。

AC代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int, int> PII;
const int N = 1e4 + 10;
int ans;//树和树苗被砍的总数
struct node
{
	int l, r;
	int sum, lazy; //sum表示当前区间(树) or (树和树苗)的数量，lazy = 0区间不进行操作，lazy = 1区间种树， lazy = -1区间砍树
}tr[2][N << 2];//tr[0]维护第一棵线段树 (树和树苗) ，tr[1]维护第二棵线段树（树）
void pushup(int p, int u) //由子节点信息更新父节点信息
{
	tr[p][u].sum = tr[p][u << 1].sum + tr[p][u << 1 | 1].sum;
}
void pushdown(int p, int u) //懒标记下传，继承
{
	if (tr[p][u].lazy == -1) //区间砍树
	{
		tr[p][u << 1].sum = tr[p][u << 1 | 1].sum = 0;
		tr[p][u << 1].lazy = tr[p][u << 1 | 1].lazy = -1;
	}
	if (tr[p][u].lazy == 1) //区间种树
	{
		tr[p][u << 1].sum = tr[p][u << 1].r - tr[p][u << 1].l + 1;
		tr[p][u << 1 | 1].sum = tr[p][u << 1 | 1].r - tr[p][u << 1 | 1].l + 1;
		tr[p][u << 1].lazy = tr[p][u << 1 | 1].lazy = 1;
	}
	tr[p][u].lazy = 0; //更新懒标记
}
void build(int l, int r, int p, int u = 1) //建树，l,r表示区间，p = 0 第一棵线段树，p = 1 第二棵线段树，u表示节点
{
	tr[p][u] = { l,r,1,0 };
	if (l == r) return;
	int mid = l + r >> 1;
	build(l, mid, p, u << 1); build(mid + 1, r, p, u << 1 | 1);
	pushup(p, u);
}
void cut(int l, int r, int p, int u = 1) //砍树
{
	if (tr[p][u].l >= l && tr[p][u].r <= r)
	{
		if (p == 0) ans += tr[p][u].sum; //记录砍掉的树和树苗的总数
		tr[p][u].sum = 0;
		tr[p][u].lazy = -1;
	}
	else
	{
		pushdown(p, u);
		int mid = tr[p][u].l + tr[p][u].r >> 1;
		if (l <= mid) cut(l, r, p, u << 1);
		if (r > mid) cut(l, r, p, u << 1 | 1);
		pushup(p, u);
	}
}
void plant(int l, int r, int p, int u = 1) //种树
{
	if (tr[p][u].l >= l && tr[p][u].r <= r)
	{
		tr[p][u].sum = tr[p][u].r - tr[p][u].l + 1;
		tr[p][u].lazy = 1;
	}
	else
	{
		pushdown(p, u);
		int mid = tr[p][u].l + tr[p][u].r >> 1;
		if (l <= mid) plant(l, r, p, u << 1);
		if (r > mid) plant(l, r, p, u << 1 | 1);
		pushup(p, u);
	}
}
int main()
{
	int l, m; cin >> l >> m;
	build(0, l, 0); build(0, l, 1); //建树
	while (m--)
	{
		int op, x, y;
		cin >> op >> x >> y;
		if (op == 0) //砍树
		{
			cut(x, y, 0);
			cut(x, y, 1);
		}
		else plant(x, y, 0); //种树
	}
	cout << tr[0][1].sum - tr[1][1].sum << endl; //剩余树和树苗的数量 - 剩余树的数量 = 剩余树苗的数量
	cout << ans - (l + 1 - tr[1][1].sum) << endl; //砍去树和树苗的数量 - 砍去树的数量 = 砍去树苗的数量
	return 0;
}