bzoj4543 长链剖分

为啥我写的都是模板题题解啊

像这种题就得耐心把式子列出来,写出来也不多。为什么某题解用的普通dsu误导人啊,不用长剖写这题的都是毒瘤。

#include<bits/stdc++.h>
using namespace std;
#define forg(i,x) for(register int i=fir[x];i;i=nxt[i])
#define uu unsigned
#define scanf a1234=scanf
#define rint register int
#define fre(x) freopen(#x".in","r",stdin),freopen(#x".out","w",stdout)
typedef long long ll;
typedef uu long long ull;
typedef pair<int,int> pii;
int a1234;
inline int rd(int l,int r){return rand()%(r-l+1)+l;}

const int mxn=1e6+3;
int to[mxn*2],nxt[mxn*2],fir[mxn],gn=1,n;
inline void gadd(int x,int y){to[++gn]=y,nxt[gn]=fir[x],fir[x]=gn;}
ll ans;
int len[mxn],zs[mxn];
ll zq[mxn*4],*zp=zq,*f[mxn],*g[mxn];
inline void dfs0(int x,int f){
    forg(i,x)if(to[i]!=f){
        dfs0(to[i],x);if(len[to[i]]>len[zs[x]])zs[x]=to[i];
    }
    len[x]=len[zs[x]]+1;
}
inline void dfs(int x,int fa){
    if(zs[x])f[zs[x]]=f[x]-1,g[zs[x]]=g[x]+1,dfs(zs[x],x),ans+=f[x][0];
    g[x][0]=1;
    
    forg(i,x)if(to[i]!=fa&&to[i]!=zs[x]){
        int v=to[i];
        g[v]=zp,zp+=len[v]*2,f[v]=zp,zp+=len[v]*2;
        dfs(v,x);
        for(int j=0;j<len[v];++j)ans+=g[v][j]*f[x][j+1]+f[v][j+1]*g[x][j];
        for(int j=1;j<len[v];++j)f[x][j-1]+=f[v][j];
        for(int j=0;j<len[v];++j)f[x][j+1]+=g[v][j]*g[x][j+1];
        for(int j=0;j<len[v];++j)g[x][j+1]+=g[v][j];
    }
}

int main(){
    scanf("%d",&n);for(int i=1,u,v;i<n;++i)scanf("%d%d",&u,&v),gadd(u,v),gadd(v,u);
    dfs0(1,0);
    g[1]=zp,zp+=len[1]*2,f[1]=zp,zp+=len[1]*2;
    dfs(1,0);
    printf("%lld\n",ans);
    return 0;
}

上一篇:redis6.0.9配置ssl认证


下一篇:函数默认值的取值