记录编号 478096 评测结果 AAAAAAAAAA
题目名称 [河南省队2012] 找第k小的数 最终得分 100
用户昵称 Gravatarwangxh 是否通过 通过
代码语言 C++ 运行时间 2.471 s
提交时间 2017-12-08 17:04:55 内存使用 233.81 MiB
显示代码纯文本
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
#define lowbit(x) (x&(-x))
struct tree{
	int id,d;
}q[201000];
int ls[20001000],rs[20001000],sum[20001000],root[201000],mova[30],movb[30],pai[201000],numa,numb,cnt,fan[201000];
int n,m,a[201000];
void insert(int,int);
void zhuan(int,int);
void change(int&,int,int,int);
int query(int,int,int);
bool cmp(const tree s1,const tree s2)
{
	if(s1.d==s2.d) return s1.id<s2.id;
	return s1.d<s2.d;
}
int main()
{
//	freopen("in.txt","r",stdin);
	freopen("kth.in","r",stdin);
	freopen("kth.out","w",stdout);
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]),q[i].d=a[i],q[i].id=i;
	sort(q+1,q+n+1,cmp);
	for(int i=1;i<=n;i++) pai[q[i].id]=i,fan[i]=q[i].d;
	for(int i=1;i<=n;i++) insert(i,pai[i]);
	for(int i=1;i<=m;i++)
	{
		int x,y,k;
		scanf("%d%d%d",&x,&y,&k);
		zhuan(x-1,y);
		printf("%d\n",fan[query(1,n,k)]);
	}
	return 0;
}
void insert(int x,int y)
{
	while(x<=n)
	{
		change(root[x],1,n,y);
		x+=lowbit(x);
	}
}
void zhuan(int x,int y)
{
	numa=numb=0;
	while(x)
	{
		numa++;
		mova[numa]=root[x];
		x-=lowbit(x);
	}
	while(y)
	{
		numb++;
		movb[numb]=root[y];
		y-=lowbit(y);
	}
}
void change(int &now,int le,int ri,int x)
{
	int pre=now;now=++cnt;
	ls[now]=ls[pre]; rs[now]=rs[pre]; sum[now]=sum[pre]+1;
	if(le==ri) return;
	int mid=(le+ri)>>1;
	if(x<=mid) change(ls[now],le,mid,x);
	else change(rs[now],mid+1,ri,x);
}
int query(int le,int ri,int k)
{
	if(le==ri) return le;
	int mid=(le+ri)>>1,tmp=0;
	for(int i=1;i<=numa;i++) tmp-=sum[ls[mova[i]]];
	for(int i=1;i<=numb;i++) tmp+=sum[ls[movb[i]]];
//	cout<<tmp<<"-->"<<k<<endl;
	if(tmp>=k)
	{
		for(int i=1;i<=numa;i++) mova[i]=ls[mova[i]];
		for(int i=1;i<=numb;i++) movb[i]=ls[movb[i]];
		return query(le,mid,k);
	}
	else
	{
		for(int i=1;i<=numa;i++) mova[i]=rs[mova[i]];
		for(int i=1;i<=numb;i++) movb[i]=rs[movb[i]];
		return query(mid+1,ri,k-tmp);
	}
}