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的编号是两回事。
两者之间关系为:
假设第 k 的编号为 i , 编号为 i 说明前面还有i - 1 个元素 ,因此要想求这个元素是第几编号,求解中要减去前面的 i -1 个元素,这样才能得到第 k编号。详见代码
#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;
}