POJ --2528--Mayor's posters--离散线段树

没看题目又想看题解的童鞋请先把题目看了。

题目给的区间是1到一千万，询问却只有一万个，一千万的数组是开不下的，但是询问的区间比如给了一个，1--100的区间，中间2--99都是浪费掉的了不用的区间，这个题要是能发现这点就好办 了，这也应该是这题的难点，最后采取的办法是离散化的线段树，我最先使用的离散方法是使用的STL里面的SET和MAP，但是比较遗憾的是超时了，然后看了别人博客上的题解代码，借用了别人使用数组离散化线段树的办法，成功搞定

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define mmaxn 10000004
#define maxn 10003
#define M(a) memset((a),0,sizeof(a))
#define lc(c) ((c)<<1)
#define rc(c) ((c)<<1|1)
using namespace std;
struct node
{
	int l,r;
	int color;
}p[maxn<<2];
struct Command
{
	int l,r;
	int color;
}com[maxn];

bool  reg[mmaxn],vis[maxn];
int d[maxn<<1],ic,ans;
unsigned short int tal[mmaxn];
int n;

void init()
{
	M(reg);M(d);M(vis);
	ans=ic=0;
	int x, y;
	scanf("%d",&n);		//	下面开始离散区间的一一映射处理
	for(int i=0;i<n;i++)
	{
		scanf("%d%d",&x,&y);
		if(!reg[x])
		{
			d[ic++]=x;
			reg[x]=1;
		}
		if(!reg[y])
		{
			d[ic++]=y;
			reg[y]=1;
		}
		com[i].l=x;
		com[i].r=y;
	}
	sort(d,d+ic);
	for(int i=0;i<ic;i++)
	tal[d[i]]=i+1;
	for(int i=0;i<n;i++)
	{
		com[i].l=tal[com[i].l];
		com[i].r=tal[com[i].r];
		com[i].color=i+1;
	}

	//搞定了，剩下的就是常规的线段树了
}

void build(int c,int x,int y)
{
	p[c].l=x;p[c].r=y;
	p[c].color=0;
	if(x==y)return;
	int m=(x+y)>>1;
	build(lc(c),x,m);
	build(rc(c),m+1,y);
}
void update(int c,int x,int y,int cl)
{
	int l=p[c].l,r=p[c].r;
	if(x==l&&r==y)
	{
		p[c].color=cl;
		return;
	}
	if(p[c].color>=0)
	{
		p[lc(c)].color=p[rc(c)].color=p[c].color;
		p[c].color=-1;
	}
	int m=(l+r)>>1;
	if(m>=y)update(lc(c),x,y,cl);
	else if (m<x)update(rc(c),x,y,cl);
	else
	{
		update(lc(c),x,m,cl);
		update(rc(c),m+1,y,cl);
	}
}
void dfs(int c)
{
	int u=p[c].color;
	if(u==0)return;
	if(p[c].color>0)
	{
		if(!vis[u])
		{
			vis[u]=1;
			ans++;
		}
		return;
	}
	dfs(lc(c));
	dfs(rc(c));
}
void solve()
{
	build(1,1,ic);
	for(int i=0;i<n;i++)
	{
		update(1,com[i].l,com[i].r,com[i].color);
	}
	dfs(1);
	printf("%d\n",ans);
}

int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		init();
		solve();
	}
	return 0;
}