CDQZ Challenge 9

说在前面:“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
#include
#include
#include
#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;
}

你可能感兴趣的:(OI-数据结构)