使用线段树。
对于取走序列中的一个数并放到尾部这样的操作,我们可以使用数组和线段树解决,方法如下:
例:1 7 3 5 7 3 取走5,放到尾部。1 7 3 5 7 3 --> 1 7 3 __ 7 3 5 (下划线表示空)
但这样查询第K个数时不能直接访问数组中的第K个元素(因为有空格),而扫一遍的复杂度又太高,所以要使用线段树进行优化,方法如下:
维护另外一个数组,数组中只有0,1。0表示空格,1表示有元素。
对于上面的例子,数组为 1 1 1 1 1 1 --> 1 1 1 0 1 1 1
这样查询第K个数就是要找到最靠左的位置,使得前缀和=K。
用线段树维护这个数组,查询时在线段树上二分即可。
但对于本题,这样做的空间复杂度为O(n(m+q)),太高。
所以要使用动态开点线段树,方法如下:
因为只有Q个操作,每次操作只修改到logn个节点,所以无需储存线段树的所有节点,只需储存修改过的节点。
对于未修改的节点,都是连续的1,总和可以直接算出,所以无需储存。这样空间复杂度就是qlogn了。使用这种方法维护每行和最后一列即可。
也可以用splay,对于连续的一段,储存在一个节点里,修改时再分裂。
代码:
#include
#define ll long long
int lc[6000010],rc[6000010],sl;
int he[6000010];
ll sz[6000010];
int gen[300010],N,M,q,ss[300010];
void pu(int i,int l,int r)
{
int m=(l+r)>>1;
he[i]=0;
if(lc[i]==-1)
he[i]+=m-l;
else
he[i]+=he[lc[i]];
if(rc[i]==-1)
he[i]+=r-m;
else
he[i]+=he[rc[i]];
}
void add(int i,int l,int r,int j,int x,ll z)
{
if(l+1==r)
{
he[i]+=x;
sz[i]=z;
return;
}
int m=(l+r)>>1;
if(j0?M:N)+q+5,l+k-1,-1,0);
return l+k-1+(ll)(x-1)*M;
}
if(l+1==r)
{
ll jg=sz[i];
add(gen[x],1,(x>0?M:N)+q+5,l,-1,0);
return jg;
}
int s,m=(l+r)>>1;
if(lc[i]==-1)
s=m-l;
else
s=he[lc[i]];
if(k>s)
return find(rc[i],m,r,k-s,x);
else
return find(lc[i],l,m,k,x);
}
int main()
{
scanf("%d%d%d",&N,&M,&q);
lc[0]=rc[0]=-1;
he[0]=N+q+5;
sl=1;
int ls=N+1;
for(int i=1;i<=N;i++)
{
add(0,1,N+q+5,i,0,(ll)i*M);
gen[i]=-1;
ss[i]=M;
}
for(int i=0;i