题目描述
给定一个含有n个数的序列a[1],a[2],a[3]……a[n],程序必须回答这样的询问:对于给定的i,j,k,在a[i],a[i+1],a[i+2]……a[j]中第k小的数是多少(1≤k≤j-i+1),并且,你可以改变一些a[i]的值,改变后,程序还能针对改变后的a继续回答上面的问题。你需要编一个这样的程序,从输入文件中读入序列a,然后读入一系列的指令,包括询问指令和修改指令。
对于每一个询问指令,你必须输出正确的回答。
输入输出格式
输入格式:
第一行有两个正整数n(1≤n≤10000),m(1≤m≤10000)。分别表示序列的长度和指令的个数。
第二行有n个数,表示a[1],a[2]……a[n],这些数都小于10^9。接下来的m行描述每条指令,每行的格式是下面两种格式中的一种。 Q i j k 或者 C i t
Q i j k (i,j,k是数字,1≤i≤j≤n, 1≤k≤j-i+1)表示询问指令,询问a[i],a[i+1]……a[j]中第k小的数。
C i t (1≤i≤n,0≤t≤10^9)表示把a[i]改变成为t。
输出格式:
对于每一次询问,你都需要输出他的答案,每一个输出占单独的一行。
输入输出样例
输入样例#1:
5 3
3 2 1 4 7
Q 1 4 3
C 2 6
Q 2 5 3
输出样例#1:
3
6
说明
20%的数据中,m,n≤100;
40%的数据中,m,n≤1000;
100%的数据中,m,n≤10000。
分析:
表示没弄懂数据结构乱套……
这道题显然不修改可以用主席树。
可以这样想,如果每棵权值线段树记录自己这一个点,显然可以O(logn)修改,O(nlogn)查询(合并l到r的线段树),如果记录前缀和,也就是第i棵线段树记录第1到第i个位置的线段树,显然查询就是T[r]-T[l-1],O(nlogn)修改,O(logn)查询。突然发现如果不管线段树的话,好像就是如何求区间和,这个显然树状数组直接解决,套上线段树就可以AC了。
代码:
#include <iostream>
#include <cstdio>
const int maxn=10005;
const int maxc=1e9+1;
using namespace std;
struct node
{
int l,r,sum;
}t[maxn*500];
int root[maxn],a[maxn],f[maxn],g[maxn];
int i,x,y,z,n,m,cnt,c,k;
char op[1];
int lowbit(int x)
{
return x&(-x);
}
void updata(int l,int r,int &x,int k,int c)
{
if (x==0)
{
cnt++;
x=cnt;
}
t[x].sum+=c;
if (l==r) return;
int mid=(l+r)/2;
if (k<=mid) updata(l,mid,t[x].l,k,c);
else updata(mid+1,r,t[x].r,k,c);
}
int ask(int l,int r,int c)
{
if (l==r) return l;
int mid=(l+r)/2;
int sum=0,j;
for (j=1;j<=k;j++)
{
sum+=t[t[f[j]].l].sum*g[j];
}
if (sum>=c)
{
for (j=1;j<=k;j++)
{
f[j]=t[f[j]].l;
}
ask(l,mid,c);
}
else
{
for (j=1;j<=k;j++)
{
f[j]=t[f[j]].r;
}
ask(mid+1,r,c-sum);
}
}
int main()
{
scanf("%d%d",&n,&m);
for (i=1;i<=n;i++)
{
scanf("%d",&a[i]);
}
for (i=1;i<=n;i++)
{
c=i;
while (c<=n)
{
updata(1,maxc,root[c],a[i],1);
c+=lowbit(c);
}
}
for (i=1;i<=m;i++)
{
scanf("%s",op);
if (op[0]=='C')
{
scanf("%d%d",&x,&y);
c=x;
while (c<=n)
{
updata(1,maxc,root[c],a[x],-1);
updata(1,maxc,root[c],y,1);
c+=lowbit(c);
}
a[x]=y;
}
else
{
scanf("%d%d%d",&x,&y,&z);
k=0;
for (c=y;c>0;c-=lowbit(c))
{
f[++k]=root[c];
g[k]=1;
}
for (c=x-1;c>0;c-=lowbit(c))
{
f[++k]=root[c];
g[k]=-1;
}
printf("%d\n",ask(1,maxc,z));
}
}
}