NOIp2017D2T3 列队（线段树）

 

题目描述

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 列的学生离队。

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

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

2. 向前看齐。这时第一行保持不动，所有学生向前填补空缺。不难发现在这条 指令之后，空位在第 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

【数据规模与约定】

题目分析

我不会平衡树啊QAQ

考虑对每一行都构建一颗线段树。由于空间限制只能考虑动态开点。

同时对每次回到队尾的点再开一颗线段树（n+1）

查询操作：

1、y!=m

若当前查询的点还没有在最后一列，就直接输出（x-1）*m+pos

若在最后一列，则输出第x个线段树的pos-m

最后将这个

2、y==m 基本同上

删除操作：

在线段树的结构体中部在开一个sum记录这一行的删除操作

则此时的区间个数为区间大小-sum

这样的话若要查的数<=区间大小-sum 的话就去左儿子

否则先将其减去（区间大小-sum）再在右儿子去查询

代码

#include<iostream>
#include<string>
#include<string.h>
#include<stdio.h>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=300100;
struct node{
	int sum,ls,rs;
}s[maxn*20];
int q,maxx,root[maxn],tot=0,n,m;
vector<long long> tree[maxn];
void update(int &x,int l,int r,int size)
{
	if (!x) {tot++;x=tot;}
	s[x].sum++;
	if (l>=r) return;
	int mid=(l+r)>>1;
	if (size<=mid) update(s[x].ls,l,mid,size);
	else update(s[x].rs,mid+1,r,size);
}

int query(int k,int l,int r,int size)
{
	if (l==r) return l;
	int mid=(l+r)>>1,siz=mid-l+1-s[s[k].ls].sum;
	if (siz>=size) return query(s[k].ls,l,mid,size);
	    else return query(s[k].rs,mid+1,r,size-siz);
}

long long work1(int x,long long y)
{
	int pos=query(root[n+1],1,maxx,x);
	update(root[n+1],1,maxx,pos);
	long long ans;
	if (pos<=n) ans=(long long)pos*m; else ans=tree[n+1][pos-n-1];
	long long tmp;
	if (y) tmp=y; else tmp=ans;
	tree[n+1].push_back(tmp);
	return ans; 
}

long long work2(int x,int y)
{
	int pos=query(root[x],1,maxx,y);
	update(root[x],1,maxx,pos);
	long long ans;
	if (pos<m) ans=(long long)(x-1)*m+pos;else ans=tree[x][pos-m];
	tree[x].push_back(work1(x,ans));
	return ans;
}

int main()
{
	scanf("%d%d%d",&n,&m,&q);
	int i;
	maxx=max(n,m)+q;
	for (i=1;i<=q;i++)
	{
		int x,y;
		long long ans;
		scanf("%d%d",&x,&y);
		if (y==m) ans=work1(x,0); else ans=work2(x,y);
		printf("%lld\n",ans);
	}
	return 0;
}