【wikioi】1282 约瑟夫问题

题目链接：http://www.wikioi.com/problem/1282/

算法：线段树（名次树）

说明在代码里有了，直接上代码。

#include <cstdio>
using namespace std;

#define lson l, m, rt << 1
#define rson m+1, r, rt << 1 | 1
#define MID (l+r) >> 1
#define lc rt << 1
#define rc rt << 1 | 1

const int maxn = 30000 + 10;
int n, m, sum[maxn << 2];

void pushup(int rt) {
	sum[rt] = sum[lc] + sum[rc];
}

void build(int l, int r, int rt) {
	if(l == r) {
		sum[rt] = 1;
		return;
	}
	int m = MID;
	build(lson); build(rson);
	pushup(rt);
}

//名次树是以名次为区间，通过对区间内的人数加减，再用相对位置与人数相比可得到他的编号所属区间，坐后递归直到得到了编号。
//维护的sum就是区间和啦~
void update(int p, int l, int r, int rt) {
	sum[rt]--;
    if(l == r) {
        printf("%d ", l);
        return;
    }
	int m = MID;
	if(p <= sum[lc]) update(p, lson); //相对位置小于左边人数说明这个人的编号在左边
	else update(p-sum[lc], rson); //反之在右边，并且人数要减去左边人数
    pushup(rt);
}

int main(){
	scanf("%d%d", &n, &m);
    build(1, n, 1);
    int seq = 1; //从1开始报数
	for(int i = 1; i <= n; ++i) {
		seq = (seq + m - 1) % sum[1]; //至于为什么要-1，自己琢磨。。
		if(!seq) seq = sum[1];
		update(seq, 1, n, 1);
	}
	return 0;
}