洛谷 P3960 列队【线段树】

用动态开点线段树分别维护每一行和最后一列，线段树的作用是记录被选的点的个数以及查询第k个没被选的点，每次修改，从行里标记被选的点，从最后一列标记向左看齐之后少的点，然后用vector维护行列的新增点

#include<iostream>
#include<cstdio>
#include<vector>
using namespace std;
const int N=600005;
int n,m,q,mx,tot,rt[N];
vector<long long>a[N];
struct qwe
{
	int ls,rs,s;
}t[10000005];
int read()
{
	int r=0,f=1;
	char p=getchar();
	while(p>'9'||p<'0')
	{
		if(p=='-')
			f=-1;
		p=getchar();
	}
	while(p>='0'&&p<='9')
	{
		r=r*10+p-48;
		p=getchar();
	}
	return r*f;
}
void update(int &ro,int l,int r,int p)
{
	if(!ro)
		ro=++tot;
	t[ro].s++;
	if(l==r)
		return;
	int mid=(l+r)>>1;
	if(p<=mid)
		update(t[ro].ls,l,mid,p);
	else
		update(t[ro].rs,mid+1,r,p);
}
int ques(int ro,int l,int r,int k)
{
	if(l==r)
		return l;
	int mid=(l+r)>>1;
	if(mid-l+1-t[t[ro].ls].s>=k)
		return ques(t[ro].ls,l,mid,k);
	else
		return ques(t[ro].rs,mid+1,r,k-(mid-l+1-t[t[ro].ls].s));
}
int main()
{
	n=read(),m=read(),q=read(),mx=max(m,n)+q;
	while(q--)
	{
		int x=read(),y=read();
		if(y==m)
		{
			int p=ques(rt[n+1],1,mx,x);
			update(rt[n+1],1,mx,p);
			long long ans=p<=n?1ll*p*m:a[n+1][p-n-1];
			a[n+1].push_back(ans);
			printf("%lld
",ans);
		}
		else
		{
			int p=ques(rt[x],1,mx,y);
			update(rt[x],1,mx,p);;
			long long ans=p<m?1ll*(x-1)*m+p:a[x][p-m];
			p=ques(rt[n+1],1,mx,x);
			update(rt[n+1],1,mx,p);
			long long ans2=p<=n?1ll*p*m:a[n+1][p-n-1];
			a[n+1].push_back(ans);
			a[x].push_back(ans2);
			printf("%lld
",ans);
		}
	}
	return 0;
}