【NOIP2017提高组正式赛】D2T3列队

Description

   Sylvia 是一个热爱学习的女孩子。   前段时间，Sylvia 参加了学校的军训。众所周知，军训的时候需要站方阵。 Sylvia所在的方阵中有n × m名学生，方阵的行数为 n，列数为 m。   为了便于管理，教官在训练开始时，按照从前到后，从左到右的顺序给方阵中从 1 到 n × m 编上了号码（参见后面的样例）。即：初始时，第 i 行第 j 列的学生的编号是(i − 1) × m + j。   然而在练习方阵的时候，经常会有学生因为各种各样的事情需要离队。在一天中，一共发生了 q 件这样的离队事件。每一次离队事件可以用数对(y,z) (1≤x≤n,1≤y≤m)描述，表示第 x 行第 y 列的学生离队。   在有学生离队后，队伍中出现了一个空位。为了队伍的整齐，教官会依次下达这样的两条指令：   1. 向左看齐。这时第一列保持不动，所有学生向左填补空缺。不难发现在这条指令之后，空位在第 x 行第 m 列。   2. 向前看齐。这时第一行保持不动，所有学生向前填补空缺。不难发现在这条指令之后，空位在第 n 行第 m 列。   教官规定不能有两个或更多学生同时离队。即在前一个离队的学生归队之后，下一个学生才能离队。因此在每一个离队的学生要归队时，队伍中有且仅有第 n 行第 m 列一个空位，这时这个学生会自然地填补到这个位置。   因为站方阵真的很无聊，所以 Sylvia 想要计算每一次离队事件中，离队的同学的编号是多少。

数据范围

题解

对于前面6个数据就直接暴力，
7、8、9、10，q都很小可以O(q2)的方法。
对于所有事件x=1这一部分，
修改的只有第一行，还有最后一列会修改，维护一棵线段树就可以了。
每次找第k大，然后在最加一个数。

其实这一部分的解法已经很接近正解了，
正解就是n+1棵线段树。
对于每一行，除去最后一个位置，建n棵线段树，
然后再对最后一列单独建一棵线段树，分别维护。
如果不在最后一列，
就先在对于那一行的线段树里面找到第k大的值，
并将这个点删掉，
再在最后一列里面找出对应的，加入到这一行里面，删掉这个位置。
最后将答案放到最后。

不能将所有点都保存下来，要动态开节点。

code

#include <cstdio>#include <algorithm>#include <cstring>#include <cmath>#include <iostream>#define ll long long#define N 300003#define x_ (x<<1)#define Mid ((l+r)>>1)using namespace std;struct node{    int l,r,s;    ll w;}t[60*N+5];ll ans,g[N*4];int tr[N*8],pos,f[N];int n,m,q,xx,yy,tot,mx;char ch;void read(int& n){    n=0;    ll w=1;    for(ch=getchar();(ch<'0' || ch>'9')&&(ch!='-');ch=getchar());    if(ch=='-')w=-1,ch=getchar();    for(;'0'<=ch && ch<='9';n=(n<<3)+(n<<1)+ch-48,ch=getchar());    n=n*w;}void write(ll x){    if(x>9)write(x/10);    putchar(x%10+48);}void Kth(int x,int l,int r,int s){    if(l==r)    {        ans=t[x].w;        t[x].s=0;        return;    }    if(!t[x].l)    {        t[x].l=++tot;        t[tot].w=t[x].w;        t[tot].s=t[x].s>Mid-l+1?Mid-l+1:t[x].s;        t[x].r=++tot;        t[tot].w=t[x].w+t[t[x].l].s;        t[tot].s=t[x].s-t[t[x].l].s;    }    if(t[t[x].l].s<s)Kth(t[x].r,Mid+1,r,s-t[t[x].l].s);        else Kth(t[x].l,l,Mid,s);    t[x].s=t[t[x].l].s+t[t[x].r].s;} void ins(int x,int l,int r,int z,ll y){    if(l==r)    {        t[x].w=y;        t[x].s=1;        return;    }    if(!t[x].l)    {        t[x].l=++tot;        t[tot].w=t[x].w;        t[tot].s=t[x].s>Mid-l+1?Mid-l+1:t[x].s;        t[x].r=++tot;        t[tot].w=t[x].w+t[t[x].l].s;        t[tot].s=t[x].s-t[t[x].l].s;    }    if(z<=Mid)ins(t[x].l,l,Mid,z,y);        else ins(t[x].r,Mid+1,r,z,y);    t[x].s=t[t[x].l].s+t[t[x].r].s;}int find(int x,int l,int r,int s){    if(l==r)return l;    if(tr[x_]<s)return find(x_+1,Mid+1,r,s-tr[x_]);        else return find(x_,l,Mid,s);}void del(int x,int l,int r,int s){    if(l==r)    {        tr[x]=0;        return;    }    if(Mid<s)del(x_+1,Mid+1,r,s);        else del(x_,l,Mid,s);    tr[x]=tr[x_]+tr[x_+1];}void add(int x,int l,int r,int s){    if(l==r)    {        tr[x]=1;        return;    }    if(Mid<s)add(x_+1,Mid+1,r,s);        else add(x_,l,Mid,s);    tr[x]=tr[x_]+tr[x_+1];}void build(int x,int l,int r){    if(l>n)return;    if(l==r)    {        tr[x]=1;        return;    }    build(x_,l,Mid);    build(x_+1,Mid+1,r);    tr[x]=tr[x_]+tr[x_+1];}int main(){    freopen("phalanx.in","r",stdin);    freopen("phalanx.out","w",stdout);    read(n);read(m);read(q);    mx=(n>m?n:m)+q;    for(ll i=1;i<=n;i++)        t[i].w=(i-1)*m+1,t[i].s=m-1,g[i]=i*m;    tot=n;    build(1,1,mx);    for(int i=1;i<=q;i++)    {        read(xx);read(yy);        pos=find(1,1,mx,xx);        if(yy!=m)f[xx]++,Kth(xx,1,mx,yy),            ins(xx,1,mx,m+f[xx]-1,g[pos]);            else ans=g[pos];        del(1,1,mx,pos);        add(1,1,mx,n+i);        g[n+i]=ans;        write(ans),putchar('\n');    }    return 0;}