openjudge 校门外的树3

P1473 校门外的树3

时间: 1000ms / 空间: 131072KiB / Java类名: Main

描述

 校门外有很多树，有苹果树，香蕉树，有会扔石头的，有可以吃掉补充体力的……
如今学校决定在某个时刻在某一段种上一种树，保证任一时刻不会出现两段相同种类的树，现有两个操作：
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号位置上可以种多种树

此题是一道不错的线段树题，做题前一定要想清楚，如果一个区间的左右子树都含有同一种树，那么对于这个区间绝不是1+1=2，而还应该是1，所以我们需要一个数组去记录左右子树相同树的个数。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
int a[2000000],i,j,n,m,p[2000000],s[2000000];
void pushdown(int now,int l,int r,int mid)
{
if (s[now]!=0)//下放时一定要想明白，既然可以下放则说明他的左右子树有，需要更新p数组
{
s[now<<1]+=s[now];
s[(now<<1)+1]+=s[now];
a[now<<1]+=s[now];
a[(now<<1)+1]+=s[now];
p[now<<1]+=s[now];
        p[(now<<1)+1]+=s[now];
s[now]=0;
}
}
void qchange(int now,int l,int r,int ll,int rr)
{
if (l>=ll&&rr>=r)
{
s[now]+=1;
a[now]++;
p[now]++;
return ;
}
int mid=(l+r)/2;
pushdown(now,l,r,mid);
a[now]++;
int pd1=0,pd2=0;
if (ll<=mid)
{
  qchange(now<<1,l,mid,ll,rr);
  pd1=1;
     }
if (rr>mid)
{
 qchange((now<<1)+1,mid+1,r,ll,rr);
 pd2=1;
     }
if (pd1==1&&pd2==1)
p[now]++;
}
int qsum(int now,int l,int r,int ll,int rr)
{
int mid=(l+r)/2;
pushdown(now,l,r,mid);
if (ll<=l&&rr>=r)
{
return a[now];
}
int ans=0;
int pd1=0,pd2=0;
if (ll<=mid)
ans+=qsum(now<<1,l,mid,ll,rr),pd1=1;
if (rr>mid)
ans+=qsum((now<<1)+1,mid+1,r,ll,rr),pd2=1;
if (pd1==1&&pd2==1)
return ans-p[now];
else
return ans;
}
int main()
{
//freopen("a.in","r",stdin);
scanf("%d%d",&n,&m);
for (i=1;i<=m;i++)
{
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
if (x==1)
{
qchange(1,1,n,y,z);
}
else
{
printf("%d\n",qsum(1,1,n,y,z));
}
}
return 0;
}