区间动态第k大模板(树状数组套主席树)

 

原理:

https://blog.csdn.net/g21wcr/article/details/86652152

https://blog.csdn.net/u014664226/article/details/47839973

板子一:

时空复杂度:O((n+q)*logn*logn)

洛谷 P2617

#include<bits/stdc++.h>
using namespace std;
 
const int maxn=1e5+5;
struct node
{
    int lc,rc,siz;
}tree[maxn*400*2];//节点
 
struct qury
{
    int a,b,c;
}q[maxn];   //离线操作
 
char op[10];//指令
int n,m,ansx,ansy,tot,len;//n个数 m个查询 tot个节点 len为离散化后的长度
                          //ansx ansy 用于记录 查询操作 所需要的节点个数
int root[maxn],a[maxn],b[maxn<<1],qx[maxn],qy[maxn];//记录根节点编号 原序列 离散化后的序列
                         // qx qy 用于记录 查询操作 所需要的节点
 
inline int lowbit(int x)
{
    return x&(-x);
}
 
inline void update(int &root,int pre,int l,int r,int val,int num)
{
    root=++tot;
	tree[root]=tree[pre];
	tree[root].siz+=num;
	if(l==r)
        return;
    int mid=(l+r)>>1;
	if(val<=mid)
        update(tree[root].lc,tree[pre].lc,l,mid,val,num);
	else
        update(tree[root].rc,tree[pre].rc,mid+1,r,val,num);
}
 
inline void add(int pos,int v)
{
	int val=lower_bound(b+1,b+len+1,a[pos])-b;
	for(int i=pos;i<=n;i+=lowbit(i))
		update(root[i],root[i],1,len,val,v);
}
 
inline int query(int l,int r,int k)
{
	if(l==r) return l;
	int sum=0;
	int mid=(l+r)>>1;
	for(int i=1;i<=ansx;++i)
        sum-=tree[tree[qx[i]].lc].siz;
	for(int i=1;i<=ansy;++i)
        sum+=tree[tree[qy[i]].lc].siz;
	if(k<=sum){
		for(int i=1;i<=ansx;++i)
            qx[i]=tree[qx[i]].lc;
		for(int i=1;i<=ansy;++i)
            qy[i]=tree[qy[i]].lc;
		return query(l,mid,k);
	}
	else{
		for(int i=1;i<=ansx;++i)
            qx[i]=tree[qx[i]].rc;
		for(int i=1;i<=ansy;++i)
            qy[i]=tree[qy[i]].rc;
		return query(mid+1,r,k-sum);
	}
}
 
inline void prework()//初始化操作 包括离散化等
{
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;++i)
    {
        scanf("%d",&a[i]);
        b[++len]=a[i];
    }
    for(int i=1;i<=m;++i)
    {
        scanf("%s",op);
        scanf("%d %d",&q[i].a,&q[i].b);
        if(op[0]=='Q')
            scanf("%d",&q[i].c);
        if(op[0]=='C')
            q[i].c=-1,b[++len]=q[i].b;
    }
    sort(b+1,b+1+len);
    len=unique(b+1,b+1+len)-b-1;
    for(int i=1;i<=n;++i)
        add(i,1);
}
 
inline void mainwork()//查询操作
{
    for(int i=1;i<=m;++i)
    {
        if(q[i].c>=0)
        {
            ansx=0,ansy=0;
            for(int j=q[i].b;j;j-=lowbit(j))
                qy[++ansy]=root[j];
            for(int j=q[i].a-1;j;j-=lowbit(j))
                qx[++ansx]=root[j];
            printf("%d\n",b[query(1,len,q[i].c)]);
        }
        else
        {
            add(q[i].a,-1);
            a[q[i].a]=q[i].b;
            add(q[i].a,1);
        }
    }
}
 
int main(){
    prework();
    mainwork();
    return 0;
}

板子二:

时空复杂度:O(n*lgn+q*logn*lgn) 因为先建立了一棵静态主席树。

ZOJ 2112

#include<bits/stdc++.h>
using namespace std;

