说在前面:“CDQZ”系列题目数据绝对良心(良苦用心233),提交网址如有需要请私信本蒟蒻。
1009:Challenge 9
查看 提交 统计 提问
总时间限制: 10000ms 单个测试点时间限制: 1000ms 内存限制: 262144kB
描述
给一个长为N的数列,有M次操作,每次操作是以下两种之一:
(1)删除某个位置的数
(2)求数列某位置的值
输入
第一行两个正整数N和M。
第二行N的整数表示这个数列。
接下来M行,每行开头是一个字符,若该字符为’D’,则表示一个删除操作,接下来一个整数x,表示删除当前在位置x的数;若该字符为’Q’,则表示一个询问操作,接下来一个整数x,表示求当前在位置x的数。
输出
对每一个询问操作单独输出一行,表示答案。
样例输入
5 3
1 2 3 4 5
Q 3
D 3
Q 3
样例输出
3
4
提示
1<=N<=10^5,1<=M<=10^5,输入保证合法,且所有整数可用带符号32位整型存储。
题解:用线段树按数组下标维护各个区间内的元素个数,modify和query都要考虑左子树贡献,尤其是modify()。每次query就相当于凑够k个数,输出刚好凑够的那个位置的数组下标对应的元素。
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define root 1,1,n
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
#define pushup(rt) sum[rt]=sum[rt<<1]+sum[rt<<1|1]
using namespace std;
const int maxn=1e5+2;
int n,m;
int a[maxn];
int sum[maxn<<2],val[maxn<<2];
inline int read() {
int x=0,f=1;char c=getchar();
while (c<'0'||c>'9') {if (c=='-') f=-1;c=getchar();}
while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
return x*f;
}
void build(int rt,int l,int r) {
if (l==r) {sum[rt]=1;return ;}
int mid=(l+r)>>1;
build(lson),build(rson);
pushup(rt);
}
void modify(int rt,int l,int r,int siz) {
if (l==r) {--sum[rt];return ;}
int mid=(l+r)>>1;
if (siz<=sum[rt<<1]) modify(lson,siz);
else modify(rson,siz-sum[rt<<1]);
pushup(rt);
}
int query(int rt,int l,int r,int siz) {
if (l==r) return a[l];
int mid=(l+r)>>1;
if (siz<=sum[rt<<1]) return query(lson,siz);
else return query(rson,siz-sum[rt<<1]);
}
int main() {
// freopen("Challenge 9.in","r",stdin);
n=read(),m=read();
for (register int i=1;i<=n;++i) a[i]=read();
build(root);
for (register int i=1;i<=m;++i) {
char ss;
while (!isalpha(ss=getchar()));
if (ss^'D') {
int siz=read();
printf("%d\n",query(root,siz));
}
else {
int siz=read();
modify(root,siz);
}
}
return 0;
}