题意:
有四种操作
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;
}