校门外有很多树
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
相当于求线段相交。
与[a,b]相交线段条数=总的线段条数-起点在[0, a-1]的线段条数-起点在[b+1, n]的线段条数。
#include <stdio.h>
#define maxn 50000
struct node{int a, b, l, r, cnt;}StartTree[maxn*2+9], EndTree[maxn*2+9];
int StartTot=0, EndTot=0;
void MakeStartTree(int x, int y)
{
int now=++StartTot;
StartTree[now].a=x;
StartTree[now].b=y;
StartTree[now].cnt=0;
if(x<y)
{
StartTree[now].l=StartTot+1;
MakeStartTree(x, (x+y)/2);
StartTree[now].r=StartTot+1;
MakeStartTree((x+y)/2+1, y);
}
}
void InsertStart(int now, int k)
{
if(StartTree[now].a<=k&&k<=StartTree[now].b)
{
StartTree[now].cnt++;
InsertStart(StartTree[now].l, k);
InsertStart(StartTree[now].r, k);
}
}
int AskStart(int now, int x, int y)
{
if(x<=StartTree[now].a&&StartTree[now].b<=y) return StartTree[now].cnt;
int lcnt=0, rcnt=0, mid=(StartTree[now].a+StartTree[now].b)/2;
if(x<=mid) lcnt=AskStart(StartTree[now].l, x, y);
if(y>mid) rcnt=AskStart(StartTree[now].r, x, y);
return lcnt+rcnt;
}
void MakeEndTree(int x, int y)
{
int now=++EndTot;
EndTree[now].a=x;
EndTree[now].b=y;
EndTree[now].cnt=0;
if(x<y)
{
EndTree[now].l=EndTot+1;
MakeEndTree(x, (x+y)/2);
EndTree[now].r=EndTot+1;
MakeEndTree((x+y)/2+1, y);
}
}
void InsertEnd(int now, int k)
{
if(EndTree[now].a<=k&&k<=EndTree[now].b)
{
EndTree[now].cnt++;
InsertEnd(EndTree[now].l, k);
InsertEnd(EndTree[now].r, k);
}
}
int AskEnd(int now, int x, int y)
{
if(x<=EndTree[now].a&&EndTree[now].b<=y) return EndTree[now].cnt;
int lcnt=0, rcnt=0, mid=(EndTree[now].a+EndTree[now].b)/2;
if(x<=mid) lcnt=AskEnd(EndTree[now].l, x, y);
if(y>mid) rcnt=AskEnd(EndTree[now].r, x, y);
return lcnt+rcnt;
}
int main()
{
int n, m, tot=0, a, b;
scanf("%d%d", &n, &m);
MakeStartTree(1, n);
MakeEndTree(1, n);
int sel;
while(m--)
{
scanf("%d%d%d", &sel, &a, &b);
if(1==sel)
{
InsertStart(1, a);
InsertEnd(1, b);
tot++;
}
else
{
printf("%d\n", tot-(AskEnd(1, 1, a-1)+AskStart(1, b+1, n)));
}
}
return 0;
}
树状数组
#include <stdio.h>
#define maxn 50000
int lowbit(int x)
{
return x&(-x);
}
int s[maxn+9], e[maxn+9];//BIT:start point and end point
int main()
{
int n, m, k, l, r, tot=0, leftend, rightstart;
scanf("%d%d", &n, &m);
for(int i=1; i<=m; i++)
{
scanf("%d%d%d", &k, &l, &r);
l++;
r++;
if(1==k)
{
tot++;
for(int j=l; j<=n; j+=lowbit(j))s[j]++;
for(int j=r; j<=n; j+=lowbit(j))e[j]++;
}
else
{
leftend=0;
for(int j=l-1; j; j-=lowbit(j))leftend+=e[j];
rightstart=0;
for(int j=n; j; j-=lowbit(j))rightstart+=s[j];
for(int j=r; j; j-=lowbit(j))rightstart-=s[j];
printf("%d\n", tot-leftend-rightstart);
}
}
return 0;
}