[bzoj3065] 带插入区间第k小值 [重量平衡树套线段树]

题面

传送门

思路

发现强制在线了......

本来可以树套树解决的问题,现在外层不能使用线段树了,拿什么替代呢?

我们需要一种支持单点插入、下套数据结构、数据结构上传合并复杂度最多单log,不能旋转的数据结构

这不是摆明了用重量平衡树吗?

我选了替罪羊树作为上层结构,下面套了一棵线段树,就做完了

查询的时候把替罪羊树上对应的log个区间提取出来,一起在底层权值线段树上二分即可

详见代码注释

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cassert>
#include<queue>
using namespace std;
inline int read(){
    int re=0,flag=1;char ch=getchar();
    while(!isdigit(ch)){
        if(ch=='-') flag=-1;
        ch=getchar();
    }
    while(isdigit(ch)) re=(re<<1)+(re<<3)+ch-'0',ch=getchar();
    return re*flag;
}
int s[20000010],ls=0,rs=0;
namespace ctrl{//内存分配器
    int cnt=0;
    int get(){
        int re;
        if(ls^rs){
            re=s[ls];
            ls++;if(ls==20000000) ls=0;
        }
        else re=(++cnt);
        return re;
    }
    void rec(int x){
        s[rs++]=x;
        if(rs==20000000) rs=0;
    }
}
vector<int>st;
namespace seg{
    int ch[20000010][2],sum[20000010];
    void insert(int &cur,int l,int r,int pos,int val){//权值线段树修改
        if(!cur){
            cur=ctrl::get();
            ch[cur][0]=ch[cur][1]=0;
        }
        if(l==r){sum[cur]+=val;return;}
        int mid=(l+r)>>1;
        if(mid>=pos) insert(ch[cur][0],l,mid,pos,val);
        else insert(ch[cur][1],mid+1,r,pos,val);
        sum[cur]=sum[ch[cur][0]]+sum[ch[cur][1]];
    }
    vector<int>s,ss;
    int query(int l,int r,int k){
        if(l==r) return l;
        int mid=(l+r)>>1,re=0,i;
        for(i=0;i<s.size();i++){//多个对象同步查询,s是对应区间,ss是对应单点
            re+=sum[ch[s[i]][0]];
        }
        for(i=0;i<ss.size();i++){
            re+=(ss[i]>=l&&ss[i]<=mid);
        }
        if(re>=k){
            for(i=0;i<s.size();i++) s[i]=ch[s[i]][0];
            return query(l,mid,k);
        }
        else{
            for(i=0;i<s.size();i++) s[i]=ch[s[i]][1];
            return query(mid+1,r,k-re);
        }
    }
    void dump(int cur){//拍扁替罪羊的过程中回收线段树节点
        if(!cur) return;
        dump(ch[cur][0]);dump(ch[cur][1]);
        ch[cur][0]=ch[cur][1]=sum[cur]=0;ctrl::rec(cur);
    }
}
int root,unb;
double alpha=0.8;
namespace scp{//替罪羊
    int ch[150010][2],siz[150010],rt[150010],v[150010],cnt;
    void update(int cur){
        siz[cur]=siz[ch[cur][0]]+siz[ch[cur][1]]+1;
    }
    void dfs(int u){//拍扁
        if(!u) return;
        seg::dump(rt[u]);
        dfs(ch[u][0]);
        st.push_back(u);
        dfs(ch[u][1]);
    }
    void build(int l,int r,int &cur){
        if(l>r) return;
        int mid=(l+r)>>1;
        cur=st[mid];
        ch[cur][0]=ch[cur][1]=0;siz[cur]=1;rt[cur]=0;
        build(l,mid-1,ch[cur][0]);build(mid+1,r,ch[cur][1]);
        update(cur);
        for(int i=l;i<=r;i++)
            seg::insert(rt[cur],0,70000,v[st[i]],1);
    }
    void rebuild(int &cur){//替罪羊重构
        st.clear();dfs(cur);
        build(0,st.size()-1,cur);
    }
    void insert(int &cur,int rk,int val){//替罪羊插入、判断重构
        if(!cur){
            cur=++cnt;
            ch[cur][0]=ch[cur][1]=0;siz[cur]=1;v[cur]=val;rt[cur]=0;
            seg::insert(rt[cur],0,70000,val,1);return;
        }
        seg::insert(rt[cur],0,70000,val,1);
        int L=siz[ch[cur][0]];
        if(L>=rk) insert(ch[cur][0],rk,val);
        else insert(ch[cur][1],rk-L-1,val);
        update(cur);
        if((double)(siz[cur])*alpha>(double)(max(siz[ch[cur][0]],siz[ch[cur][1]]))){
            if(unb){
                if(ch[cur][0]==unb) rebuild(ch[cur][0]);
                else rebuild(ch[cur][1]);
                unb=0;
            }
        }
        else unb=cur;
    }
    int ori;
    void change(int cur,int rk,int val){//替罪羊区间修改
        seg::insert(rt[cur],0,70000,val,1);
        if(siz[ch[cur][0]]+1==rk){ori=v[cur];v[cur]=val;}
        else if(siz[ch[cur][0]]>=rk) change(ch[cur][0],rk,val);
        else change(ch[cur][1],rk-siz[ch[cur][0]]-1,val);
        seg::insert(rt[cur],0,70000,ori,-1);
    }
    void query(int cur,int l,int r){//替罪羊上提取查询区间,如果区间覆盖了当前点,就把当前点加入查询序列
        assert(cur);
        if(l==1&&r==siz[cur]){
            seg::s.push_back(rt[cur]);return;
        }
        if(siz[ch[cur][0]]+1>=l&&siz[ch[cur][0]]+1<=r) seg::ss.push_back(v[cur]);
        if(r<=siz[ch[cur][0]]) query(ch[cur][0],l,r);
        else if(l>siz[ch[cur][0]]+1) query(ch[cur][1],l-siz[ch[cur][0]]-1,r-siz[ch[cur][0]]-1);
        else{
            if(l<=siz[ch[cur][0]]) query(ch[cur][0],l,siz[ch[cur][0]]);
            if(r>siz[ch[cur][0]]+1) query(ch[cur][1],1,r-siz[ch[cur][0]]-1);
        }
    }
}
int n,Q;
int main(){
    n=read();int i,t1,t2,t3;char ss[10];int lastans=0;
    for(i=1;i<=n;i++){
        t1=read();
        scp::insert(root,i,t1);
        if(unb){
            unb=0;
            scp::rebuild(root);
        }
    }
    Q=read();
    while(Q--){
        scanf("%s",ss);
        if(ss[0]=='Q'){
            t1=read();t2=read();t3=read();
            t1^=lastans;t2^=lastans;t3^=lastans;
            seg::s.clear();seg::ss.clear();
            scp::query(root,t1,t2);
            printf("%d\n",lastans=seg::query(0,70000,t3));
        }
        if(ss[0]=='M'){
            t1=read();t2=read();
            t1^=lastans;t2^=lastans;
            scp::change(root,t1,t2);
        }
        if(ss[0]=='I'){
            t1=read();t2=read();unb=0;
            t1^=lastans;t2^=lastans;
            scp::insert(root,t1-1,t2);
            if(unb){
                unb=0;
                scp::rebuild(root);
            }
        }
    }
}

