[ZJOI2013]K大数查询

区间求第k小,变为了区间求第k大。
那么现在,对于插入数的结构体来说,如果q[i].k>mid,就把它划分到右区间,并进行单点累加;如果q[i].k<=mid,则划分到左区间。而对于查询的结构体来说,如果查询区间内的值,记为sum,当q[i].k>sum时,可知是由于mid不够小的缘故,所以应该把mid再变小写,即划分到左区间;当q[i].k<=sum时,说明mid已经够大了,所有划分到右区间。
这里的划分,和求第k小相比,好像不是那么“显然”与“对称”了,所以需要注意。
#include <bits/stdc++.h>
#define int long long
using namespace std;
const int N=5e4+5;
int n,m,Q;
int sum[N<<2],tag[N<<2],ans[N];
struct node{int x,y,k,id,opt;}q[N<<1],q1[N<<1],q2[N<<1];

inline void pushdown(int k,int l,int r,int mid)
{
	if (tag[k])
	{
		tag[k<<1]+=tag[k]; tag[k<<1|1]+=tag[k];
		sum[k<<1]+=(mid-l+1)*tag[k];
		sum[k<<1|1]+=(r-(mid+1)+1)*tag[k];
		tag[k]=0;
	}
}
void change(int k,int l,int r,int qx,int qy,int v)
{
	if (qx<=l && r<=qy)
	{
		sum[k]+=(r-l+1)*v;
		tag[k]+=v;
		return;
	}
	int mid=l+r>>1;
	pushdown(k,l,r,mid);
	if (qx<=mid) change(k<<1,l,mid,qx,qy,v);
	if (mid<qy) change(k<<1|1,mid+1,r,qx,qy,v);
	sum[k]=sum[k<<1]+sum[k<<1|1];
}
int query(int k,int l,int r,int qx,int qy)
{
	if (qx<=l && r<=qy) return sum[k];
	int mid=l+r>>1;
	pushdown(k,l,r,mid);
	int res=0;
	if (qx<=mid) res+=query(k<<1,l,mid,qx,qy);
	if (mid<qy) res+=query(k<<1|1,mid+1,r,qx,qy);
	return res;	
}

void solve(int ql,int qr,int l,int r)
{
	if (ql>qr) return;
	if (l==r)
	{
		for (register int i=ql; i<=qr; ++i) if (q[i].opt==2) ans[q[i].id]=l;	
		return;
	}
	int mid=l+r>>1;
	int p1=0,p2=0;
	for (register int i=ql; i<=qr; ++i)
	if (q[i].opt==1)
	{
		if (q[i].k>mid)
		{
			change(1,1,n,q[i].x,q[i].y,1);
			q2[++p2]=q[i];
		}
		else q1[++p1]=q[i];
	}
	else
	{
		int res=query(1,1,n,q[i].x,q[i].y);
		if (q[i].k>res) 
		{
			q[i].k-=res;
			q1[++p1]=q[i];	
		}
		else
		{
			q2[++p2]=q[i];
		}
	}
	for (register int i=1; i<=p2; ++i) if (q2[i].opt==1) change(1,1,n,q2[i].x,q2[i].y,-1);
	for (register int i=1; i<=p1; ++i) q[ql+i-1]=q1[i];
	for (register int i=1; i<=p2; ++i) q[ql+p1-1+i]=q2[i];
	solve(ql,ql+p1-1,l,mid);
	solve(ql+p1,qr,mid+1,r);
}

signed main(){
	scanf("%lld%lld",&n,&m);
	for (register int i=1; i<=m; ++i)
	{
		scanf("%lld%lld%lld%lld",&q[i].opt,&q[i].x,&q[i].y,&q[i].k);
		if (q[i].opt==2) q[i].id=++Q;			
	}
	solve(1,m,-n,n);
	for (register int i=1; i<=Q; ++i) printf("%d\n",ans[i]);
return 0;	
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值