[ZJOI2013]K大数查询

一、题目

点此看题

二、解法

0x01 树套树

这道题的思路特别巧妙，树套树不一定要用区间线段树套权值线段树，还可以反过来套。
我们维护一个动态开点的权值线段树，每个点代表权值$[l,r]$在整个区间的出现情况，套上一个动态开点的区间线段树，操作$1$对权值线段树单点修改，然后对每个点的$[a,b]$区间修改。
操作$2$先算右子树根的$[a,b]$和，如果答案在里面，找右子树，否则减掉之后找左子树，最后跑到叶子就得到了答案，时间复杂度$O(n{\log }^2)$，空间复杂度$O(n{\log }^2)$。
这道题是需要卡常的，$sum$需要开$unsigned\_int$，输入的$c$要开$long\_long$。

#include <cstdio>
#define uint unsigned int
#define LL long long
const int MAXN = 500005;
int read()
{
	int x=0,flag=1;char c;
	while((c=getchar())<'0' || c>'9') if(c=='-') flag=-1;
	while(c>='0' && c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
	return x*flag;
}
int n,m,a,b,rt,cnt1,cnt2;LL c;
struct node1
{
	int ls,rs,lazy;
	uint sum;
}tr[MAXN*40];
struct node2
{
	int ls,rs,rt;
}Tr[MAXN*20];
void modify(int x,int l,int r,int v)
{
	tr[x].sum+=v*(r-l+1);
	tr[x].lazy+=v;
}
void up(int x)
{
	tr[x].sum=tr[tr[x].ls].sum+tr[tr[x].rs].sum;
}
void down(int x,int l,int r)
{
	if(!tr[x].lazy) return ;
	int mid=(l+r)>>1;
	if(!tr[x].ls) tr[x].ls=++cnt1;
	if(!tr[x].rs) tr[x].rs=++cnt1;
	modify(tr[x].ls,l,mid,tr[x].lazy);
	modify(tr[x].rs,mid+1,r,tr[x].lazy);
	tr[x].lazy=0;
}
void add(int &x,int l,int r,int L,int R)
{
	if(l>R || L>r) return ;
	if(!x) x=++cnt1;
	if(L<=l && r<=R)
	{
		modify(x,l,r,1);
		return ;
	}
	down(x,l,r);
	int mid=(l+r)>>1;
	add(tr[x].ls,l,mid,L,R);
	add(tr[x].rs,mid+1,r,L,R);
	up(x);
}
uint query(int x,int l,int r,int L,int R)
{
	if(!x || l>R || L>r) return 0;
	if(L<=l && r<=R)
		return tr[x].sum;
	down(x,l,r);
	int mid=(l+r)>>1;
	return query(tr[x].ls,l,mid,L,R)+query(tr[x].rs,mid+1,r,L,R);
}
void Modify(int &x,int l,int r,int id)
{
	if(!x) x=++cnt2;
	add(Tr[x].rt,1,n,a,b);
	if(l==r) return ;
	int mid=(l+r)>>1;
	if(mid>=id)
		Modify(Tr[x].ls,l,mid,id);
	else
		Modify(Tr[x].rs,mid+1,r,id);
}
int Query(int &x,int l,int r,int s)
{
	if(!x) x=++cnt2;
	if(l==r) return l;
	int mid=(l+r)>>1;uint t=query(Tr[Tr[x].rs].rt,1,n,a,b);
	if(t>=s)
		return Query(Tr[x].rs,mid+1,r,s);
	return Query(Tr[x].ls,l,mid,s-t);
}
int main()
{
	n=read();m=read();
	while(m--)
	{
		int op=read();a=read();b=read();scanf("%lld",&c);
		if(op==1)
		{
			Modify(rt,-n,n,c);
		}
		if(op==2)
		{
			printf("%d\n",Query(rt,-n,n,c));
		}
	}
}

0x02 整体二分

首先我们把询问和修改放在一起，然后在递归的时候维护$(ql,qr,l,r)$，就表示$[ql,qr]$之间询问的答案在$[l,r]$之中，每次拆分的时候考虑权值的中点$mid$，然后维护出一个树状数组（这里的区间修改为加$1$，可以同时支持区间修改和区间查询），修改的话考虑是否大于$mid$，如果大于$mid$就把这个修改归类到右边，同时修改树状数组，否则归类到左边，不修改。

询问的话就先算出这个区间中大于$mid$的个数，如果大于$mid$就分到左边（因为答案在左边，参考线段树这类问题的求法），然后减去这个个数，否则分到右边。最后我们把左右分别递归，知道权值区间只剩一个值就确定了答案。

#include <cstdio>
#define int long long
const int M = 50005;
int read()
{
	int x=0,flag=1;char c;
	while((c=getchar())<'0' || c>'9') if(c=='-') flag=-1;
	while(c>='0' && c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
	return x*flag;
}
int n,m,c1[M],c2[M],pd[M],ans[M];
struct node
{
    int op,l,r,v,id;
}p[M],L[M],R[M];
int lowbit(int x)
{
    return x&(-x);
}
void upd(int x,int f)
{
    for(int i=x;i<=n;i+=lowbit(i))
        c1[i]+=f,c2[i]+=f*x;
}
int ask(int x)
{
    int r=0;
    for(int i=x;i>=1;i-=lowbit(i))
        r+=c1[i]*(x+1)-c2[i];
    return r;
}
void cdq(int ql,int qr,int l,int r)
{
    if(ql>qr || l>r) return ;
    if(l==r)
    {
        for(int i=ql;i<=qr;i++)
            ans[p[i].id]=l;
        return ;
    }
    int mid=(l+r)>>1,nl=0,nr=0;
    for(int i=ql;i<=qr;i++)
    {
        if(p[i].op==1)
        {
            if(p[i].v>mid)
                upd(p[i].l,1),upd(p[i].r+1,-1),R[++nr]=p[i];
            else L[++nl]=p[i];
        }
        else
        {
            int tmp=ask(p[i].r)-ask(p[i].l-1);
            if(tmp<p[i].v) p[i].v-=tmp,L[++nl]=p[i];
            else R[++nr]=p[i];
        }
    }
    for(int i=ql;i<=qr;i++)
        if(p[i].op==1 && p[i].v>mid)
            upd(p[i].l,-1),upd(p[i].r+1,1);
    for(int i=1;i<=nl;i++) p[ql+i-1]=L[i];
    for(int i=1;i<=nr;i++) p[ql+nl+i-1]=R[i];
    cdq(ql,ql+nl-1,l,mid);
    cdq(ql+nl,qr,mid+1,r);
}
signed main()
{
    n=read();m=read();
    for(int i=1;i<=m;i++)
    {
        int op=read(),l=read(),r=read(),v=read();
        p[i]=node{op,l,r,v,i};
        if(op==2) pd[i]=1;
    }
    cdq(1,m,1,n);
    for(int i=1;i<=m;i++)
        if(pd[i])
            printf("%d\n",ans[i]);
}