Template(Updating)

1、Splay

(Tyvj1728)

#include <bits/stdc++.h>

using namespace std;
const int MAXN=1e5+,INF=<<;
int n,op,x,root,ch[MAXN][],sz[MAXN],cnt[MAXN],val[MAXN],f[MAXN],tot=;

void pushup(int x)
{sz[x]=sz[ch[x][]]+sz[ch[x][]]+cnt[x];}

void Rotate(int x)
{
    int y=f[x],z=f[y],k=(x==ch[y][]);
    ch[z][y==ch[z][]]=x;f[x]=z;
    ch[y][k]=ch[x][k^];f[ch[x][k^]]=y;
    ch[x][k^]=y;f[y]=x;
    pushup(x);pushup(y);
}

void Splay(int x,int up)
{
    while(f[x]!=up)
    {
        int y=f[x],z=f[y];
        if(z!=up) (ch[y][]==x)^(ch[z][]==y)?Rotate(x):Rotate(y);
        Rotate(x);
    }
    if(!up) root=x;
}

void Insert(int x)
{
    int k=root,anc=;
    while(k&&x!=val[k])
        anc=k,k=ch[k][x>val[k]];

    if(k) cnt[k]++;
    else
    {
        k=++tot;
        if(anc) ch[anc][x>val[anc]]=k;
        ch[k][]=ch[k][]=;
        val[k]=x;cnt[k]=;sz[k]=;f[k]=anc;
    }
    Splay(k,);
}

void Find(int x)
{
    int k=root;
    if(!k) return;
    while(ch[k][x>val[k]]&&x!=val[k])
        k=ch[k][x>val[k]];
    Splay(k,);
}

int Next(int x,int flag)
{
    Find(x);
    int k=root;
    if((val[k]<x&&!flag)||(val[k]>x&&flag)) return k;

    k=ch[k][flag];
    while(ch[k][flag^]) k=ch[k][flag^];

    return k;
}

int Kth(int x)
{
    int k=root;
    if(sz[k]<x) return ;
    while(true)
    {
        if(x>sz[ch[k][]]+cnt[k])
            x-=sz[ch[k][]]+cnt[k],k=ch[k][];
        else if(sz[ch[k][]]>=x)
            k=ch[k][];
        else return val[k];
    }
}

void Delete(int x)
{
    int lst=Next(x,),nxt=Next(x,);
    Splay(lst,);Splay(nxt,lst);

    if(cnt[ch[nxt][]]>)
        cnt[ch[nxt][]]--,Splay(ch[nxt][],);
    else ch[nxt][]=;
}

int main()
{
    scanf("%d",&n);
    Insert(INF);Insert(-INF);//记得先插入边界
    for(int i=;i<=n;i++)
    {
        scanf("%d%d",&op,&x);
        if(op==) Insert(x);
        else if(op==) Delete(x);
        else if(op==) Find(x),printf("%d\n",sz[ch[root][]]);
        else if(op==) printf("%d\n",Kth(x+));
        else if(op==) printf("%d\n",val[Next(x,)]);
        else if(op==) printf("%d\n",val[Next(x,)]);
    }
    return ;
}

Splay

Tip:由于采取求前驱后继来删除节点的方式，要先$insert(INF)$和$insert(-INF)$

2、Treap

(Tyvj1728)

#include <bits/stdc++.h>

using namespace std;
const int MAXN=;
const int INF=<<;
int cnt=,root,n,s[MAXN][],siz[MAXN],val[MAXN],pri[MAXN];

void update(int i)
{siz[i]=siz[s[i][]]+siz[s[i][]]+;}

void spin(int &i,int f)
{
    int v=s[i][f^];
    s[i][f^]=s[v][f];s[v][f]=i;
    update(i);update(i=v);
}

void ins(int &i,int k)
{
    if(!i){i=++cnt;siz[i]=;val[i]=k;pri[i]=rand();return;}

    siz[i]++;
    if(val[i]>=k){ins(s[i][],k);if(pri[i]>pri[s[i][]]) spin(i,);}
    else{ins(s[i][],k);if(pri[i]>pri[s[i][]]) spin(i,);}
}

