标签：长链 剖分 int len mxn bzoj4543 define zp zs

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

像这种题就得耐心把式子列出来，写出来也不多。为什么某题解用的普通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;
}

标签：长链,剖分,int,len,mxn,bzoj4543,define,zp,zs
来源： https://www.cnblogs.com/happyguy/p/14614393.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。