【tyvj1473】校门外的树3

描述

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

输入格式

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

输出格式

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

测试样例1

输入

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

输出

1 
2

备注

范围：20%的数据保证，n,m<=100
      60%的数据保证，n <=1000,m<=50000
      100%的数据保证，n,m<=50000
注意：树是可以重叠的，比如1号位置上可以种多种树

【题解】

p和线段树sum同步，表示这个结点的左区间和右区间有多少重合的种数（即有多少种树在根结点的左右子树所表示的区间都有分布）。更新的条件是：①如果区间被完全覆盖，那么一定左右子树都有分布；②如果在进行下一步操作时（l,mid）和（mid+1,r）都做了，那么也证明有分布；

sum线段树表示该区间的种类数，最后查询还是应该求和，但是如果要查询的区间在结点的左右子树都有分布的话，应该减去p[now]

刚开始有一种奇怪的思路，就是在维护的时候根结点不管怎样都加1。这样可以实现维护，但是查询的时候还是有问题。

【代码】

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
int sum[1000000],p[1000000],delta[1000000];
int lrange,rrange,ty,ans,n,m;
inline void update(int now){
	sum[now]=sum[now<<1]+sum[now<<1|1]-p[now];
}
inline void pushdown(int now,int l,int r,int mid){
	if (delta[now]!=0){
		sum[now<<1]+=delta[now];
		delta[now<<1]+=delta[now];
		p[now<<1]+=delta[now];
		sum[now<<1|1]+=delta[now];
		delta[now<<1|1]+=delta[now];
		p[now<<1|1]+=delta[now];
		delta[now]=0;
	}
}
inline void interval_change(int now,int l,int r,int lrange,int rrange,int v){
	int mid=(l+r)>>1;
	if (lrange<=l&&r<=rrange){
		sum[now]+=v;
		delta[now]+=v;
		p[now]++; 
		return;
	}
	pushdown(now,l,r,mid);
	bool pd1=false,pd2=false;
	if (lrange<=mid){
		pd1=true;
		interval_change(now<<1,l,mid,lrange,rrange,v);
	}
	if (mid+1<=rrange){
		pd2=true;
		interval_change(now<<1|1,mid+1,r,lrange,rrange,v);
	}
	if (pd1&&pd2) p[now]++;
	update(now);
}
inline int query(int now,int l,int r,int lrange,int rrange){
	int mid=(l+r)>>1,ans=0;
	if (lrange<=l&&r<=rrange){
		return sum[now];
	}
	pushdown(now,l,r,mid);
	bool pd1=false,pd2=false;
	if (lrange<=mid){
		ans+=query(now<<1,l,mid,lrange,rrange);
		pd1=true;
	}
	if (mid+1<=rrange){
		ans+=query(now<<1|1,mid+1,r,lrange,rrange);
		pd2=true;
	}
	if (pd1&&pd2) return ans-p[now];
	else return ans;
}
int main(){
	scanf("%d%d",&n,&m);
	for (int i=1;i<=m;++i){
		scanf("%d",&ty);
		if (ty==1){
			scanf("%d%d",&lrange,&rrange);
			interval_change(1,1,n,lrange,rrange,1);
		}
		if (ty==2){
			scanf("%d%d",&lrange,&rrange);
			ans=query(1,1,n,lrange,rrange);
			printf("%d\n",ans);
		}
	}
}