BestCoder Round #18 1004 &&hdu 5107 K-short Problem 线段树

题解：

1004 K-short Problem
一个线段树问题，将坐标离散化，将建筑和询问放在一起离线排序，先排x，再排y，再先排建筑。按y轴建线段树，因为询问最多第10矮，只要将每一个坐标按大小顺序存10个节点即可，每读到一个建筑，将其插入当前y坐标。询问则询问[-inf，y]的区间里低k小的高度。

写完心好累……

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<vector>
#include<set>
#include<stack>
#include<queue>
#include<string>
using namespace std;
#define inf 0x3f3f3f
#define ll __int64
#define mem(a,b) memset(a,b,sizeof(a))
#define inf 0x3f3f3f3f
const ll mod=1e9+7;

int n,m,t;
vector<int> tmp,pos;
struct node
{
	int x,y,h,id;
	void init()
	{
		scanf("%d%d%d",&x,&y,&h);
	}
}p[60060];
struct Tree
{
	int l,r,h[10],cnt;
	Tree()
	{
		cnt=0;
	}
	int mid()
	{
		return (l+r)>>1;
	}
	void insert(int x)
	{
		if(cnt<10) h[cnt++]=x;
		else if(x<h[cnt-1]) h[cnt-1]=x;
		sort(h,h+cnt);
	}
}tree[4*60060];

void build(int id,int x,int y)
{
	tree[id].l=x;
	tree[id].r=y;
	tree[id].cnt=0;
	memset(tree[id].h,0,sizeof(tree[id].h));
	if(x==y)
		return ;
	int mid=tree[id].mid();
	build(id<<1,x,mid);
	build(id<<1|1,mid+1,y);
}

void init()
{
	int i,j,k;
	t=n+m;
	tmp.clear();
	pos.clear();
	for(i=0;i<n;i++)
	{
		p[i].init();
		p[i].id=0;
		tmp.push_back(p[i].y);
	}
	for(i=n;i<t;i++)
	{
		p[i].init();
		p[i].id=i+1-n;
		tmp.push_back(p[i].y);
	}
	sort(tmp.begin(),tmp.end());
	pos.push_back(tmp[0]);
	for(i=1;i<tmp.size();i++)
	{
		if(tmp[i]!=tmp[i-1])
			pos.push_back(tmp[i]);
	}
	build(1,1,pos.size());
}
int ans[30030];
int find(int x)
{
	return lower_bound(pos.begin(),pos.end(),x)-pos.begin()+1;
}
Tree merge(Tree a,Tree b)
{
	Tree tt;
	int i=0,j=0;
	while(i<a.cnt && j<b.cnt && tt.cnt<10)
	{
		if(a.h[i]<b.h[j])
		{
			tt.h[tt.cnt++]=a.h[i];
			i++;
		}
		else
		{
			tt.h[tt.cnt++]=b.h[j];
			j++;
		}
	}
	while(tt.cnt<10 && i<a.cnt)
		tt.h[tt.cnt++]=a.h[i++];
	while(tt.cnt<10 && j<b.cnt)
		tt.h[tt.cnt++]=b.h[j++];
	return tt;
}
void add(int id,int x,int y)
{
	if(tree[id].l==x && tree[id].r==x)
	{
		tree[id].insert(y);
		return ;
	}
	int mid=tree[id].mid();
	if(x<=mid)
		add(id<<1,x,y);
	else
		add(id<<1|1,x,y);
	int tx=tree[id].l,ty=tree[id].r;
	tree[id]=merge(tree[id<<1],tree[id<<1|1]);
	tree[id].l=tx,tree[id].r=ty;
}
Tree query(int id,int x,int y)
{
	if(tree[id].l==x && tree[id].r==y)
	{
		return tree[id];
	}
	int mid=tree[id].mid();
	if(y<=mid)
		return query(id<<1,x,y);
	else if(mid+1<=x)
		return query(id<<1|1,x,y);
	else
	{
		return merge(query(id<<1,x,mid),query(id<<1|1,mid+1,y));
	}
}
bool cmp(node a,node b)
{
	if(a.x!=b.x) return a.x<b.x;
	if(a.y!=b.y) return a.y<b.y;
	return a.id<b.id;
}
void solve()
{
	int i,j,k;
	sort(p,p+t,cmp);
	for(i=0;i<t;i++)
	{
		if(p[i].id)
		{
			Tree he=query(1,1,find(p[i].y));
			k=p[i].h;
			if(he.cnt>=k)
				ans[p[i].id]=he.h[k-1];
			else
				ans[p[i].id]=-1;
		}
		else
			add(1,find(p[i].y),p[i].h);
	}
	for(i=1;i<=m;i++)
		printf("%d\n",ans[i]);
}
int main()
{
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		init();
		solve();
	}
}