【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);}}}