主席树WA40分求助
查看原帖
主席树WA40分求助
104963
Gary88楼主2020/7/31 15:35
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
int m,n,tot,cnt;
int a[200001],root[200001],aa[200001],ls[4000001],rs[4000001],t[4000001];
struct popo
{
	int v,no;
}b[100001];
bool sss(popo aa,popo bb)
{
	return aa.v<bb.v;
}
void build(int &rt,int l,int r)
{
	rt=++tot;
	if(l==r)
	{
		return;
	}
	int mid=(l+r)>>1;
	build(ls[rt],l,mid);
	build(rs[rt],mid+1,r);
}
void add(int &rt,int bef,int l,int r,int x)
{
	rt=++tot;
	t[rt]=t[bef]+1;
	if(l==r)
	{
		return;
	}
	int mid=(l+r)>>1;
	if(x<=mid)
	{
		rs[rt]=rs[bef];
		add(ls[rt],ls[bef],l,mid,x);
	}
	else
	{
		ls[rt]=ls[bef];
		add(rs[rt],rs[bef],mid+1,r,x);
	}
}
int query(int rt1,int rt2,int l,int r,int k)
{
	if(l==r)
	{
		return aa[l];
	}
	int x=t[ls[rt2]]-t[ls[rt1]];
	int mid=(l+r)>>1;
	if(x<k)
	{
		return query(rs[rt1],rs[rt2],mid+1,r,k-x);
	}
	else
	{
		return query(ls[rt1],ls[rt2],l,mid,k);
	}
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		b[i].v=a[i];
		b[i].no=i;
	}
	sort(b+1,b+n+1,sss);
	for(int i=1;i<=n;i++)
	{
		if(b[i].v>b[i-1].v)
		a[b[i].no]=++cnt;
		else
		a[b[i].no]=cnt;
		aa[cnt]=b[i].v;
	}
	build(root[0],1,cnt);
	for(int i=1;i<=n;i++)
	{
		add(root[i],root[i-1],1,cnt,a[i]);
	}
	for(int i=1;i<=m;i++)
	{
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		printf("%d\n",query(root[x-1],root[y],1,cnt,z));
	}
	return 0;
}
2020/7/31 15:35
加载中...