题目背景
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;
}