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

*/
posted @ 2018-10-12 08:39  Pickupwin  阅读(92)  评论(0编辑  收藏  举报