题目大意:
维护一个数列,要求单点修改以及求区间第
k
k
k大。
n
≤
1
0
4
,
m
≤
1
0
4
,
a
i
∈
[
0
,
1
0
9
]
n≤10^4,m≤10^4,a_i\in[0,10^9]
n≤104,m≤104,ai∈[0,109]
分析:
对每一个位置维护一颗权值线段树,然后外面套树状数组即可。
修改直接在数状数组对应位置的线段树修改。查询同主席树。
代码:
/**************************************************************
Problem: 1901
User: ypxrain
Language: C++
Result: Accepted
Time:516 ms
Memory:49428 kb
****************************************************************/
#include <iostream>
#include <cstdio>
#include <cmath>
const int maxn=2e4+7;
const int maxa=1e9;
using namespace std;
int n,m,tot,x,y,k,num;
int a[maxn],root[maxn],p[maxn],ka[maxn];
char op[3];
struct node{
int l,r,data;
}t[maxn*204];
void ins(int &p,int l,int r,int x,int k)
{
if (!p) p=++tot;
t[p].data+=k;
if (l==r) return;
int mid=(l+r)/2;
if (x<=mid) ins(t[p].l,l,mid,x,k);
else ins(t[p].r,mid+1,r,x,k);
}
void updata(int num,int x,int k)
{
for (int i=num;i<=n;i+=i&(-i))
{
ins(root[i],0,maxa,x,k);
}
}
int getans(int l,int r,int k)
{
if (l==r) return l;
int sum=0;
for (int i=1;i<=num;i++) sum+=t[t[p[i]].l].data*ka[i];
int mid=(l+r)/2;
if (sum>=k)
{
for (int i=1;i<=num;i++) p[i]=t[p[i]].l;
getans(l,mid,k);
}
else
{
for (int i=1;i<=num;i++) p[i]=t[p[i]].r;
getans(mid+1,r,k-sum);
}
}
int main()
{
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
updata(i,a[i],1);
}
for (int i=1;i<=m;i++)
{
scanf("%s",op);
if (op[0]=='C')
{
scanf("%d%d",&x,&y);
updata(x,a[x],-1);
a[x]=y;
updata(x,a[x],1);
}
else
{
scanf("%d%d%d",&x,&y,&k);
num=0;
for (int i=x-1;i>0;i-=i&(-i)) p[++num]=root[i],ka[num]=-1;
for (int i=y;i>0;i-=i&(-i)) p[++num]=root[i],ka[num]=1;
printf("%d\n",getans(0,maxa,k));
}
}
}