题目描述
小 FF 最后一道防线是一条长度为 nn 的战壕,小 FF 拥有无数多种地雷,而 SCV 每次可以在 [L, R]区间埋放同一种不同于之前已经埋放的地雷。由于情况已经十万火急,小 FF 在某些时候可能会询问你在 [L,R]区间内有多少种不同的地雷,他希望你能尽快的给予答复。
输入格式
第一行为两个整数n和m,n表示防线长度,m 表示 SCV 布雷次数及小 FF 询问的次数总和。
接下来有m 行,每行三个整数 q,l,r:
若 q=1,则表示 SCV 在 [l, r]这段区间布上一种地雷;
若 q=2,则表示小 FF 询问当前 [l, r] 区间总共有多少种地雷。
第一眼看到就可以发现是线段树,但如何对它进行维护呢?很显然,直接维护是做不到的,因为区间统计时可能会有重复。其实不难看出两个端点是重中之重,我们应该从两个端点下手,怎么操作呢,可以维护两个东西,一个suml(起点),sumr(终点),每次操作就在起点(suml)和终点(sumr)++,让其次查询的时候求出1~r的suml - 1~l的sumr就可以了。
至于为什么这么操作,嘿嘿,我们可以将1~l分解成 l~r , 1~l 区间的suml与sumr可以互相抵消,那么剩下的就是l~r的suml。
#include<bits/stdc++.h>
using namespace std;
int n,m;
int len=0;
int root;
struct node
{
int l,r,lc,rc,suml,sumr;
};
node e[500000];
int bt(int x,int y)
{
len++;
int now=len,lc=-1,rc=-1;
if(x==y) ;
else
{
int mid=(x+y)/2;
lc=bt(x,mid);
rc=bt(mid+1,y);
}
e[now]={x,y,lc,rc,0,0};
return now;
}
void cg1(int now,int x)
{
int l=e[now].l,r=e[now].r;
if(l==r)
{
e[now].suml++;
return ;
}
else
{
int mid=(l+r)/2;
int lc=e[now].lc,rc=e[now].rc;
if(x>=mid+1) cg1(rc,x);
else cg1(lc,x);
e[now].suml=e[lc].suml+e[rc].suml;
return ;
}
}
void cg2(int now,int x)
{
int l=e[now].l,r=e[now].r;
if(l==r)
{
e[now].sumr++;
return ;
}
else
{
int mid=(l+r)/2;
int lc=e[now].lc,rc=e[now].rc;
if(x>=mid+1) cg2(rc,x);
else cg2(lc,x);
e[now].sumr=e[lc].sumr+e[rc].sumr;
return ;
}
}
int find1(int now,int x,int y)
{
int l=e[now].l,r=e[now].r;
if(l==x&&y==r)
{
return e[now].suml;
}
else
{
int lc=e[now].lc,rc=e[now].rc;
int mid=(l+r)/2;
if(x>=mid+1) return find1(rc,x,y);
else if(y<=mid) return find1(lc,x,y);
else return find1(lc,x,mid)+find1(rc,mid+1,y);
}
}
int find2(int now,int x,int y)
{
int l=e[now].l,r=e[now].r;
if(l==x&&y==r)
{
return e[now].sumr;
}
else
{
int lc=e[now].lc,rc=e[now].rc;
int mid=(l+r)/2;
if(x>=mid+1) return find2(rc,x,y);
else if(y<=mid) return find2(lc,x,y);
else return find2(lc,x,mid)+find2(rc,mid+1,y);
}
}
int find(int x,int y)
{
return find1(root,1,y)-find2(root,1,x-1);
}
int main()
{
scanf("%d%d",&n,&m);
root=bt(1,n);
for(int i=1;i<=m;i++)
{
int p,x,y;
scanf("%d",&p);
if(p==1)
{
scanf("%d%d",&x,&y);
// printf("-");
cg1(root,x);
cg2(root,y);
}
else
{
scanf("%d%d",&x,&y);
if(x>y) swap(x,y);
printf("%d\n",find(x,y));
}
}
return 0;
}