[洛谷]P3870 [TJOI2009]开关 (#线段树)

题目描述

现有N（2 ≤ N ≤ 100000）盏灯排成一排，从左到右依次编号为：1，2，......，N。然后依次执行M（1 ≤ M ≤ 100000）项操作，操作分为两种：第一种操作指定一个区间[a, b]，然后改变编号在这个区间内的灯的状态（把开着的灯关上，关着的灯打开），第二种操作是指定一个区间[a, b]，要求你输出这个区间内有多少盏灯是打开的。灯在初始时都是关着的。

输入格式

第一行有两个整数N和M，分别表示灯的数目和操作的数目。接下来有M行，每行有三个整数，依次为：c, a, b。其中c表示操作的种类，当c的值为0时，表示是第一种操作。当c的值为1时表示是第二种操作。a和b则分别表示了操作区间的左右边界（1 ≤ a ≤ b ≤ N）。

输出格式

每当遇到第二种操作时，输出一行，包含一个整数：此时在查询的区间中打开的灯的数目。

输入输出样例

输入 #1

4 5
0 1 2
0 2 4
1 2 3
0 2 4
1 1 4

输出 #1

1
2

---

思路

线段树好题，且难度还是比较大的，最起码对于我这种辣鸡难度大。

本题框架:区间修改&区间查询

显然树状数组也可以，分块也可以。

某一区间的长度减去这个区间的亮灯数目就是这个区间在一次修改中所需要打开灯的数目，那么打开灯的数目变了，自然会带动这个区间sum的改变，因此我们需要维护父亲和儿子的sum值。

#include <stdio.h>
#include <iostream>
#define maxn 100001
using namespace std;
int a[maxn],n,m,cnt;
int tree[maxn<<2],tag[maxn<<2];
inline int ln(int p) {return p<<1;}
inline int rn(int p) {return p<<1|1;}
inline void push_up(int node)
{
	tree[node]=tree[ln(node)]+tree[rn(node)];
}
inline void push_down(int root,int l,int r,int mid)
{
	if(tag[root]==0) return;//如果父节点不需要改，儿子节点也不需要 
	tag[ln(root)]^=1,tag[rn(root)]^=1;//取反 
	tree[ln(root)]=(mid-l+1)-tree[ln(root)];//某一区间的长度减去这个区间的亮灯数目就是这个区间在一次修改中所需要打开灯的数目
	tree[rn(root)]=(r-mid)-tree[rn(root)];//某一区间的长度减去这个区间的亮灯数目就是这个区间在一次修改中所需要打开灯的数目
	tag[root]=0;
}
void update(int node,int l,int r,int cl,int cr)
{
	if(cl<=l && r<=cr)
	{
		tree[node]=(r-l+1)-tree[node];
		tag[node]^=1;
		return;
	}
	else
	{
		int mid=(l+r)>>1;
		push_down(node,l,r,mid);
		if(cl<=mid)
		{
			update(ln(node),l,mid,cl,cr);
		}
		if(mid<cr)
		{
			update(rn(node),mid+1,r,cl,cr);
		}
		push_up(node);
	}
}
int query(int node,int l,int r,int cl,int cr)
{
	if(cl<=l && r<=cr)
	{
		return tree[node];
	}
	int mid=(l+r)>>1;
	push_down(node,l,r,mid);
	int s(0);
	if(cl<=mid)
	{
		s=s+query(ln(node),l,mid,cl,cr);
	}
	if(mid<cr)
	{
		s=s+query(rn(node),mid+1,r,cl,cr);
	}
	return s;
}
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	register int i;
	cin>>n>>m;
	while(m--)
	{
		int Case,x,y;
		cin>>Case>>x>>y;
		if(Case==0)
		{
			update(1,1,n,x,y);
		}
		else
		{
			cout<<query(1,1,n,x,y)<<endl;
		}
	}
	return 0;
}