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