HDU 3727 Jewel

题意:

有四种操作

1、在序列最优插入一个数字(该数字从没出现过)

2、询问序列内某区间第k小值

3、询问当前序列内数字x是第几小的(x一定在序列中)

4、询问当前序列内第k小的值


思路:

由于插入操作不改变前面的元素  因此可以离线操作  离线后处理区间第k小值可以使用划分树

操作2、4其实都是区间第k小值的查询  划分树模版即可解决

操作3是操作2的逆过程  可以通过二分+区间第k小操作搞定(慢!)

快的方法是将划分树看成二叉查找树  递归计算(用此方法可提高100+ms)


PS:

划分树是通过kuangbin巨巨的模版学习的  其实划分树就是记录了快排的过程并巧妙的利用toleft数组计数


代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
typedef __int64 LL;
#define N 300010
#define M 20

int tree[M][N],toleft[M][N],sorted[N];
//tree[dep][i] 表示第dep层第i个节点  toleft[dep][i]表示第dep层1~i节点中有几个位于左子树
LL ans[5];
struct ask
{
	int kind,u,v,k;
}ak[N];
int t,m,n,cnt;

void build(int l,int r,int dep) //建树 模拟快排过程
{
	if(l==r) return ;
	int i,mid=(l+r)>>1;
	int y=sorted[mid],same=mid-l+1,lpos=l,rpos=mid+1;
	for(i=l;i<=r;i++) //same计算与sorted[mid]相同的数字有几个会被放在左子树(根节点也算左子树)
	{
		if(tree[dep][i]<y) same--;
	}
	for(i=l;i<=r;i++)
	{
		if(tree[dep][i]<y) tree[dep+1][lpos++]=tree[dep][i];
		else if(tree[dep][i]==y&&same>0)
		{
			tree[dep+1][lpos++]=tree[dep][i];
			same--;
		}
		else tree[dep+1][rpos++]=tree[dep][i];
		toleft[dep][i]=toleft[dep][l-1]+lpos-l;
	}
	build(l,mid,dep+1);
	build(mid+1,r,dep+1);
}

int query(int L,int R,int l,int r,int dep,int k) //[L,R]为大区间 [l,r]为查询区间 查询[l,r]内第k小值
{
	if(l==r) return tree[dep][l];
	int mid=(L+R)>>1,amt=toleft[dep][r]-toleft[dep][l-1]; //amt表示[l,r]区间内分在左子树的节点个数
	if(amt>=k)
	{
		int fl=L+toleft[dep][l-1]-toleft[dep][L-1]; // L + [L,l-1]区间分在左子树的节点个数
		int fr=fl+amt-1;
		return query(L,mid,fl,fr,dep+1,k);
	}
	else
	{
		int fr=r+toleft[dep][R]-toleft[dep][r]; // r + [r+1,R]区间分在左子树的节点个数
		int fl=fr-(r-l-amt);
		return query(mid+1,R,fl,fr,dep+1,k-amt);
	}
}

int findrank(int L,int R,int l,int r,int dep,int x) //查询[l,r]区间内数字x第几小
{
	if(l==r) return 1;
	int mid=(L+R)>>1,amt=toleft[dep][r]-toleft[dep][l-1];
	if(x<=sorted[mid]) //x在左子树
	{
		int fl=L+toleft[dep][l-1]-toleft[dep][L-1];
		int fr=fl+amt-1;
		return findrank(L,mid,fl,fr,dep+1,x);
	}
	else
	{
		int fr=r+toleft[dep][R]-toleft[dep][r];
		int fl=fr-(r-l-amt);
		return amt+findrank(mid+1,R,fl,fr,dep+1,x);
	}
}

int main()
{
	int i,j;
	char str[10];
	while(~scanf("%d",&m))
	{
		cnt=n=0;
		while(m--)
		{
			scanf("%s",str);
			if(str[0]=='I')
			{
				n++;
				scanf("%d",&sorted[n]);
				tree[0][n]=sorted[n];
			}
			else
			{
				if(str[6]=='1')
				{
					ak[cnt].kind=1;
					scanf("%d%d%d",&ak[cnt].u,&ak[cnt].v,&ak[cnt].k);
				}
				else
				{
					ak[cnt].kind=str[6]-'1'+1;
					ak[cnt].u=1;
					ak[cnt].v=n;
					scanf("%d",&ak[cnt].k);
				}
				cnt++;
			}
		}
		sort(sorted+1,sorted+1+n);
		build(1,n,0);
		for(i=0;i<cnt;i++)
		{
			j=ak[i].kind;
			if(j!=2) ans[j]+=query(1,n,ak[i].u,ak[i].v,0,ak[i].k);
			else ans[2]+=findrank(1,n,ak[i].u,ak[i].v,0,ak[i].k);
		}
		printf("Case %d:\n",++t);
		for(i=1;i<=3;i++)
		{
			printf("%I64d\n",ans[i]);
			ans[i]=0;
		}
	}
	return 0;
}


  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值