NJUST 黑黑更健康

题意是模拟约瑟夫环的问题.

有n个人 (1 <= n <= 100000)排成一行队伍(头尾相连)

每次都从队里第一个人开始数t下,第t个人出队,t是一个随机数.

求每次出队人的编号.

普通的链表模拟肯定是超时的,但是可以用线段树nlogn的复杂度来做.

线段树结点[x, y]存的绝对编号x - y中还在队列中的人数.初始化每个人都在队里.

然后每次计算要出队的相对位置 pos = t % segs[1](当前队伍认输)(pos == 0 -> pos = segs[1])

然后去查询线段树中还在队里的第pos个人的编号,顺便更新线段树(路径上的每个结点都减1)

做题到现在发现规模达到10W以上就应该考虑线段树了.

#include <vector>
#include <cstdio>
#include <iostream>
using namespace std;
#define lson l, m, rt << 1
#define rson m + 1, r, rt << 1 | 1
const int maxn = 100100;
int segs[maxn << 2];
int n;
inline void pushUp(int rt){
	segs[rt] = segs[rt << 1] + segs[rt << 1 | 1];
}
void build(int l, int r, int rt){
	if(l == r){
		segs[rt] = 1;
		return;
	}
	int m = (l + r) >> 1;
	build(lson), build(rson);
	pushUp(rt);
}
int update(int p, int l ,int r, int rt){
	segs[rt]--;
	if(l == r){
		return l;
	}
	int m = (l + r) >> 1;
	if(p <= segs[rt << 1])return update(p, lson);
	else return update(p - segs[rt << 1], rson);//左边的还在队里的人数不够,找右边的第（p - 左边还在队里的人数)个人
}
int main(){
	while (~scanf("%d", &n)){
		build(1, n, 1);
		int pos = 0;
		for (int i = 1; i <= n; ++i){
			int t;
			scanf("%d", &t);
			pos = t % segs[1]; //取得当前要出队的相对位置
			if(pos == 0) pos = segs[1];
			pos = update(pos, 1, n, 1);
			printf("%d\n", pos);
		}
	}
}