校门外有很多树

校门外有很多树

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