【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;

}