欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

珂朵莉树(Chtholly Tree)学习笔记

程序员文章站 2023-04-08 13:39:15
珂朵莉树(Chtholly Tree)学习笔记 珂朵莉树原理 其原理在于运用一颗树(set,treap,splay......)其中要求所有元素有序,并且支持基本的操作(删除,添加,查找......)来实现区间压缩。 那么区间压缩的意义在于 区间推平 这是珂朵莉树的 核心 (如果没有这个操作实际上不 ......

珂朵莉树(chtholly tree)学习笔记


珂朵莉树原理

其原理在于运用一颗树(set,treap,splay......)其中要求所有元素有序,并且支持基本的操作(删除,添加,查找......)来实现区间压缩。

那么区间压缩的意义在于区间推平这是珂朵莉树的核心(如果没有这个操作实际上不一定需要这种算法)

ps:若保证有连续相等甚至递增的区间,也可以的(吧?)。

可想而知它的操作在于对区间的分裂合并操作

(为什么?因为这样可以方便而快捷的区间推平

珂朵莉树的实现

在众多树中因为set这个c++自带,所以决定选择它。

结构

一个node结构体——把它叫区间(已typedef long long ll)

struct node
{
    int l,r;ll v;//这里官方写法加了一个mutable,看个人写法
    node(int l,int r,ll v):l(l),r(r),v(v){}
    bool operator < (const node& o) const
    {
        return l<o.l;
    }
};

声明集合

set<node>s;

初始化

for(int i=1;i<=n;++i)
    s.insert(i,i,val);
s.insert(n+1,n+1,0);//见下面中it!=s.end()的前提

分裂

#define it_ set<node>::iterator
it_ split(int pos)
{
    it_ it=s.lower_bound(node(pos));
    if(it!=s.end()&& it->l==pos) return it;
    --it;
    int ll=it->l,rr=it->r;
    ll vv=it->v;
    s.erase(it);
    s.insert(node(ll,pos-1,vv));
    return s.insert(node(pos,rr,vv)).first;
}

区间推平

void assign(int l,int r,ll val=0)
{
    it_ itl=split(l),itr=split(r+1);
    s.erase(itl,itr);
    s.insert(node(l,r,val));
}

如上。


例子

见cf896c

仅仅给出代码

#include <bits/stdc++.h>
#define it_ set<node>::iterator
// #define swap(x,y) {int (tmp)=(x),(x)=(y),(y)=(tmp);}
using namespace std;
typedef long long ll;
const int m7=1e9+7;
const int maxn=1e5+5;
int n,m;
ll seed,vmax;
ll a[maxn];
struct node
{
    int l,r;mutable ll v;
    node(int l,int r=-1,ll v=0):l(l),r(r),v(v){}
    bool operator < (const node& o) const
    {
        return l<o.l;
    }
};
set<node>s;
it_ split(int pos)
{
    it_ it=s.lower_bound(node(pos));
    if(it!=s.end()&& it->l==pos) return it;
    --it;
    int ll=it->l,rr=it->r;
    ll vv=it->v;
    s.erase(it);
    s.insert(node(ll,pos-1,vv));
    return s.insert(node(pos,rr,vv)).first;
}
void assign(int l,int r,ll val=0)
{
    it_ itl=split(l),itr=split(r+1);
    s.erase(itl,itr);
    s.insert(node(l,r,val));
}
void add(int l,int r,ll val=0)
{
    it_ itl=split(l),itr=split(r+1);
    for(it_ i=itl;i!=itr;++i)
    {
        i->v+=val;
    }
}
ll so(int l,int r,int k)
{
    it_ itl=split(l),itr=split(r+1);
    vector<pair<ll,int>>v;
    v.clear();
    for(;itl!=itr;++itl)
    {
        v.push_back(pair<ll,int>(itl->v,itl->r-itl->l+1));
    }
    sort(v.begin(),v.end());
    for(vector<pair<ll,int>>::iterator it=v.begin();it!=v.end();++it)
    {
        k-=it->second;
        if(k<=0)return it->first;
    }
}
ll pow(ll a,ll b,ll mod)
{
    ll ret=1;a%=mod;
    while(b)
    {
        if(b&1) ret=ret*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ret%mod;
}
ll ok(int l,int r,ll k,ll mod)
{
    it_ itl=split(l),itr=split(r+1);
    ll ans=0;
    for(;itl!=itr;++itl)
    {
        ans+=(ll)(itl->r-itl->l+1)*pow(itl->v,k,mod)%mod;
        ans%=mod;
    }
    return ans%mod;
}
ll rnd()
{
    ll ret=seed;
    seed=(seed*7+13)%m7;
    return ret;
}

int main(int argc, char const *argv[])
{
    ios::sync_with_stdio(0);
    cin>>n>>m>>seed>>vmax;
    for(int i=1;i<=n;++i)
    {
        a[i]=(rnd()%vmax)+1;
        s.insert(node(i,i,a[i]));
    }
    s.insert(node(n+1,n+1,0));
    for(int i=1;i<=m;++i)
    {
        int op=(rnd()%4)+1,
        l=(rnd()%n)+1,
        r=(rnd()%n)+1;ll x,y;
        if(l>r)
            swap(l,r);
        if(op==3)
            x=(rnd()%(ll)(r-l+1))+1;
        else 
            x=(rnd()%vmax)+1;
        if(op==4)
            y=(rnd()%vmax)+1;
        if(op==1) add(l,r,x); 
        else if(op==2) assign(l,r,x);
        else if(op==3) cout<<so(l,r,x)<<endl;
        else if(op==4) cout<<ok(l,r,x,y)<<endl;
    }
    return 0;
}