进食后入+疑问

f[v]=new ll[len[v]]();

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+10;
int n;
vector<int>e[N];
int len[N],son[N];
void dfs1(int u,int fa){
    len[u]=1;
    for(int& v:e[u]){
        if(v==fa)continue;
        dfs1(v,u);
        if(len[v]>len[son[u]]){
            len[u]=len[v]+1;
            son[u]=v;
        }
    }
}
ll ans=0;
ll *f[N],*g[N];
void tdp1(int u,int fa){
    if(son[u]){
        f[son[u]]=f[u]+1;
        g[son[u]]=g[u]-1;
        tdp1(son[u],u);
    }
    f[u][0]=1;ans+=g[u][0];
    for(int& v:e[u]){
        if(v==fa||v==son[u])continue;
        f[v]=new ll[len[v]]();
        g[v]=new ll[len[v]<<1]();
        g[v]+=len[v];
        tdp1(v,u);
        for(int i=0;i<len[v];i++){
            if(i)ans+=f[u][i-1]*g[v][i];
            ans+=f[v][i]*g[u][i+1];
        }
        for(int i=0;i<len[v];i++){
            g[u][i+1]+=f[u][i+1]*f[v][i];
            if(i)g[u][i-1]+=g[v][i];
            f[u][i+1]+=f[v][i];
        }
    }
}
int main(){
    ios::sync_with_stdio(0);
    cin.tie(0);
    cin>>n;
    int u,v;
    for(int i=1;i<n;i++){
        cin>>u>>v;
        e[u].push_back(v);
        e[v].push_back(u);
    }
    dfs1(1,0);
    f[1]=new ll[len[1]]();
    g[1]=new ll[len[1]<<1]();
    g[1]+=len[1];
    tdp1(1,0);
    cout<<ans<<'\n';
    return 0;
}