int pre(int i,int k)
{
    if(!i) return -INF;
    if(val[i]<k) return max(val[i],pre(s[i][],k));
    else return pre(s[i][],k);
}

int nxt(int i,int k)
{
    if(!i) return INF;
    if(val[i]>k) return min(val[i],nxt(s[i][],k));
    else return nxt(s[i][],k);
}

int find_key(int i,int k)
{
    if(siz[s[i][]]==k-) return val[i];
    if(siz[s[i][]]>=k) return find_key(s[i][],k);
    return find_key(s[i][],k-siz[s[i][]]-);
}

int find_rank(int i,int k)
{
    if(!i) return ;
    if(val[i]>=k) return find_rank(s[i][],k);
    return siz[s[i][]]++find_rank(s[i][],k);
}

void del(int &i,int k)
{
    if(val[i]==k)
    {
        if(s[i][]*s[i][]==)
        {
            i=s[i][]+s[i][];
            return;
        }
        if(pri[s[i][]]<pri[s[i][]]) spin(i,),del(s[i][],k);
        else spin(i,),del(s[i][],k);
    }
    else if(val[i]>=k) del(s[i][],k);
    else del(s[i][],k);
    update(i);
}

int main()
{
    scanf("%d",&n);srand(time(NULL));
    while(n--)
    {
        int op,x;scanf("%d%d",&op,&x);
        if(op==) ins(root,x);
        else if(op==) del(root,x);
        else if(op==) printf("%d\n",find_rank(root,x));
        else if(op==) printf("%d\n",find_key(root,x));
        else if(op==) printf("%d\n",pre(root,x));
        else printf("%d\n",nxt(root,x));
    }
    return ;
}

Treap

Tip:注意递归边界

3、计算几何模板

我现在发现了我在计算几何方面就是个sillycross

在这里总结一下基本模块吧，

（使用complex<double>模板类）

点乘：

double dot(point a,point b){return real(a*conj(b));}

叉乘：

double det(point a,point b){return imag(a*conj(b));}

判断点x是否在线段[L,R]上：

bool on_seg(point x,point L,point R){return det(L-x,R-x)==0 && dot(L-x,R-x)<=0;}

判断三点是否在一条直线 + L,R是否在x的两侧

判断两线段是否相交（非严格）：

bool seg_cross(point a,point b,point c,point d)
{
    double s1=det(c-a,b-a)*det(b-a,d-a);
    double s2=det(a-c,d-c)*det(d-c,b-c);
    if(s1<0 || s2<0) return false;
    if(s1==0 && s2==0) return on_seg(c,a,b) || on_seg(d,a,b);
    return true;
}

对于每个点判断另一线段的两点是否在其两端  +   对一条线段的端点恰在另一线段上的特殊处理

求点x关于线段[A,B]的对称点：

point sym(point x,point A,point B){return 2*dot(A,B)/dot(B,B)*B-A+x;}

