【NOIP2017提高组】列队

题目背景

NOIP2017提高组 DAY2 T3

题目描述

Sylvia 是一个热爱学习的女孩子。
前段时间，Sylvia 参加了学校的军训。众所周知，军训的时候需要站方阵。Sylvia 所在的方阵中有 n×m 名学生，方阵的行数为 n，列数为 m 。

为了便于管理，教官在训练开始时，按照从前到后，从左到右的顺序给方阵中的学生从 1 到 n×m 编上了号码（参见后面的样例）。即：初始时，第 i 行第 j 列的学生的编号是 (i-1)×m+j。

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

在有学生离队后，队伍中出现了一个空位。为了队伍的整齐，教官会依次下达这样的两条指令：
1. 向左看齐。这时第一列保持不动，所有学生向左填补空缺。不难发现在这条指令之后，空位在第 x 行第 m 列。
2. 向前看齐。这时第一行保持不动，所有学生向前填补空缺。不难发现在这条指令之后，空位在第 n 行第 m 列。

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

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

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

输入格式

输入共 q+1 行。
第 1 行包含 3 个用空格分隔的正整数 n，m，q，表示方阵大小是 n 行 m 列，一共发生了 q 次事件。
接下来 q 行按照事件发生顺序描述了 q 件事件。每一行是两个整数 x，y，用一个空格分隔，表示这个离队事件中离队的学生当时排在第 x 行第 y 列。

输出格式

按照事件输入的顺序，每一个事件输出一行一个整数，表示这个离队事件中离队学生的编号。

样例数据 1

输入

2 2 3
1 1
2 2
1 2

输出

1
1
4

备注

【输入输出样例1说明】

列队的过程如上图所示，每一行描述了一个事件。

在第一个事件中，编号为1的同学离队，这时空位在第一行第一列。接着所有同学向左标齐，这时编号为 2 的同学向左移动一步，空位移动到第一行第二列。然后所有同学向上标齐，这时编号为 4 的同学向上一步，这时空位移动到第二行第二列。最后编号为 1 的同学返回填补到空位中。

【数据规模与约定】

数据保证每一个事件满足 1≤x≤n；1≤y≤m。

 

解析：

       线段树。

       实现前30分直接模拟。

       然后我们来看n=1的情况，只需要找出第K个数，将它移到队尾即可，于是就可以愉快地 Treap/Splay 啦。

       因为M不大，于是考虑用权值线段树，每次找出第K个没有使用的位置。

       1.如果找到的数小等于M，将它标记为使用，加进 vector 中。

       2.如果找道的数大于M，则实际值即为 vector 里的第 num-M 个数。

       对于100%的数据，我们仍可以使用以上做法，对每一行都建一颗权值线段树。特别的，我们还要对最后一列建一颗权值线段树。对于每个操作，分两种情况：

       1.该位置 (x，y) 中若 y<m，则先修改第 x 行的线段树，然后当加入 vector 时，我们不能直接加入这个数，因为由于最后一列要整体上移，所以，所以要先在最后一列的权值线段树上查询第 x 个位置（x，m）对应的数加入 vector ，然后再把这个删除的数加入最后一列的线段树的末尾。

       2.该位置 (x，y) 中若 y<=m，则直接在最后一列线段树上查询和修改。

       线段树要动态开点，注意开 。

       分析一下时间与空间复杂度：

       首先时间复杂度容易知道是：

       其次空间复杂度，由于是动态开点，所以每次查询Q次，每次更新 logN 个节点，所以大致还是

 

代码：

#pragma GCC optimize(3)
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int Max=300005;
int n,m,q,sum,tot;
int tree[Max*20],lc[Max*20],rc[Max*20],rt[Max];
vector<ll>num[Max];
inline int get_int()
{
	int x=0;char c;
	for(c=getchar();!isdigit(c);c=getchar());
	for(;isdigit(c);c=getchar()) x=(x<<3)+(x<<1)+(c^48);
	return x;
}
inline void print(ll x)
{
	if(x>9) print(x/10);
	putchar('0'+x%10);
}

inline void add(int &now,int l,int r,int pos)
{
	if(!now) now=++tot;tree[now]++;
	if(l==r) return;
	int mid=(l+r)>>1;
	if(pos<=mid) add(lc[now],l,mid,pos);
	else add(rc[now],mid+1,r,pos);
}
inline int Q(int root,int l,int r,int k)
{
	if(l==r) return l;
	int mid=(l+r)>>1,sum=(mid-l+1)-tree[lc[root]];
	if(sum>=k) Q(lc[root],l,mid,k);
	else Q(rc[root],mid+1,r,k-sum);
}
inline ll Qr(int x,ll v)
{
	int pos=Q(rt[n+1],1,sum,x);add(rt[n+1],1,sum,pos);
	ll ans=pos<=n?(ll)m*pos:num[n+1][pos-n-1];
	return num[n+1].push_back(v?v:ans),ans;
}
inline ll Ql(int x,int y)
{
	int pos=Q(rt[x],1,sum,y);add(rt[x],1,sum,pos);
	ll ans=pos<m?(ll)(x-1)*m+pos:num[x][pos-m];
	return num[x].push_back(Qr(x,ans)),ans;
}

int main()
{
  	n=get_int(),m=get_int(),q=get_int(),sum=max(n,m)+q;
	while(q--)
	{
	  int x=get_int(),y=get_int();
	  if(y==m) print(Qr(x,0)),putchar('\n');
	  else print(Ql(x,y)),putchar('\n');
	}
	return 0;
}