模板合集

Posted 2021-01-11 ngualexzhang

第一个堆模板

二叉堆性质优先级最大的在堆顶

插入操作是开出一个空间然后从堆底切换到一个合适的位置

删除操作就是把最下面的和最上面的换位置，再进行调整

有stl，但还是了解一下原理为好

#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
int heap[1000005],n;
int siz;
void push(int x){
    heap[++siz]=x;
    int now=siz;
    while(now){
        int nxt=now>>1;
        if(heap[nxt]>heap[now]) swap(heap[nxt],heap[now]);
        else break;
        now=nxt;
    }
    return;
}
void pop(){
    swap(heap[siz],heap[1]);siz--;
    int now=1;
    while((now<<1)<=siz){
        int nxt=now<<1;
        if(nxt+1<=siz&&heap[nxt+1]<heap[nxt]) nxt++;
        if(heap[nxt]<heap[now]) swap(heap[nxt],heap[now]);
        else break;
        now=nxt;
    }
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        int opt,x;
        scanf("%d",&opt);
        if(opt==1){
            scanf("%d",&x);
            push(x);
        }
        if(opt==3){
            pop();
        }
        if(opt==2) printf("%d
",heap[1]);
    }
    return 0;
}

 

第二个快排

系统也为其提供了函数

知其然，知其所以然

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int n,a[100005];
void q_sort(int l,int r){
    int i=l,j=r;
    int mid=a[(l+r)>>1];
    if(i>=j)
        return;
    while(i<=j){
        while(a[j]>mid) j--;
        while(a[i]<mid) i++;
        if(i<=j){
            swap(a[i],a[j]);
            i++,j--;
        }
    }
    if(l<j) q_sort(l,j);
    if(i<r) q_sort(i,r);
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    q_sort(1,n);
    for(int i=1;i<=n;i++)
        printf("%d ",a[i]);
    return 0;
}

第三个ST表模板

利用的倍增法

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
int n,m,maxn[100005][21];
int query(int l,int r){
    int k=log2(r-l+1);
    return max(maxn[l][k],maxn[r-(1<<k)+1][k]);
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%d",&maxn[i][0]);
    for(int i=1;i<=21;i++)
        for(int j=1;j+(1<<i)-1<=n;j++)
                maxn[j][i]=max(maxn[j][i-1],maxn[j+(1<<(i-1))][i-1]);
    while(m--){
        int l,r;
        scanf("%d%d",&l,&r);
        printf("%d
",query(l,r));
    }
    return 0;
}

第四个lca最近公共祖先模板

也是利用了倍增法

#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
int n,m,s,head[500005],nxt[1200000],to[1200000],ce,dep[500005],fa[500005][22],lg[500005];
void add(int u,int v){
    to[++ce]=v;nxt[ce]=head[u];head[u]=ce;
}
void dfs(int now,int fat){
    dep[now]=dep[fat]+1;
    fa[now][0]=fat;
    for(int i=1;(1<<i)<=dep[now];i++)
        fa[now][i]=fa[fa[now][i-1]][i-1];
    for(int i=head[now];i;i=nxt[i])
        if(to[i]!=fat)
            dfs(to[i],now);
}
int lca(int x,int y){
    if(dep[x]<dep[y])
        swap(x,y);
    while(dep[x]>dep[y])
        x=fa[x][lg[dep[x]-dep[y]]-1];
    if(x==y) return x;
    for(int i=lg[dep[x]]-1;i>=0;i--)
        if(fa[x][i]!=fa[y][i])
            x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
int main(){
    scanf("%d%d%d",&n,&m,&s);
    for(int i=1;i<n;i++){
        int a,b;
        scanf("%d%d",&a,&b);
        add(a,b);add(b,a);
    }
    dfs(s,0);
    for(int i=1;i<=n;i++)
        lg[i]=lg[i-1]+((1<<lg[i-1])==i);
    for(int i=1;i<=m;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        printf("%d
",lca(x,y));    
    }
    return 0;
}