Codevs_P1282 约瑟夫问题(线段树)

时间限制: 1 s
空间限制: 128000 KB
题目等级 : 大师 Master
题目描述 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

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
#define N 500005
int n,m,k,v;
int d[N];
void print(int x){
    int w=0;char ch[10];memset(ch,0,sizeof(ch));
    while(x){
        ch[++w]=x%10+'0';x/=10;
    }
    for(int i=w;i>0;i--) putchar(ch[i]);
    putchar(' ');
    return;
}
void build(int o,int l,int r){
    if(l==r) {d[o]=1;return;}
    int mid=(l+r)>>1;
    build(o*2,l,mid);build(o*2+1,mid+1,r);
    d[o]=r-l+1;
    return;
}
void change(int o,int w,int l,int r){
    d[o]--;
    if(l==r) {print(l);return;}
    int mid=(l+r)>>1;
    if(d[o*2]>=w) change(o*2,w,l,mid);
    else change(o*2+1,w-d[o*2],mid+1,r);
    return;
}
int main(){
    scanf("%d%d",&n,&m);
    build(1,1,n);k=n;
    while(k--){
        v=(v-1+m)%(k+1);
        change(1,v+1,1,n);
    }
    return 0;
}