将线段[x,A]延长一倍，求出线段[x,x']的向量，再行加减即可

4、FFT

#include <bits/stdc++.h>

using namespace std;
#define X first
#define Y second
#define pb push_back
typedef double db;
typedef long long ll;
typedef pair<int,int> P;
const int MAXN=3e6+;
struct Complex
{
    db x,y;
    Complex(db a=,db b=){x=a;y=b;}
    Complex operator + (const Complex& rhs)
    {return Complex(x+rhs.x,y+rhs.y);}
    Complex operator - (const Complex& rhs)
    {return Complex(x-rhs.x,y-rhs.y);}
    Complex operator * (const Complex& rhs)
    {return Complex(x*rhs.x-y*rhs.y,x*rhs.y+y*rhs.x);}
}a[MAXN],b[MAXN];
int n,m,lmt=,dgt,par[MAXN];

void FFT(Complex *a,int flag)
{
    for(int i=;i<lmt;i++)
        if(i<par[i]) swap(a[i],a[par[i]]);

    for(int len=;len<lmt;len<<=)
    {
        Complex unit(cos(M_PI/len),flag*sin(M_PI/len));
        for(int st=;st<lmt;st+=(len<<))
        {
            Complex w(,);
            for(int k=st;k<st+len;k++,w=w*unit)
            {
                Complex A=a[k],B=w*a[k+len];
                a[k]=A+B;a[k+len]=A-B;
            }
        }
    }
    if(flag==-)
        for(int i=;i<=n+m;i++)
            a[i].x=floor(a[i].x/lmt+0.5);
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=;i<=n;i++) scanf("%lf",&a[i].x);
    for(int i=;i<=m;i++) scanf("%lf",&b[i].x);
    while(lmt<=n+m) lmt<<=,dgt++;
    for(int i=;i<lmt;i++)
        par[i]=(par[i>>]>>)|((i&)<<(dgt-));    

    FFT(a,);FFT(b,);
    for(int i=;i<lmt;i++)
        a[i]=a[i]*b[i];
    FFT(a,-);
    for(int i=;i<=n+m;i++)
        printf("%d ",(int)a[i].x);
    return ;
}

FFT

5、NTT

#include <bits/stdc++.h>

using namespace std;
#define X first
#define Y second
#define pb push_back
typedef double db;
typedef long long ll;
typedef pair<int,int> P;
const int MAXN=4e6+,MOD=;
ll n,m,a[MAXN],b[MAXN],dgt,lmt=,par[MAXN];

ll quick_pow(ll a,ll b)
{
    ll ret=;
    for(;b;b>>=,a=a*a%MOD)
        if(b&) ret=ret*a%MOD;
    return ret;
}
void FFT(ll *a,int flag)
{
    for(int i=;i<lmt;i++)
        if(i<par[i]) swap(a[i],a[par[i]]);
    for(int len=;len<lmt;len<<=)
    {
        ll unit=quick_pow(,(MOD-)/(len<<));
        if(flag==-) unit=quick_pow(unit,MOD-);
        for(int st=;st<lmt;st+=(len<<))
        {
            ll w=;
            for(int k=st;k<st+len;k++,w=w*unit%MOD)
            {
                ll A=a[k],B=w*a[k+len]%MOD;
                a[k]=(A+B)%MOD;a[k+len]=(A-B+MOD)%MOD;
            }
        }
    }
}

int main()
{
    scanf("%lld%lld",&n,&m);
    for(int i=;i<=n;i++) scanf("%lld",&a[i]);
    for(int i=;i<=m;i++) scanf("%lld",&b[i]);
    while(lmt<=n+m) lmt<<=,dgt++;
    for(int i=;i<lmt;i++)
        par[i]=(par[i>>]>>)|((i&)<<(dgt-));

    FFT(a,);FFT(b,);
    for(int i=;i<lmt;i++)
        (a[i]*=b[i])%=MOD;
    FFT(a,-);
    ll inv=quick_pow(lmt,MOD-);
    for(int i=;i<=n+m;i++)
        printf("%lld ",a[i]*inv%MOD);
    return ;
}

NTT

6、MTT

#include <bits/stdc++.h>

using namespace std;
#define X first
#define Y second
#define pb push_back
typedef double db;
typedef long long ll;
typedef pair<int,int> P;
const int MAXN=4e5+;
ll p[]={,,};
int n,m,MOD,F[MAXN],G[MAXN],dgt,lmt=;
ll a[][MAXN],b[MAXN],res[MAXN],par[MAXN];