原文地址:https://www.cnblogs.com/dedicatus545/p/9780329.html

时间: 2024-10-06 21:35:05

[bzoj3065] 带插入区间第k小值 [重量平衡树套线段树]的相关文章

【XSY2720】区间第k小 整体二分 可持久化线段树

题目描述 给你你个序列,每次求区间第\(k\)小的数. 本题中,如果一个数在询问区间中出现了超过\(w\)次,那么就把这个数视为\(n\). 强制在线. \(n\leq 100000,a_i<n,w\leq n\) 题解 考虑整体二分. 先看看离线要怎么做. 现在我们要计算每个数对每个区间的贡献. 对于每个询问区间和每种数,让这个区间最右边\(w\)个数对这个询问的贡献为\(1\),第\(w+1\)个数对这个询问的贡献为\(-w\). 这样每个数的贡献就是二维平面上的一个矩形.可以用扫描线+线段

[BZOJ3065]带插入区间K小值 解题报告 替罪羊树+值域线段树

刚了一天的题终于切掉了,数据结构题的代码真**难调,这是我做过的第一道树套树题,做完后感觉对树套树都有阴影了......下面写一下做题记录. Portal Gun:[BZOJ3065]带插入区间k小值. 这道题的题面其实都提醒怎么做了,维护区间k小值用值域线段树,但要维护一个插入操作,树状数组套主席树也用不了,那么这道题还剩下平衡树可以搞,那就上平衡树吧. 我这里的做法,因为要维护序列的顺序,所以我这里用到替罪羊树套值域线段树:我们在替罪羊树的每个节点都套一颗值域线段树,记录以该节点为根的子树的

