LG3960 列队

题意

传送门

分析

参照博客

树状数组+离线处理即可。

利用树状数组下标本质即可\(O(\log n)\)求第k大。

代码

#include<iostream>
#include<cstring>
#include<cmath>
#include<cstdio>
#include<set>
#include<map>
#include<queue>
#include<vector>
#include<algorithm>
#include<string>
#define co const
#define il inline
#define rg register
const int MAXBUF=20000005;
char B[MAXBUF],*Si=B,*Ti=B;
inline char getc()
{
    if (Si==Ti) Ti=(Si=B)+fread(B,1,MAXBUF,stdin);
    if (Si==Ti) return 0;
    return *Si++;
}
template <class T>
inline void read(T &a)
{
    static char c;
    while ((c=getc())<'0'||c>'9');
    a=c-'0';
    while ((c=getc())>='0'&&c<='9') a=(a<<3)+(a<<1)+c-'0';
}
using namespace std;
typedef long long ll;

co int MAXN=3e5+7;
int n,m,nq;
int q[MAXN][2];

struct $
{
	int p,id;

	$(){}

	$(int p,int id):p(p),id(id){}
};
vector <$> t[MAXN];
typedef vector<$>::iterator it;

int c[MAXN<<1],mx; // edit 1

int lowbit(int x)
{
	return x&-x;
}

void add(int p,int v)
{
	for(int i=p;i<=mx;i+=lowbit(i))
		c[i]+=v;
}

int sum(int p)
{
	int res=0;
	for(int i=p;i;i-=lowbit(i))
		res+=c[i];
	return res;
}

int find(int p)
{
	int res=0;
	for(int i=20;i>=0;--i)
	{
		res+=(1<<i);
		if(res>mx||c[res]>=p) // 不足最后加1，过剩则无法处理
			res-=(1<<i);
		else
			p-=c[res];
	}
	return res+1;
}

int d[MAXN];
vector<ll>v[MAXN],lst;

int main()
{
//	freopen(".in","r",stdin);
//	freopen(".out","w",stdout);
	read(n);read(m);read(nq);
	mx=max(n,m)+nq;
	for(int i=1;i<=nq;++i)
	{
		read(q[i][0]);read(q[i][1]);
		if(q[i][1]!=m)
			t[q[i][0]].push_back($(q[i][1],i));
	}
	for(int i=1;i<=mx;++i)
		add(i,1);
	for(int i=1;i<=n;++i)
	{
		for(it j=t[i].begin();j!=t[i].end();++j)
			add(d[j->id]=find(j->p),-1);
		for(it j=t[i].begin();j!=t[i].end();++j)
			add(d[j->id],1);
	}
	ll ans;
	for(int i=1;i<=nq;++i)
	{
		int x=find(q[i][0]);
		ans=x<=n?(ll)x*m:lst[x-n-1];
		add(x,-1);
		if(q[i][1]!=m)
		{
			v[q[i][0]].push_back(ans);
			ans=d[i]<m?((ll)q[i][0]-1)*m+d[i]:v[q[i][0]][d[i]-m];
		}
		lst.push_back(ans);
		printf("%lld\n",ans);
	}
	return 0;
}