ll quickpow(ll a,ll b,ll MOD)
{
    a%=MOD;ll ret=;
    for(;b;b>>=,a=a*a%MOD)
        if(b&) ret=ret*a%MOD;
    return ret;
}
ll mul(ll a,ll b,ll MOD)
{
    a=(a%MOD+MOD)%MOD;
    b=(b%MOD+MOD)%MOD;ll ret=;
    for(;b;b>>=,a=(a+a)%MOD)
        if(b&) (ret+=a)%=MOD;
    return ret;
}
ll inv(ll a,ll MOD)
{return quickpow(a,MOD-,MOD);}
void FFT(ll *a,int flag,ll MOD)
{
    for(int i=;i<lmt;i++)
        if(i<par[i]) swap(a[i],a[par[i]]);
    for(int len=;len<lmt;len<<=)
    {
        ll unit=quickpow(,(MOD-)/(len<<),MOD);
        if(flag==-) unit=inv(unit,MOD);
        for(int st=;st<lmt;st+=(len<<))
        {
            ll w=;
            for(int k=st;k<st+len;k++,w=w*unit%MOD)
            {
                ll A=a[k],B=w*a[k+len]%MOD;
                a[k]=(A+B)%MOD;a[k+len]=(A-B+MOD)%MOD;
            }
        }
    }
    if(flag==-)
    {
        ll INV=inv(lmt,MOD);
        for(int i=;i<lmt;i++)
            a[i]=a[i]*INV%MOD;
    }
}
void solve(ll *a,ll *b,ll MOD)
{
    for(int i=;i<=n;i++) a[i]=F[i];
    for(int i=;i<=m;i++) b[i]=G[i];
    for(int i=m+;i<lmt;i++) b[i]=;
    FFT(a,,MOD);FFT(b,,MOD);
    for(int i=;i<lmt;i++) a[i]=a[i]*b[i]%MOD;
    FFT(a,-,MOD);
}

int main()
{
    scanf("%d%d%d",&n,&m,&MOD);
    for(int i=;i<=n;i++) scanf("%d",&F[i]);
    for(int i=;i<=m;i++) scanf("%d",&G[i]);
    while(lmt<=n+m) lmt<<=,dgt++;
    for(int i=;i<lmt;i++) par[i]=(par[i>>]>>)|((i&)<<(dgt-));

    for(int i=;i<;i++) solve(a[i],b,p[i]);
    for(int i=;i<=n+m;i++)
    {
        ll M=p[]*p[];
        ll A=(mul(a[][i]*p[],inv(p[],p[]),M)+
             mul(a[][i]*p[],inv(p[],p[]),M))%M;
        ll K=mul(a[][i]-A,inv(M,p[]),p[]);
        res[i]=(mul(K,M,MOD)+A%MOD)%MOD;
    }
    for(int i=;i<=n+m;i++)
        printf("%lld ",res[i]);
    return ;
}

CRT写法

#include <bits/stdc++.h>

using namespace std;
#define X first
#define Y second
#define pb push_back
typedef double db;
typedef long long ll;
typedef pair<int,int> P;
const int MAXN=1e6+;
struct Complex
{
    db x,y;
    Complex(db a=,db b=){x=a;y=b;}
    Complex operator +(const Complex& rhs)
    {return Complex(x+rhs.x,y+rhs.y);}
    Complex operator -(const Complex& rhs)
    {return Complex(x-rhs.x,y-rhs.y);}
    Complex operator *(const Complex& rhs)
    {return Complex(x*rhs.x-y*rhs.y,x*rhs.y+y*rhs.x);}
}a[MAXN],b[MAXN],w[MAXN],t1[MAXN],t2[MAXN],t3[MAXN];
int n,m,MOD,lmt=,dgt,par[MAXN];ll x,res[MAXN];

