FZU 1962 新击鼓传花游戏

新击鼓传花游戏

Time Limit: 1000ms

Memory Limit: 32768KB

This problem will be judged on FZU. Original ID: 1962
64-bit integer IO format: %I64d      Java class name: Main

N个小朋友，编号为1..N，玩新击鼓传花游戏。游戏由老师来主持，老师将下达M条命令，命令包含以下三种，

L k，让将当前游戏中剩余的小朋友中，编号第k小的小朋友离开游戏。

R x，让编号为x的小朋友重新回到游戏中，如果编号为x的小朋友已经在游戏中，那么不做任何变化。

Q k，查询并输出当前剩余小朋友中，编号第k小的小朋友的编号。

 

Input

第一行两个整数N, M（1<=N, M<=500,000），表示小朋友的人数和老师命令的条数。接下来m行，每行按指定格式给出一条命令，输入数据保证k不大于当前剩余的小朋友数量，而1<=x<=N。

 

Output

对测试数据中的查询语句，输出一行，表示当前编号第k小的小朋友编号。

 

Sample Input

5 5
L 1
L 1
Q 1
R 1
Q 1

Sample Output

3
1

Source

Contest for 2010 lecture I

 

解题：线段树或者树状数组

 

 #include <iostream>
 #include <cstdio>
 #include <cstring>
 using namespace std;
 const int maxn = ;
 int tree[maxn<<];
 void pushup(int v){
     tree[v] = tree[v<<] + tree[v<<|];
 }
 void build(int L,int R,int v){
     if(L == R) {
         tree[v] = ;
         return;
     }
     int mid = (L + R)>>;
     build(L,mid,v<<);
     build(mid+,R,v<<|);
     pushup(v);
 }
 void update(int L,int R,int id,int val,int v){
     if(L == R){
         if(tree[v] != val) tree[v] = val;
         return;
     }
     int mid = (L + R)>>;
     if(id <= mid) update(L,mid,id,val,v<<);
     if(id > mid) update(mid+,R,id,val,v<<|);
     pushup(v);
 }
 int query(int L,int R,int p,int v){
     if(L == R) return L;
     int mid = (L + R)>>;
     if(p <= tree[v<<]) return query(L,mid,p,v<<);
     else return query(mid+,R,p-tree[v<<],v<<|);
 }
 int main(){
     int n,m,o;
     char tmp[];
     while(~scanf("%d %d",&n,&m)){
         build(,n,);
         while(m--){
             scanf("%s%d",tmp,&o);
             if(tmp[] == 'L') update(,n,query(,n,o,),,);
             else if(tmp[] == 'R') update(,n,o,,);
             else if(tmp[] == 'Q') printf("%d\n",query(,n,o,));
         }
     }
     return ;
 }