这道题在洛谷可以找到:https://www.luogu.org/problemnew/show/P3960#sub
题目描述
Sylvia
是一个热爱学习的女孩子。
前段时间,Sylvia
参加了学校的军训。众所周知,军训的时候需要站方阵。
Sylvia 所在的方阵中有n×mn \times mn×m名学生,方阵的行数为 nnn,列数为 mmm。
为了便于管理,教官在训练开始时,按照从前到后,从左到右的顺序给方阵中 的学生从 1 到 n×mn \times mn×m 编上了号码(参见后面的样例)。即:初始时,第 iii 行第 jjj 列 的学生的编号是(i−1)×m+j(i-1)\times m + j(i−1)×m+j。
然而在练习方阵的时候,经常会有学生因为各种各样的事情需要离队。在一天 中,一共发生了 qq q件这样的离队事件。每一次离队事件可以用数对(x,y)(1≤x≤n,1≤y≤m)(x,y) (1 \le x \le n, 1 \le y \le m)(x,y)(1≤x≤n,1≤y≤m)描述,表示第 xxx 行第 yyy 列的学生离队。
在有学生离队后,队伍中出现了一个空位。为了队伍的整齐,教官会依次下达 这样的两条指令:
-
向左看齐。这时第一列保持不动,所有学生向左填补空缺。不难发现在这条 指令之后,空位在第 xxx 行第 mmm 列。
-
向前看齐。这时第一行保持不动,所有学生向前填补空缺。不难发现在这条 指令之后,空位在第 nnn 行第 mmm 列。
教官规定不能有两个或更多学生同时离队。即在前一个离队的学生归队之后, 下一个学生才能离队。因此在每一个离队的学生要归队时,队伍中有且仅有第 nnn 行 第 mmm 列一个空位,这时这个学生会自然地填补到这个位置。
因为站方阵真的很无聊,所以 Sylvia
想要计算每一次离队事件中,离队的同学 的编号是多少。
注意:每一个同学的编号不会随着离队事件的发生而改变,在发生离队事件后 方阵中同学的编号可能是乱序的。
输入输出格式
输入格式:
输入共 q+1q+1q+1 行。
第 1 行包含 3 个用空格分隔的正整数 n,m,qn, m, qn,m,q,表示方阵大小是 nnn 行 mmm 列,一共发 生了 qqq 次事件。
接下来 qqq 行按照事件发生顺序描述了 qqq 件事件。每一行是两个整数 x,yx, yx,y,用一个空 格分隔,表示这个离队事件中离队的学生当时排在第 xxx 行第 yyy 列。
输出格式:
按照事件输入的顺序,每一个事件输出一行一个整数,表示这个离队事件中离队学 生的编号。
输入输出样例
输入样例#1: 复制
2 2 3
1 1
2 2
1 2
输出样例#1: 复制
1
1
4
说明
【输入输出样例 1 说明】
列队的过程如上图所示,每一行描述了一个事件。 在第一个事件中,编号为1 11 的同学离队,这时空位在第一行第一列。接着所有同学 向左标齐,这时编号为 22 2的同学向左移动一步,空位移动到第一行第二列。然后所有同 学向上标齐,这时编号为4 4 4的同学向上一步,这时空位移动到第二行第二列。最后编号 为1 11 的同学返回填补到空位中。
【数据规模与约定】
数据保证每一个事件满足 1≤x≤n,1≤y≤m1 \le x \le n,1 \le y \le m1≤x≤n,1≤y≤m
这个题应该有不止一种做法,这里还是用线段树吧(貌似只会这个...orz)
这里开n+1棵线段树,前n棵维护n行,第n+1棵维护第m列,
分两种情况,当y==m时,找到第n+1棵线段树,将第x个删除,加到第m列最后
当y!=m时,
找到第x棵线段树,将y位置的值删除并插入第n+1棵线段树最后,找到第m列(即第n+1棵线段树)中x的位置,将其删除插入到第x棵线段树最后。
#include<bits/stdc++.h>
#define exp 1e-8
#define mian main
#define pii pair<int,int>
#define pll pair<ll,ll>
#define ll long long
#define pb push_back
#define PI acos(-1.0)
#define inf 0x3f3f3f3f
#define w(x) while(x--)
#define int_max 2147483647
#define lowbit(x) (x)&(-x)
#define gcd(a,b) __gcd(a,b)
#define pq(x) priority_queue<x>
#define ull unsigned long long
#define scn(x) scanf("%d",&x)
#define scl(x) scanf("%lld",&x)
#define pl(a,n) next_permutation(a,a+n)
#define ios ios::sync_with_stdio(false)
#define met(a,x) memset((a),(x),sizeof((a)))
using namespace std;
const int maxn=6e5+10,maxm=2e7+10;
int n,m,q,Max,rt[maxn],pos,tot=0;
vector<ll>v[maxn];
struct tree{
int sum;
int ls,re;
}t[maxm];
void kd(int& u,int l,int r)
{
if(!u)
u=++tot;
t[u].sum++;
if(l<r){
int mid=(l+r)>>1;
if(mid>=pos)
kd(t[u].ls,l,mid);
else
kd(t[u].re,mid+1,r);
}
}
int query(int u,int l,int r,int k)
{
if(l==r)
return l;
int mid=(l+r)>>1;
int num=mid+1-l-t[t[u].ls].sum;
if(num>=k)
return query(t[u].ls,l,mid,k);
else return query(t[u].re,mid+1,r,k-num);
}
ll workl(int x,ll w)
{
pos=query(rt[n+1],1,Max,x);
kd(rt[n+1],1,Max);
ll ans=pos<=n? 1ll*m*pos:v[n+1][pos-n-1];
v[n+1].push_back(w?w:ans);
return ans;
}
ll workr(int x,int y)
{
pos=query(rt[x],1,Max,y);
kd(rt[x],1,Max);
ll ans=pos<m? 1ll*(x-1)*m+pos:v[x][pos-m];
v[x].push_back(workl(x,ans));
return ans;
}
int main()
{
while(~scanf("%d%d%d",&n,&m,&q)){
tot=0;
Max=q+max(n,m);
int x,y;
while(q--){
scanf("%d%d",&x,&y);
if(y==m)
printf("%lld\n",workl(x,0));
else
printf("%lld\n",workr(x,y));
}
}
}