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