BZOJ 3522 DFS+DP

思路：

f[]表示选1个点的

g[]表示选2个点的

dp一下

            ans+=(ll)g[k]*deep[k];
            g[k]+=(ll)f[k]*deep[k];
            f[k]+=deep[k];

听说有O(n)做法但是我不会

//By SiriusRen
#include <cstdio>
#include <cstring>
using namespace std;
const int N=10050;
int n,xx,yy,first[N],next[N],v[N],tot,deep[N],f[N],g[N];
long long ans;
typedef long long ll;
void add(int x,int y){v[tot]=y,next[tot]=first[x],first[x]=tot++;}
void dfs(int x,int fa,int dep){
    deep[dep]++;
    for(int i=first[x];~i;i=next[i])
        if(v[i]!=fa)dfs(v[i],x,dep+1);
}
int main(){
    memset(first,-1,sizeof(first));
    scanf("%d",&n);
    for(int i=1;i!=n;i++){
        scanf("%d%d",&xx,&yy);
        add(xx,yy),add(yy,xx);
    }
    for(int i=1;i<=n;i++){
        memset(f,0,sizeof(f)),memset(g,0,sizeof(g));
        for(int j=first[i];~j;j=next[j]){
            memset(deep,0,sizeof(deep));
            dfs(v[j],i,1);
            for(int k=1;k<=n;k++){
                ans+=(ll)g[k]*deep[k];
                g[k]+=(ll)f[k]*deep[k];
                f[k]+=deep[k];
            }
        }
    }
    printf("%lld\n",ans);
}