codevs 1282 约瑟夫问题 树状数组正解 解题报告

题目描述 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;
}