洛谷P3960 列队【noip2017D2T3】（多颗线段树求第k大）

题目描述

Sylvia 是一个热爱学习的女♂孩子。

前段时间，Sylvia 参加了学校的军训。众所周知，军训的时候需要站方阵。

Sylvia 所在的方阵中有n \times mn×m 名学生，方阵的行数为 nn ，列数为 mm 。

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

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

在有学生离队后，队伍中出现了一个空位。为了队伍的整齐，教官会依次下达 这样的两条指令：

向左看齐。这时第一列保持不动，所有学生向左填补空缺。不难发现在这条 指令之后，空位在第 xx 行第 mm 列。

向前看齐。这时第一行保持不动，所有学生向前填补空缺。不难发现在这条 指令之后，空位在第 nn 行第 mm 列。
教官规定不能有两个或更多学生同时离队。即在前一个离队的学生归队之后， 下一个学生才能离队。因此在每一个离队的学生要归队时，队伍中有且仅有第 nn 行 第 mm 列一个空位，这时这个学生会自然地填补到这个位置。

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

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

输入输出格式

输入格式：

输入共 q+1q+1 行。

第 1 行包含 3 个用空格分隔的正整数 n, m, qn,m,q ，表示方阵大小是 nn 行 mm 列，一共发 生了 qq 次事件。

接下来 qq 行按照事件发生顺序描述了 qq 件事件。每一行是两个整数 x, yx,y ，用一个空 格分隔，表示这个离队事件中离队的学生当时排在第 xx 行第 yy 列。

输出格式：

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

————————————————————————————————————

题解

---

1.
前n棵是维护每一行的第1列到第m-1列的
第n+1棵是维护第m列的。
关键在于：300000*300000咋整？
只能用动态开点来解决。
2.
y！=m时
把第x行第y个的值找到记为ans并删除
把第m列第x个找到并删除，插入第x行最后一个
把ans插入第m列最后一个那里
y==m直接在x=m那一列操作即可

---

注意

temp2判断是不是原始队列的人（包括初始状态和被迫移动的人）
如果是，就可以用目测（正方形内求每一个位置的坐标）法来获得编号，所以初始状态下c的值不需要
至于主动离队的人，就需要维护c值
离开的人的位置实际上不从线段树中去掉，只是标记为没人（size=1），用处就是判断第k大时去掉这些位置，
阅读代码时容易把l所在的位置当做实际长（或宽），其实去掉size找到的第k大才是当下实际所在坐标(x,y)

---

代码

#include<iostream>
#include<cstring> 
#include<cstdio>
#include<cstdlib> 
using namespace std;
#define ll long long
struct position
{
    int x,y;
}a[310000];
struct segmenttree
{
    int lc,rc;ll c;
}seg[6100000];
int tail[310000],root[310000];
int tot=0,size[6100000];
void insert(int &now,int l,int r,int k,ll num)
{
    if(now==0) now=++tot;
    if(l==r)
    {
        seg[now].c=num;
        size[now]=0;
        return ;
    }
    int mid=(l+r)>>1;
    if(k<=mid) insert(seg[now].lc,l,mid,k,num);
    else       insert(seg[now].rc,mid+1,r,k,num);
    size[now]=size[seg[now].lc]+size[seg[now].rc];
}
ll findkth(int &now,int l,int r,int k,int &ans)
{
    if(now==0) now=++tot;
    if(l==r)
    {
        ans=l;
        ll temp=seg[now].c;seg[now].c=0;
        size[now]=1;
        return temp;
    }
    int mid=(l+r)>>1;ll temp;
    if(mid-l+1-size[seg[now].lc]>=k)
    temp=findkth(seg[now].lc,l,mid,k,ans);
    else 
    temp=findkth(seg[now].rc,mid+1,r,k-(mid-l+1-size[seg[now].lc]),ans);
    size[now]=size[seg[now].lc]+size[seg[now].rc];
    return temp;
}
int main()
{
    ll n,m,q;
    scanf("%lld%lld%lld",&n,&m,&q);
    for(int i=1;i<=n;i++)tail[i]=m-1;
    tail[n+1]=n;
    for(int i=1;i<=q;i++)
    {
        scanf("%d%d",&a[i].x,&a[i].y);
        if(a[i].y==m)
        {
            int temp2;
            ll temp=findkth(root[n+1],1,n+q,a[i].x,temp2);
            if(temp2<=n)temp=temp2*m;
            printf("%lld\n",temp);
            insert(root[n+1],1,n+q,++tail[n+1],temp);
        }
        else 
        {
            int temp2;
            ll temp=findkth(root[a[i].x],1,m+q,a[i].y,temp2);
            if(temp2<m)temp=(a[i].x-1)*m+temp2;
            printf("%lld\n",temp);
            insert(root[n+1],1,n+q,++tail[n+1],temp);
            temp=findkth(root[n+1],1,n+q,a[i].x,temp2);
            if(temp2<=n)temp=temp2*m;
            insert(root[a[i].x],1,m+q,++tail[a[i].x],temp);
        }
    }
//system("pause");
return 0;
}