题目描述 Description
有编号从1到N的N个小朋友在玩一种出圈的游戏。开始时N个小朋友围成一圈,编号为I+1的小朋友站在编号为I小朋友左边。编号为1的小朋友站在编号为N的小朋友左边。首先编号为1的小朋友开始报数,接着站在左边的小朋友顺序报数,直到数到某个数字M时就出圈。直到只剩下1个小朋友,则游戏完毕。
现在给定N,M,求N个小朋友的出圈顺序。
输入描述 Input Description
唯一的一行包含两个整数N,M。(1<=N,M<=30000)
输出描述 Output Description
唯一的一行包含N个整数,每两个整数中间用空格隔开,第I个整数表示第I个出圈的小朋友的编号。
样例输入 Sample Input
5 3
样例输出 Sample Output
3 1 5 2 4
思路
没错刚刚写了大暴力水过了这道题,但是作为优秀七中学子,我得钻研。
遥相呼应
那么树状数组操作,标准lowbit、getsum、add,多一个getk来快速查找位置并且输出。
其中查找的时候,注意是整数,所以循环得次数要够!
代码
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<vector>
using namespace std;
const int N=30000+5;
int n,m,w,line[N];
int lowbit(int x)
{
return x&(-x);
}
void add(int x,int y)
{
for (;x<=n;x+=lowbit(x))
line[x]+=y;
}
int getsum(int x)
{
int ret=0;
for (;x;x-=lowbit(x))
ret+=line[x];
return ret;
}
int getk(int k)
{
int b=0;
for (int i=15;i>=0;i--)
if (b+(1<<i)<=n&&line[b+(1<<i)]<k)
{k-=line[b+(1<<i)];b+=1<<i;}
return b+1;
}
int main()
{
scanf("%d%d",&n,&m);
memset(line,0,sizeof(line));
for (int i=1;i<=n;i++)
add(i,1);
for (int i=1;i<=n;i++)
{
if (getsum(n)-getsum(w)>=m)
{
w=getk(getsum(w)+m);
add(w,-1);
printf("%d ",w);
}
else
{
w=getk((m-getsum(n)+getsum(w)-1)%getsum(n)+1);
add(w,-1);
printf("%d ",w);
}
}
printf("\n");
return 0;
}