CSU_WJJJ的ACM板子（持续更新）

我的ACM板子

基础算法

二分

// 满足ok(x)的最小x
int l=1,r=n;
while(l<r){
    int mid=l+r>>1;
    if(ok(mid)){
        r=mid;
    }else{
        l=mid+1;
    }
}
// 满足ok(x)的最大x
int l=1,r=n;
while(l<r){
    int mid=l+r+1>>1;
    if(ok(mid)){
        l=mid;
    }else{
        r=mid-1;
    }
}
//最后l==r即为答案 

离散化

int a[10050],b[10050];
vector<int>alls;
int findx(int x)
{
    return lower_bound(alls.begin(),alls.end(),x)-alls.begin()+1;
}
for(int i=1;i<=m;i++){
    cin>>a[i]>>b[i];
    alls.push_back(a[i]); alls.push_back(b[i]);
}
sort(alls.begin(),alls.end());
alls.erase(unique(alls.begin(),alls.end()),alls.end());

ST表

int n,m;
int a[100050];
int st[100050][100];
//st[i][j]为从位置i开始连续(1<<j)个数的最值 
void init()
{
	for(int i=1;i<=n;i++){
		st[i][0]=a[i];
	}
	for(int j=1;(1<<j)<=n;j++){
		for(int i=1;i+(1<<j)-1<=n;i++){
			st[i][j]=max(st[i][j-1],st[i+(1<<(j-1))][j-1]);
		}
	}
}
int search(int l,int r)
{
	int k=log2(r-l+1);
	return max(st[l][k],st[r-(1<<k)+1][k]);	
}

倍增

求LCA

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
#define endl '\n'
int n,m,s,head[500050];
int fa[500050];
struct edge{
    int nxt,to;
}e[1000050];
int cnt,dep[500050],p[500050][21];
void add(int u,int v)
{
    e[++cnt].to=v;
    e[cnt].nxt=head[u];
    head[u]=cnt;
}
void dfs(int u,int fa)
{
    dep[u]=dep[fa]+1;
    p[u][0]=fa;
    for(int i=1;(1<<i)<=dep[u];i++){
        p[u][i]=p[p[u][i-1]][i-1];
    }
    for(int i=head[u];i;i=e[i].nxt){
        int v=e[i].to;
        if(v!=fa){
            dfs(v,u);
        }
    }
}
int lca(int a,int b)
{
    if(dep[a]>dep[b]){
        swap(a,b);
    }
    for(int i=20;i>=0;i--){
        if(dep[a]<=dep[b]-(1<<i)){
            b=p[b][i];
        }
    }
    if(a==b) return a;
    for(int i=20;i>=0;i--){
        if(p[a][i]==p[b][i]){
            continue;
        }
        else{
            a=p[a][i],b=p[b][i];
        }
    }
    return p[a][0];
}
int main(){
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
cin>>n>>m>>s; // s为树根的序号
for(int i=1;i<n;i++){
    int x,y;
    cin>>x>>y;
    add(x,y); add(y,x);
}
dfs(s,0);
for(int i=1;i<=m;i++){
    int x,y;
    cin>>x>>y;
    cout<<lca(x,y)<<endl;
}

    return 0;
}

数据结构

并查集

int find(int x){
	return x==fa[x]?fa[x]:find(fa[x]);
}

int find(int x){
	if(x==fa[x]) return x;
    int rt=find(fa[x]);
    d[x]+=d[fa[x]];
    return fa[x]=rt;
}
//带权并查集 先递归 回溯的时候维护父子关系

树状数组

int lowbit(int x)
{
    return x&-x;
}
int ask(int x)
{
	int ans=0;
    for(;x;x-=lowbit(x)) ans+=c[x];
    return ans;
}
void add(int x,int y)
{
    for(;x<=n;x+=lowbit(x)) c[x]+=y;
}

单调队列

cin>>n>>k;
for(int i=1;i<=n;i++){
	cin>>a[i];
}
int head=1,tail=0;
for(int i=1;i<=n;i++){
	while(head<=tail&&que[tail]>=a[i]) tail--;
	que[++tail]=a[i]; pos[tail]=i;
	while(pos[head]<=i-k) head++;
	if(i>=k){
		cout<<que[head]<<endl;
	}
}

线段树

线段树的板子好丑啊。。要重新写

