BZOJ 4415 发牌
线段树
记录当前位于牌堆第几张,不搬牌,发牌即询问第K个可用位置
注意对Size取模即可
#include <cstdio>
const int MAXN=700111;
int N;
int R[MAXN];
int At, Cnt;
struct Node{
int l, r, cnt;
} T[MAXN<<2];
void BuildTree(int l, int r, int at){
T[at].l=l;T[at].r=r;T[at].cnt=(r-l+1);
if(l==r) return;
int m=(l+r)>>1;
BuildTree(l, m, at<<1);
BuildTree(m+1, r, (at<<1)|1);
}
void Update(int at, int p){
--T[at].cnt;
if(T[at].l==T[at].r) return;
int m=(T[at].l+T[at].r)>>1;
if(p<=m) Update(at<<1, p);
else Update((at<<1)|1, p);
}
int Ask(int at, int k){
int m=(T[at].l+T[at].r)>>1;
if(T[at].l==T[at].r) return m;
int ls=T[at<<1].cnt;
if(k<=ls) return Ask(at<<1, k);
return Ask((at<<1)|1, k-ls);
}
int main(){
scanf("%d", &N);
BuildTree(1, N, 1);
At=1;Cnt=N;
for(int i=1, r, v;i<=N;++i){
scanf("%d", &r);
At+=r;At%=Cnt;if(!At) At+=Cnt;
v=Ask(1, At);Update(1, v);
printf("%d\n", v);--Cnt;
}
return 0;
}
/*
4
2
0
3
2
3
4
2
1
*/