POJ2104 K-th Number【主席树】

题意：求区间第k小

思路：如果让我们求[1,n]上的第k小我们会这么求：建线段树，每个节点记录，这个区间内出现值得个数。找第k小的时候，如果左子树个数大于等于k，进入左子树；反之，进入右子树，此时找的是第（k-左子树的值）小。那么这个任意区间的话，我们给每个前缀都建线段树，求[L,R]的话，[1,R],[1,L-1]这两颗树上点的值的差，就是[L,R]上的点，用之前说到的相同的方法就可以求出第k小。建这么多树，内存要炸，怎么办？假设现在建好了第n棵树，我们建n+1棵树，第n+1个数，要是只更新第n棵树的右子树，那么第n+1棵树的左子树没变，就是第n棵树的左子树，直接指向它就好了，右子树去递归，和第n棵树一样的还是直接指向，不一样的只能新建点了。这题还要离散，要让线段树上的每个点都用上，不然又耗时，又耗内存。

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<string>
#include<stdlib.h>
#include<math.h>
#include<vector>
#include<list>
#include<map>
#include<stack>
#include<queue>
#include<algorithm>
#include<numeric>
#include<functional>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int maxn = 1e5+5;
struct data
{
	int l,r,num;
}node[maxn*20];
int root[maxn],tot;
int a[maxn],pai[maxn];

void init()
{
	tot = 0;
}

int add(int x,int l,int r,int cnt)
{
	node[++tot].num = node[cnt].num + 1;
	int save = tot;
	int mid = (l+r) / 2;
	if(l == r)
		return save;
	else if(x <= mid)
	{
		node[save].l = add(x,l,mid,node[cnt].l);
		node[save].r = node[cnt].r;
	}
	else
	{
		node[save].l = node[cnt].l;
		node[save].r = add(x,mid+1,r,node[cnt].r);
	}
	return save;
}

int query(int x,int y,int l,int r,int k)
{
	if(l == r) return l;
	int num = node[ node[y].l ].num - node[ node[x].l ].num;
	int mid = (l+r) / 2;
	if(k <= num)
		return query(node[x].l,node[y].l,l,mid,k);
	else
		return query(node[x].r,node[y].r,mid+1,r,k - num);
}

int main(void)
{
	int n,m;
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		init();
		for(int i = 1; i <= n; i++)
		{
			scanf("%d",&a[i]);
			pai[i] = a[i];
		}
		sort(pai+1,pai+1+n);
		int newn = unique(pai+1,pai+1+n) - (pai+1);
		for(int i = 1; i <= n; i++)
		{
			int nx = lower_bound(pai+1,pai+1+newn,a[i]) - pai;
			int tp = add(nx,1,newn,root[i-1]);
			root[i] = tp;
		}
		for(int i = 0; i < m; i++)
		{
			int b,c,k;
			scanf("%d%d%d",&b,&c,&k);
			int ans = query(root[b-1],root[c],1,newn,k);
			printf("%d\n",pai[ans]);
		}
	}
	return 0;
}