【整体二分】[ZJOI2013]K大数查询

题目

https://darkbzoj.tk/problem/3110

思路

这题有好几种做法，我选择整体二分。

整体二分就是对于所有询问二分答案，分别加入对其有贡献的区间。

对于一个修改操作，如果对当前答案有贡献，就在线段树上做区间加
对于查询操作，在线段树上查询看看是否大于mid来确定放在哪边。

代码

#include<bits/stdc++.h>
#define ll long long
#define lc p<<1
#define rc p<<1|1
using namespace std;
const int N=5e4+77;
struct Q
{
	int op,l,r,id;
	ll v;
}q[N],tl[N],tr[N];
int bj[N<<2],cl[N<<2],ans[N];
ll sum[N<<2];
int n,m,tot;
void pushdown(int p,int l,int r){
	if(cl[p]){
		cl[p]=0;
		bj[lc]=bj[rc]=sum[lc]=sum[rc]=0;
		cl[lc]=1,cl[rc]=1;
	}
	if(bj[p]){
		int mid=(l+r)>>1;
		bj[lc]+=bj[p],bj[rc]+=bj[p];
		sum[lc]+=bj[p]*(mid-l+1);
		sum[rc]+=bj[p]*(r-mid);
		bj[p]=0;
	}
}	
void add(int ql,int qr,int w,int p=1,int l=1,int r=n)
{
	if(ql<=l&&r<=qr)
	{
		bj[p]+=w;sum[p]+=w*(r-l+1);
		return;
	}
	if(bj[p]||cl[p]) pushdown(p,l,r);
	int mid=(l+r)>>1;
	if(ql<=mid) add(ql,qr,w,lc,l,mid);
	if(mid<qr) add(ql,qr,w,rc,mid+1,r);
	sum[p]=sum[lc]+sum[rc];
}
ll query(int ql,int qr,int p=1,int l=1,int r=n)
{
	if(ql<=l&&r<=qr) return sum[p];
	int mid=(l+r)>>1;
	ll tt=0;
	if(bj[p]||cl[p]) pushdown(p,l,r);
	if(ql<=mid) tt+=query(ql,qr,lc,l,mid);
	if(mid<qr) tt+=query(ql,qr,rc,mid+1,r);
	return tt;
}
void solve(int st,int en,int l,int r)
{
	if(l==r)
	{
		for(int i=st; i<=en; i++)
			if(q[i].op==2) ans[q[i].id]=l;
		return;
	}
	int mid=(l+r)>>1;
	bool fl=0,fr=0;
	int L=0,R=0;
	cl[1]=1;bj[1]=sum[1]=0;
	for(int i=st; i<=en; i++)
	{
		if(q[i].op==1)
		{
			if(q[i].v>mid) add(q[i].l,q[i].r,1),tr[++R]=q[i];
			else tl[++L]=q[i];
		}
		else
		{
			ll val=query(q[i].l,q[i].r);
			if(val<q[i].v)
			{
				q[i].v-=val; fl=1; tl[++L]=q[i];
			}
			else
			{
				fr=1; tr[++R]=q[i];
			}
		}
	}
	for(int i=1; i<=L; i++) q[st+i-1]=tl[i];
	for(int i=L+1; i<=L+R; i++) q[st+i-1]=tr[i-L];
	if(fl) solve(st,st+L-1,l,mid);
	if(fr) solve(st+L,en,mid+1,r);
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1; i<=m; i++)
	{
		scanf("%d%d%d%d",&q[i].op,&q[i].l,&q[i].r,&q[i].v);
		if(q[i].op==2)q[i].id=++tot;
	}
	solve(1,m,-n,n);
	for(int i=1; i<=tot; i++) printf("%d\n",ans[i]);
}