NKOI 1317 校门外有很多树

校门外有很多树

Time Limit:10000MS  Memory Limit:65536K
Total Submit:96 Accepted:73 
Case Time Limit:1000MS

Description

校门外有很多树，有苹果树，香蕉树，有会扔石头的，有可以吃掉补充体力的…… 
如今学校决定在某个时刻在某一段种上一种树，保证任一时刻不会出现两段相同种类的树，现有两个操作： 
K=1，读入l,r表示在l~r之间种上的一种树 
K=2，读入l,r表示询问l~r之间能见到多少种树 
（0<l,r）

Input

第一行n,m表示道路总长为n，共有m个操作 
接下来m行为m个操作 

Output

对于每个k=2输出一个答案

Sample Input

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

Sample Output

1
2

Hint

20%的数据保证，n,m<=100 
60%的数据保证，n <=1000,m<=50000 
100%的数据保证，n,m<=50000

Source

vijos

tree[r].v=tree[tree[r].left].v+tree[tree[r].right].v-左右儿子重复的个数

因此加一个参数ex记录当前区间左右儿子重复的个数

如果涂的区间与讨论的区间的左右儿子都相交，那么tree[r].ex++

#include<cstdio>
#include<iostream>
using namespace std;
const int maxn=50005;
int n,m,tot,a,b,k;
struct wk{int a,b,left,right,v,lazy,ex;}tree[2*maxn];
inline void _read(int &x){
	char t=getchar();bool sign=true;
	while(t<'0'||t>'9')
	{if(t=='-')sign=false;t=getchar();}
	for(x=0;t>='0'&&t<='9';t=getchar())x=x*10+t-'0';
	if(!sign)x=-x;
}
void buildtree(int x,int y){
	int r=++tot;
	tree[r].a=x;tree[r].b=y;
	if(x<y){
		int mid=(x+y)>>1;
		tree[r].left=tot+1;
		buildtree(x,mid);
		tree[r].right=tot+1;
		buildtree(mid+1,y);
	}
}
void putdown(int r){
	tree[r].ex+=tree[r].lazy;
	tree[tree[r].left].lazy+=tree[r].lazy;
	tree[tree[r].right].lazy+=tree[r].lazy;
	tree[tree[r].left].v+=tree[r].lazy;
	tree[tree[r].right].v+=tree[r].lazy;
	tree[r].lazy=0;
}
void add(int r){
	if(tree[r].a>b||tree[r].b<a)return;
	tree[r].v++;
	if(tree[r].a>=a&&tree[r].b<=b){
		tree[r].lazy++;
		return;
	}
	if(tree[r].lazy)putdown(r);
	int mid=(tree[r].a+tree[r].b)>>1;
	if(a<=mid)add(tree[r].left);
	if(b>mid)add(tree[r].right);
	if(a<=mid&&b>mid)tree[r].ex++;
}
int getnum(int r){
	if(tree[r].a>b||tree[r].b<a)return 0;
	if(tree[r].a>=a&&tree[r].b<=b)return tree[r].v;
	if(tree[r].lazy)putdown(r);
	int mid=(tree[r].a+tree[r].b)>>1;
	int sum=0;
	if(a<=mid)sum+=getnum(tree[r].left);
	if(b>mid)sum+=getnum(tree[r].right);
	if(a<=mid&&b>mid)sum-=tree[r].ex;
	return sum;
}
int main(){
	_read(n);_read(m);
	buildtree(1,n);
	for(int i=1;i<=m;i++){
		_read(k);_read(a);_read(b);
		if(k==1)add(1);
		else printf("%d\n",getnum(1));
	}
}