题意是模拟约瑟夫环的问题.
有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);
}
}
}