fzu_1962新击鼓传花游戏 线段树_结点更新_元素位置

Problem 1962 新击鼓传花游戏

Accept: 92    Submit: 241
Time Limit: 1000 mSec    Memory Limit : 32768 KB

Problem Description

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

 

此题和zoj 3279 Ants 思路基本一样

注意做此题时，编号为i和第i的编号是两回事。

两者之间关系为：

假设第　ｋ　的编号为 i ， 编号为 i 说明前面还有i - 1 个元素 ，因此要想求这个元素是第几编号，求解中要减去前面的 i -1 个元素，这样才能得到第 ｋ编号。详见代码

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
#define INF 0x3f3f3f3f
#define inf -0x3f3f3f3f
#define mem0(a) memset(a,0,sizeof(a))
const int maxn = 500000+10;
int a[maxn<<2];
void Getsum(int cur){
    a[cur]=a[cur<<1]+a[cur<<1|1];
}
void build(int l,int r ,int cur){
    if(l == r){
        a[cur]=1;
        return ;
    }
    int mid = (l  + r)>>1;
    build(l,mid,cur<<1);
    build(mid+1,r,cur<<1|1);
    Getsum(cur);
}
void leave(int k,int l,int r,int cur){
    if(l == r){
        a[cur] = 0;return ;
    }
    int mid = (l + r )>>1;
    if(k <= a[cur<<1] )
        leave(k,l,mid,cur<<1);
    else{
        k-=a[cur<<1];//第k个离开，求它的编号时要减去前面元素的个数
        leave(k,mid+1,r,cur<<1|1);
    }
    Getsum(cur);
}
void receive(int k,int l,int r,int cur){
    if(l == r){
        a[cur] = 1;return ;
    }
    int mid = (l + r )>>1;
    if(k <= mid )  /*加入编号k，则要直接判断编号k在线段树中的位置
                    直到找到相应的子叶节点*/
        receive(k,l,mid,cur<<1);
    else
        receive(k,mid+1,r,cur<<1|1);
    Getsum(cur);
}
int query(int k,int l,int r,int cur){
    if( l == r)
        return  l;
    int mid = (l + r )>>1;
    if(k <= a[cur<<1])
        return query(k,l,mid,cur<<1);
    else{
        k-=a[cur<<1];  //查询和删除最终目的都是找到第k个的编号
        return query(k,mid+1,r,cur<<1|1);
    }
}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF){
        build(1,n,1);
        while(m--){
            char c[2];
            int k;
            scanf("%s %d",c,&k);
            if(c[0]=='L'){
                leave(k,1,n,1);
            }
            else if(c[0]=='Q'){
                printf("%d\n",query(k,1,n,1));
            }
            else if(c[0]=='R'){
                receive(k,1,n,1);
            }
        }
    }
    return 0;
}