// 区间修改 区间查询
struct node{
	int l,r;
	ll sum,lazy;
	int length;
}tree[maxn<<2];
char ch[2];
int a[maxn];
int n,m,x,y,t;
void pushup(int k)
{
	tree[k].sum=tree[k<<1].sum+tree[k<<1|1].sum;
}
void pushdown(int k)
{
	if(tree[k].lazy){
		tree[k<<1].sum=tree[k<<1].sum+tree[k].lazy*tree[k<<1].length;
		tree[k<<1|1].sum=tree[k<<1|1].sum+tree[k].lazy*tree[k<<1|1].length;
		tree[k<<1].lazy+=tree[k].lazy;
		tree[k<<1|1].lazy+=tree[k].lazy;
		tree[k].lazy=0;
	}
}
void build(int k,int l,int r)
{
	tree[k].l=l;tree[k].r=r;tree[k].lazy=0;tree[k].length=r-l+1;
	if(l==r){
		tree[k].sum=a[l];
		return;
	}
	int mid=(l+r)>>1;
	build(k<<1,l,mid);
	build(k<<1|1,mid+1,r);
	pushup(k);
}
void update(int k,int l,int r,int lazy)
{
	if(tree[k].l==l&&tree[k].r==r){
		tree[k].sum+=tree[k].length*lazy;
		tree[k].lazy+=lazy;
		return;
	}
	pushdown(k);
	int mid=(tree[k].l+tree[k].r)>>1;
	if(mid>=r){
		update(k<<1,l,r,lazy);	
	}
	else if(l>mid){
		update(k<<1|1,l,r,lazy);
	}
	else{
		update(k<<1,l,mid,lazy);
		update(k<<1|1,mid+1,r,lazy);
	}
	pushup(k);
}
ll query(int k,int l,int r)
{
	if(tree[k].l==l&&tree[k].r==r){
		return tree[k].sum;
	}
	pushdown(k);
	ll res=0;
	int mid=(tree[k].l+tree[k].r)>>1;
	if(r<=mid){
		res+=query(2*k,l,r);
	}
	else if(l>mid){
		res+=query(2*k+1,l,r);
	}
	else{
		res+=query(2*k,l,mid);
		res+=query(2*k+1,mid+1,r);
	}
	return res;
}
int main(){
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
	scanf("%d",&a[i]);
}
memset(tree,0,sizeof(tree));
build(1,1,n);
for(int i=1;i<=m;i++){
	scanf("%s",ch);
	if(ch[0]=='Q'){
		scanf("%d%d",&x,&y);
		printf("%lld\n",query(1,x,y));
	}
	if(ch[0]=='C'){
		scanf("%d%d%d",&x,&y,&t);
		update(1,x,y,t);
	}
}
	return 0;
}

树链剖分

已经看不懂当时的板子了orz

struct edge{
	int to,nxt;
}e[M];
struct tree{
	int l,r,sum,maxm;
}t[N<<2];
int num,cnt,n,q;
int f[N],dep[N],siz[N],son[N],head[N];
int top[N],id[N],rk[N],w[N];
void add(int u,int v)
{
	e[++cnt].nxt=head[u];
	e[cnt].to=v;
	head[u]=cnt;
}
void dfs1(int u,int fa)
{
	f[u]=fa,dep[u]=dep[fa]+1,siz[u]=1;
	for(int i=head[u];i;i=e[i].nxt)
	{
		int v=e[i].to;
		if(v!=fa){
			dfs1(v,u);
			siz[u]+=siz[v];
			if(siz[son[u]]<siz[v]){
				son[u]=v;
			}
		}
	}
}
void dfs2(int p,int t)
{
	top[p]=t,id[p]=++num,rk[num]=p;
	if(son[p]) dfs2(son[p],t);
	for(int i=head[p];i;i=e[i].nxt){
		int v=e[i].to;
		if(v!=f[p]&&v!=son[p]){
			dfs2(v,v);
		}
	}
}
void pushup(int k)
{
	t[k].sum=t[ls].sum+t[rs].sum;
	t[k].maxm=max(t[ls].maxm,t[rs].maxm);
}
void build(int k,int l,int r)
{
	t[k].l=l; t[k].r=r;
	if(l==r){
		t[k].sum=t[k].maxm=w[rk[l]];
		return;
	}
	int mid=(l+r)>>1;
	build(ls,l,mid); build(rs,mid+1,r);
	pushup(k);
}
void change(int k,int d,int x)
{
	if(t[k].l==t[k].r&&t[k].r==d){
		t[k].maxm=x;
		t[k].sum=x;
		return;
	}
	int mid=(t[k].l+t[k].r)>>1;
	if(d<=mid){
		change(k<<1,d,x);
	}
	else{
		change(k<<1|1,d,x);
	}
	pushup(k);
}
int Querysum(int k,int l,int r)
{
	int ans=0;
	if(t[k].l>=l&&t[k].r<=r){
		return t[k].sum;
	}
	int mid=(t[k].l+t[k].r)>>1;
	if(l<=mid){
		ans+=Querysum(ls,l,r);
	}
	if(r>mid){
		ans+=Querysum(rs,l,r);
	}
	return ans;
}
int Querymax(int k,int l,int r)
{
	int res=-inf;
	if(t[k].l>=l&&t[k].r<=r){
		return t[k].maxm;
	}
	int mid=(t[k].l+t[k].r)>>1;
	if(l<=mid){
		res=max(res,Querymax(ls,l,r));
	}
	if(r>mid){
		res=max(res,Querymax(rs,l,r));
	}
	return res;
}
int querysum(int x,int y)
{
	int res=0;
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]]){
			swap(x,y);
		}
		res+=Querysum(1,id[top[x]],id[x]);
		x=f[top[x]];
	}
	if(dep[x]>dep[y]){
		swap(x,y);
	}
	res+=Querysum(1,id[x],id[y]);
	return res;
}
int querymax(int x,int y)
{
	int res=-inf;
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]]){
			swap(x,y);
		}
		res=max(res,Querymax(1,id[top[x]],id[x]));
		x=f[top[x]];
	}
	if(dep[x]>dep[y]){
		swap(x,y);
	}
	return res=max(res,Querymax(1,id[x],id[y]));
}
int main(){
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
cin>>n;
for(int i=1;i<=n-1;i++){
	int x,y;
	cin>>x>>y;
	add(x,y); add(y,x);
}
for(int i=1;i<=n;i++){
	cin>>w[i];
}
dfs1(1,1);
dfs2(1,1);
build(1,1,n);
cin>>q;
for(int i=1;i<=q;i++){
	string s;
	int x,y;
	cin>>s>>x>>y;
	if(s[1]=='S'){
		cout<<querysum(x,y)<<endl;
	}
	else if(s[1]=='M'){
		cout<<querymax(x,y)<<endl;
	}
	else if(s[1]=='H'){ 
		change(1,id[x],y);
	}
}
	return 0;
}

