【NOIP 2017 提高组】列队

传送门

---

problem

前段时间，Sylvia 参加了学校的军训。众所周知，军训的时候需要站方阵。Sylvia 所在的方阵中有 $n\times m$ 名学生，方阵的行数为 $n$，列数为 $m$。

为了便于管理，教官在训练开始时，按照从前到后，从左到右的顺序给方阵中的学生从 $1$ 到 $n\times m$ 编上了号码。初始时，第 $i$ 行第 $j$ 列的学生的编号是 $(i-1)\times m+j$。

然而在练习方阵的时候，经常会有学生因为各种各样的事情需要离队。在一天中，一共发生了 $q$ 件这样的离队事件。每一次离队事件可以用数对 $(x,y)(1\le x\le n,1\le y\le m)$ 描述， 表示第 $x$ 行第 $y$ 列的学生离队。

在有学生离队后，队伍中出现了一个空位。为了队伍的整齐，教官会依次下达这样的两条指令：

• 向左看齐。这时第一列保持不动，所有学生向左填补空缺。不难发现在这条指令之后，空位在第 $x$ 行第 $m$ 列。
• 向前看齐。这时第一行保持不动，所有学生向前填补空缺。不难发现在这条指令之后，空位在第 $n$ 行第 $m$ 列。

教官规定不能有两个或更多学生同时离队。即在前一个离队的学生归队之后，下一个学生才能离队。因此在每一个离队的学生要归队时，队伍中有且仅有第 $n$ 行第 $m$ 列一个空位，这时这个学生会自然地填补到这个位置。

因为站方阵真的很无聊，所以 Sylvia 想要计算每一次离队事件中，离队的同学的编号是多少。

注意：每一个同学的编号不会随着离队事件的发生而改变，在发生离队事件后方阵中同学的编号可能是乱序的。

数据范围：$n,m,q\le 3\times 10^5$。

---

solution

这道题好像并没有那么难。。。

我们对 $n$ 行各自维护一颗线段树，对最后一列单独维护一颗线段树。

考虑将 $(x,y),y\ne m$ 离队，实际上会有两种操作：

1. 找到第 $x$ 行未被删除的第 $y$ 的点，并把它删掉。
2. 把这个点加到最后一列的末尾。
3. 把 $(x+1,m)$ 的点加到第 $x$ 行来。

这里的删除不用真正的删，打一个删除标记即可。

然后加点的话开 $n+1$ 个 vector 分别记录每一行以及最后一列多加的点（$2,3$ 操作）。

如果 $1$ 操作找到的点在 $n\times m$ 的矩阵内，就按照题目的方法算；否则就在 vector 中找。

貌似有点口胡。。。

时间复杂度 $O(n\log n)$。

---

code

#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#define N 300005
#define ll long long
using namespace std;
int n,m,q,Max,pos,tot,root[N];
struct Seg{int num,l,r;}T[N*40];
vector<ll>Q[N];
#define mid ((l+r)>>1)
int Query(int &root,int l,int r,int x){
	if(l==r)  return l;
	int Size=mid-l+1-T[T[root].l].num;
	if(Size>=x)  return Query(T[root].l,l,mid,x);
	return Query(T[root].r,mid+1,r,x-Size);
}
void Modify(int &root,int l,int r,int pos){
	if(!root)  root=++tot;
	T[root].num++;
	if(l==r)  return;
	if(pos<=mid)  Modify(T[root].l,l,mid,pos);
	else  Modify(T[root].r,mid+1,r,pos);
}
ll DelR(int x,ll val){
	pos=Query(root[n+1],1,Max,x);
	Modify(root[n+1],1,Max,pos);
	ll now=(pos<=n)?(ll)pos*m:Q[n+1][pos-n-1];
	Q[n+1].push_back(val?val:now);
	return now;
}
ll DelC(int x,int y){
	pos=Query(root[x],1,Max,y);
	Modify(root[x],1,Max,pos);
	ll now=(pos<m)?(ll)(x-1)*m+pos:Q[x][pos-m];
	Q[x].push_back(DelR(x,now));
	return now;
}
int main(){
	int x,y;
	scanf("%d%d%d",&n,&m,&q),Max=max(m,n)+q;
	while(q--){
		scanf("%d%d",&x,&y);
		printf("%lld\n",(y==m)?DelR(x,0):DelC(x,y));
	}
	return 0;
}