const int maxn=6e4+5;
const int M=2400005;

struct node
{
    int lc,rc,siz;
}tree[M];//节点

struct qury
{
    int a,b,c;
}q[10010];   //离线操作

char op[10];//指令
int n,m,ansx,ansy,tot,len;//n个数 m个查询 tot个节点 len为离散化后的长度
                          //ansx ansy 用于记录 查询操作 所需要的节点个数
int srt[maxn],rt[maxn],a[maxn],b[maxn],qx[maxn],qy[maxn];//记录静态主席树根节点编号 新增节点编号 原序列 离散化后的序列
                         // qx qy 用于记录 查询操作 所需要的节点

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

inline void update(int &root,int pre,int l,int r,int val,int num)
{
    root=++tot;
	tree[root]=tree[pre];
	tree[root].siz+=num;
	if(l==r)
        return;
    int mid=(l+r)>>1;
	if(val<=mid)
        update(tree[root].lc,tree[pre].lc,l,mid,val,num);
	else
        update(tree[root].rc,tree[pre].rc,mid+1,r,val,num);
}

inline void add(int pos,int v)
{
	int val=lower_bound(b+1,b+len+1,a[pos])-b;
	for(int i=pos;i<=n;i+=lowbit(i))
		update(rt[i],rt[i],1,len,val,v);
}

inline int query(int x,int y,int l,int r,int k)
{
	if(l==r) return l;
	int mid=(l+r)>>1;
    int sum=tree[tree[y].lc].siz-tree[tree[x].lc].siz;
	for(int i=1;i<=ansx;++i)
        sum-=tree[tree[qx[i]].lc].siz;
	for(int i=1;i<=ansy;++i)
        sum+=tree[tree[qy[i]].lc].siz;
	if(k<=sum){
		for(int i=1;i<=ansx;++i)
            qx[i]=tree[qx[i]].lc;
		for(int i=1;i<=ansy;++i)
            qy[i]=tree[qy[i]].lc;
		return query(tree[x].lc,tree[y].lc,l,mid,k);
	}
	else{
		for(int i=1;i<=ansx;++i)
            qx[i]=tree[qx[i]].rc;
		for(int i=1;i<=ansy;++i)
            qy[i]=tree[qy[i]].rc;
		return query(tree[x].rc,tree[y].rc,mid+1,r,k-sum);
	}
}


inline void prework()//初始化操作 包括离散化等
{
    tot=len=rt[0]=0;
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;++i)
    {
        scanf("%d",&a[i]);
        b[++len]=a[i];
    }
    for(int i=1;i<=m;++i)
    {
        scanf("%s",op);
        scanf("%d %d",&q[i].a,&q[i].b);
        if(op[0]=='Q') //查询
            scanf("%d",&q[i].c);
        else if(op[0]=='C')  //修改
            q[i].c=-1,b[++len]=q[i].b;
    }
    sort(b+1,b+1+len);
    len=unique(b+1,b+1+len)-b-1;
    for(int i=1;i<=n;++i)
    {
        int val=lower_bound(b+1,b+len+1,a[i])-b;
        update(srt[i],srt[i-1],1,len,val,1);
    }
    for(int i=1;i<=n;i++)
        rt[i]=srt[0];
}

inline void mainwork()//查询操作
{
    for(int i=1;i<=m;++i)
    {
        if(q[i].c>=0)
        {
            ansx=0,ansy=0;
            for(int j=q[i].b;j;j-=lowbit(j))
                qy[++ansy]=rt[j];
            for(int j=q[i].a-1;j;j-=lowbit(j))
                qx[++ansx]=rt[j];
            printf("%d\n",b[query(srt[q[i].a-1],srt[q[i].b],1,len,q[i].c)]);
        }
        else
        {
            add(q[i].a,-1);
            a[q[i].a]=q[i].b;
            add(q[i].a,1);
        }
    }
}

int main(){
    int t;
    scanf("%d",&t);
    while(t--)
    {
        prework();
        mainwork();
    }
    return 0;
}

CDQ分治 整体二分 待补。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值