树上启发式合并

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
#define endl '\n'
struct edge{
    int nxt,to;
}e[2000050];
int ans[1000050],mx,anss;
int head[1000050],ct,sz[1000050],son[1000050],dep[1000050],cnt[1000050];
void add(int u,int v)
{
    e[++ct].to=v;
    e[ct].nxt=head[u];
    head[u]=ct;
}
void dfs1(int u,int fa)
{
    sz[u]=1; dep[u]=dep[fa]+1;
    for(int i=head[u];i;i=e[i].nxt){
        int v=e[i].to;
        if(v!=fa){
            dfs1(v,u); sz[u]+=sz[v];
            if(sz[v]>sz[son[u]]){
                son[u]=v;
            }
        }
    }
}
void change(int u,int fa,int val,int p)
{
    cnt[dep[u]]+=val;
    if(cnt[dep[u]]>mx){
        mx=cnt[dep[u]]; anss=dep[u];
    }
    else if(cnt[dep[u]]==mx){
        anss=min(anss,dep[u]);
    }
    for(int i=head[u];i;i=e[i].nxt){
        int v=e[i].to;
        if(v==fa||v==p){
            continue;
        }
        change(v,u,val,p);
    }
}
void dfs2(int u,int fa,int op)
{
    for(int i=head[u];i;i=e[i].nxt){
        int v=e[i].to;
        if(v==fa||v==son[u]) continue;
        dfs2(v,u,0);
    }
    if(son[u]) dfs2(son[u],u,1);
    change(u,fa,1,son[u]);
    ans[u]=anss;
    if(op==0){
        change(u,fa,-1,0); mx=anss=0;
    }
}
int main(){
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
int n;cin>>n;
for(int i=1;i<n;i++){
    int x,y; cin>>x>>y;
    add(x,y); add(y,x);
}
dfs1(1,0); dfs2(1,0,0);
for(int i=1;i<=n;i++){
    cout<<ans[i]-dep[i]<<endl;
}
    return 0;
}

杂项

莫队

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
#define endl '\n'
int n,m,a[2000050],pos[2000050],ans,cnt[2000050];
struct node{
    int l,r,id,anss;
}q[2000050];
bool cmp(node a,node b)
{
    return (pos[a.l]^pos[b.l])?pos[a.l]<pos[b.l]:a.r<b.r;
}
void add(int x)
{
    if(!cnt[a[x]]) ans++;
    cnt[a[x]]++;
}
void del(int x)
{
    cnt[a[x]]--;
    if(!cnt[a[x]]) ans--;
}
int main(){
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
cin>>n; for(int i=1;i<=n;i++) cin>>a[i];
cin>>m;
for(int i=1;i<=m;i++){
    int x,y; cin>>q[i].l>>q[i].r; q[i].id=i;
}
int block=sqrt(n);
for(int i=1;i<=n;i++) pos[i]=(i-1)/block+1;
sort(q+1,q+1+m,cmp);
int l=1,r=0;
for(int i=1;i<=m;i++){
    while(l<q[i].l) del(l++);
    while(l>q[i].l) add(--l);
    while(r>q[i].r) del(r--);
    while(r<q[i].r) add(++r);
    q[q[i].id].anss=ans;
}
for(int i=1;i<=m;i++){
    cout<<q[i].anss<<endl;
}
    return 0;
}