少年,想学带修改主席树吗 | BZOJ1901 带修改区间第k小

少年,想学带修改主席树吗 | BZOJ1901 带修改区间第k小 有一道题(BZOJ 1901)是这样的:n个数,m个询问,询问有两种:修改某个数/询问区间第k小. 不带修改的区间第k小用主席树很好写,不会的同学可以看一下这个. 加上修改怎么做呢?我们可以用数学老师成天讲的类比思想: 可以发现,不修改的区间k小问题中,每加入一个原序列中的数,对应的主席树在上一个的基础上进行修改,而查询的时候用右端点主席树减去左端点左边的主席树.这样的操作就像是维护前缀和:每次加入一个元素的时候,sum[i] =

Bzoj3065 带插入区间K小值

Time Limit: 60 Sec  Memory Limit: 512 MBSubmit: 3436  Solved: 1103 Description 从前有n只跳蚤排成一行做早操,每只跳蚤都有自己的一个弹跳力a[i].跳蚤国王看着这些跳蚤国欣欣向荣的情景,感到非常高兴.这时跳蚤国王决定理性愉悦一下,查询区间k小值.他每次向它的随从伏特提出这样的问题: 从左往右第x个到第y个跳蚤中,a[i]第k小的值是多少.这可难不倒伏特,他在脑袋里使用函数式线段树前缀和的方法水掉了跳蚤国王的询问.这时伏

POJ 2761(求区间第k小值)

Feed the dogs Time Limit: 6000MS   Memory Limit: 65536KB   64bit IO Format: %I64d & %I64u Submit Status Description Wind loves pretty dogs very much, and she has n pet dogs. So Jiajia has to feed the dogs every day for Wind. Jiajia loves Wind, but no

bzoj3065带插入区间K小值

这题其实好像很难,但是听werkeytom_ftd说可以用块链水,于是就很开心地去打了个块状链表套主席树,插入操作就直接插到一个块中,注意如果块的大小2*block就将块分开,注意每一个修改或插入都要修改后继的状态,贴代码: #include<iostream> #include<algorithm> #include<cstdio> #include<cmath> #include<cstring> #define fo(i,a,b) for(

Codevs-4919 线段树练习4(区间加上一个值并求摸个区间整除k的数的个数,线段树+数组维护)

给你N个数,有两种操作 1:给区间[a,b]内的所有数都增加X 2:询问区间[a,b]能被7整除的个数 输入描述 Input Description 第一行一个正整数n,接下来n行n个整数,再接下来一个正整数Q,表示操作的个数. 接下来Q行每行若干个整数.如果第一个数是add,后接3个正整数a,b,X,表示在区间[a,b]内每个数增加X,如果是count,表示统计区间[a,b]能被7整除的个数 输出描述 Output Description 对于每个询问输出一行一个答案 样例输入 Sample

POJ - 2155 Matrix (二维树状数组 + 区间改动 + 单点求值 或者 二维线段树 + 区间更新 + 单点求值)

POJ - 2155 Matrix Time Limit: 3000MS   Memory Limit: 65536KB   64bit IO Format: %I64d & %I64u Submit Status Description Given an N*N matrix A, whose elements are either 0 or 1. A[i, j] means the number in the i-th row and j-th column. Initially we ha

【块状链表】【权值分块】bzoj3065 带插入区间K小值

显然是块状链表的经典题.但是经典做法的复杂度是O(n*sqrt(n)*log(n)*sqrt(log(n)))的,出题人明确说了会卡掉. 于是我们考虑每个块内记录前n个块的权值分块. 查询的时候差分什么的,复杂度就是O(n*sqrt(n))的了. 插入的时候为了防止块过大,要考虑裂块(细节较多). 感谢bzoj提供O2,我的STL块链才能通过(list+vector). #include<cstdio> #include<list> #include<vector> #