void FFT(Complex *a,int flag)
{
    for(int i=;i<lmt;i++)
        if(i<par[i]) swap(a[i],a[par[i]]);
    for(int len=;len<lmt;len<<=)
        for(int st=;st<lmt;st+=(len<<))
        {
            int cur=;
            for(int k=st;k<st+len;k++)
            {
                Complex A=a[k],B=w[cur]*a[k+len];
                a[k]=A+B;a[k+len]=A-B;
                //预处理的写法
                cur=(cur+flag*lmt/(len<<)+lmt)&(lmt-);
            }
        }
    if(flag==-)
        for(int i=;i<lmt;i++)
            a[i].x=floor(a[i].x/lmt+0.5);
}
void solve()
{
    FFT(a,);FFT(b,);
    for(int i=;i<lmt;i++)
    {
        Complex d1,d2,d3,d4;
        int j=(lmt-i)&(lmt-);
        d1=(a[i]+Complex(a[j].x,-a[j].y))*Complex(0.5,);
        d2=(a[i]-Complex(a[j].x,-a[j].y))*Complex(,-0.5);
        d3=(b[i]+Complex(b[j].x,-b[j].y))*Complex(0.5,);
        d4=(b[i]-Complex(b[j].x,-b[j].y))*Complex(,-0.5);
        //必须先用临时变量存，因为后面还要用
        t1[i]=d1*d3;t2[i]=d1*d4+d2*d3;t3[i]=d2*d4;
    }
    for(int i=;i<lmt;i++)
        //充分利用虚部空间（可看成逆过程）
        b[i]=t2[i],a[i]=t1[i]+t3[i]*Complex(,);
    FFT(a,-);FFT(b,-);
    for(int i=;i<lmt;i++)
    {
        ll k1=(ll)a[i].x%MOD,k2=(ll)b[i].x%MOD;
        ll k3=(ll)floor(a[i].y/lmt+0.5)%MOD;
        res[i]=((k3<<)%MOD+(k2<<)%MOD+k1)%MOD;
    }
}

int main()
{
    scanf("%d%d%d",&n,&m,&MOD);
    for(int i=;i<=n;i++)
        scanf("%lld",&x),a[i]=Complex(x&,x>>);
    for(int i=;i<=m;i++)
        scanf("%lld",&x),b[i]=Complex(x&,x>>);
    while(lmt<=n+m) lmt<<=,dgt++;
    for(int i=;i<lmt;i++)
        par[i]=(par[i>>]>>)|((i&)<<(dgt-));
    for(int i=;i<lmt;i++)
        w[i]=Complex(cos(*M_PI*i/lmt),sin(*M_PI*i/lmt));

    solve();
    for(int i=;i<=n+m;i++)
        printf("%lld ",res[i]);
    return ;
}

拆系数4次DFT/IDFT

实测$myy$论文里拆系数优化至4次$DFT/IDFT$的方法比三模数$CRT$快7倍左右

7、SA

#include <bits/stdc++.h>

using namespace std;
#define X first
#define Y second
#define pb push_back
#define debug(x) cerr<<#x<<"="<<x<<endl
typedef double db;
typedef long long ll;
typedef pair<int,int> P;
const int MAXN=1e6+;
char s[MAXN];
int len,lmt,cnt[MAXN],sa[MAXN],x[MAXN],y[MAXN],cur;

void solve()
{
    for(int i=;i<=len;i++)
        cnt[x[i]=s[i]]++;
    for(int i=;i<=lmt;i++)
        cnt[i]+=cnt[i-];
    for(int i=len;i>=;i--)
        sa[cnt[x[i]]--]=i;

    for(int k=;k<=len;k<<=,lmt=cur)
    {
        cur=;
        for(int i=len-k+;i<=len;i++)
            y[++cur]=i;
        for(int i=;i<=len;i++)
            if(sa[i]>k) y[++cur]=sa[i]-k;

        for(int i=;i<=lmt;i++) cnt[i]=;
        for(int i=;i<=len;i++) cnt[x[i]]++;
        for(int i=;i<=lmt;i++) cnt[i]+=cnt[i-];
        //一定要按第二关键字从大往小枚举！
        for(int i=len;i>=;i--)
            sa[cnt[x[y[i]]]--]=y[i];

        swap(x,y);cur=;x[sa[]]=;
        for(int i=;i<=len;i++)
            x[sa[i]]=(y[sa[i]]==y[sa[i-]]&&y[sa[i]+k]==y[sa[i-]+k])?cur:++cur;
        if(cur==len) break;
    }
}

int main()
{
    scanf("%s",s+);
    len=strlen(s+);lmt=;
    solve();
    for(int i=;i<=len;i++)
        printf("%d ",sa[i]);
    return ;
}

SA