说明:提交一直segmentation fault,没有accepted
题意:
带修改的区间第k大查询;
#include<bits/stdc++.h>
using namespace std;
#define lson l,mid
#define rson mid+1,r
const int maxn=60001;
int hush[maxn],a[50001],ls[maxn<<5],rs[maxn<<5],sum[maxn<<5],root[maxn];
int n,m,tot,tota,totb,totn,A[50],B[50];
struct qu
{
int l,r,k;
} q[10001];
int build(int l,int r)
{
int rt=++tot;
sum[rt]=0;
if(l<r)
{
int mid=l+r>>1;
ls[rt]=build(lson);
rs[rt]=build(rson);
}
return rt;
}
int lowbit(int x)
{
return x&(-x);
}
int fin(int x)
{
return lower_bound(hush+1,hush+1+totn,x)-hush;
}
int update(int last,int p,int val,int l,int r)
{
int rt=++tot;
sum[rt]=sum[last]+val;
ls[rt]=ls[last],rs[rt]=rs[last];
if(l<r)
{
int mid=l+r>>1;
if(p<=mid)
ls[rt]=update(ls[rt],p,val,lson);
else
rs[rt]=update(rs[rt],p,val,rson);
}
return rt;
}
void Add(int x,int p,int val)
{
for(int i=x; i<=n; i+=lowbit(i))
root[i]=update(root[i],p,val,1,totn);
}
int query(int k,int l,int r)
{
if(l==r)return l;
int sum1=0,sum2=0;
for(int i=1; i<=tota; i++)
sum1+=sum[A[i]];
for(int i=1; i<=totb; i++)
sum2+=sum[B[i]];
int tmp=sum2-sum1;
int mid=l+r>>1;
if(tmp>=k)
{
for(int i=1; i<=tota; i++)
A[i]=ls[A[i]];
for(int i=1; i<=totb; i++)
B[i]=ls[B[i]];
return query(k,lson);
}
else
{
for(int i=1; i<=tota; i++)
A[i]=rs[A[i]];
for(int i=1; i<=totb; i++)
B[i]=rs[B[i]];
return query(k-tmp,rson);
}
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
tot=tota=totb=totn=0;
scanf("%d%d",&n,&m);
for(int i=1; i<=n; i++)
{
scanf("%d",&a[i]);
hush[i]=a[i];
}
totn=n;
char s[3];
for(int i=1; i<=m; i++)
{
scanf("%s",s);
if(s[0]=='Q')
scanf("%d%d%d",&q[i].l,&q[i].r,&q[i].k);
else
{
q[i].l=0;
scanf("%d%d",&q[i].r,&q[i].k);
hush[++totn]=q[i].k;
}
}
sort(hush+1,hush+totn+1);
totn=unique(hush+1,hush+1+totn)-hush-1;
root[0]=build(1,totn);
for(int i=1;i<=n;i++)
root[i]=root[0];
for(int i=1; i<=n; i++)
{
Add(i,fin(a[i]),1);
}
for(int i=1; i<=m; i++)
{
if(q[i].l)
{
tota=totb=0;
for(int j=q[i].l-1; j; j-=lowbit(j))
A[++tota]=ls[root[j]];
for(int j=q[i].r; j; j-=lowbit(j))
B[++totb]=ls[root[j]];
printf("%d\n",query(q[i].k,1,totn));
}
else
{
int tp=fin(a[q[i].r]);
Add(q[i].r,tp,-1);
a[q[i].r]=q[i].k;
Add(q[i].r,fin(a[q[i].r]),1);
}
}
}
return 0;
}