NOIP2017 Day2t3 队列

Description

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 想要计算每一次离队事件中，离队的同学的编号是多少。
注意：每一个同学的编号不会随着离队事件的发生而改变，在发生离队事件后方阵中同学的编号可能是乱序的。

Input

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

Output

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

Sample Input

2 2 3
1 1
2 2
1 2

Sample Output

1
1
4

Solution

显然平衡树是不难的
可以用一个点表示一个等差数列
那么初始时每一行就是一个等差数列，最后一列单独维护
然而我没有用这种方法
由于NOIP题，用的是线段树

同样对于每一行和最后一列开一个线段树，表示每个点是否被删掉过
那么初始时全部为0
每次询问二分出一个位置，当然不能包括已经被标记为1的
这个位置如果还没有超过m，就是它不是后面移上来的，可以直接用公式给出编号
否则就在队列中找到这个点
同时一次询问会使一行和最后一列改变
把最后一列对应的位置也删掉，加到询问所在行的后面，用队列维护
把删掉的元素加到最后一列的后面，也是用队列维护
那么共维护n+1棵线段树和n+1个队列，开个vector，动态开节点之类的
也可以离线，就不详细说了
语文太差，有问题请问

Solution

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
#define N 301000
#define ll long long
using namespace std;
int n,m,q,w,tot;
struct node{
    int l,r,d;
}t[N*100];
vector<ll>a[N];
int find(int v,int i,int j,int x)
{
    if(i==j) return i;
    int m=(i+j)/2;
    if(x<=m-i+1-t[t[v].l].d) return find(t[v].l,i,m,x);
    return find(t[v].r,m+1,j,x-(m-i+1-t[t[v].l].d));
}
void change(int v,int i,int j,int x)
{
    t[v].d++;
    if(i==j) return;
    int m=(i+j)/2;
    if(x<=m) t[v].l=t[v].l==0?++tot:t[v].l,change(t[v].l,i,m,x);
    else t[v].r=t[v].r==0?++tot:t[v].r,change(t[v].r,m+1,j,x);
}
int main()
{
    freopen("phalanx.in","r",stdin);
    freopen("phalanx.out","w",stdout);
    scanf("%d%d%d",&n,&m,&q);
    w=max(n,m)+q+1;
    tot=n+1;
    for(;q;q--)
    {
        int x,y;
        ll ans=0;scanf("%d%d",&x,&y);
        if(y!=m)
        {
            int z=find(x,1,w,y);
            if(z<m) ans=(x-1)*(ll)m+z;
            else ans=a[x][z-m];
            change(x,1,w,z);
            a[n+1].push_back(ans);
            z=find(n+1,1,w,x);
            if(z>n) a[x].push_back(a[n+1][z-n-1]);
            else a[x].push_back((ll)z*(ll)m);
            change(n+1,1,w,z);
        }
        else
        {
            int z=find(n+1,1,w,x);
            if(z>n) ans=a[n+1][z-n-1];
            else ans=(ll)z*(ll)m;
            change(n+1,1,w,z);
            a[n+1].push_back(ans);
        }
        printf("%lld\n",ans);
    }
}