[杂题] BZOJ3522: [Poi2014]Hotel

这题不要想的太复杂，要暴力一点。不需要定树根搞，如果是有根树的话，发现沿父边往上走的点很难统计。
所以就直接枚举中间点，把他作为根，相当于求在不同子树中选 3 <script type="math/tex" id="MathJax-Element-19">3</script> 个深度相同的点的方案数。直接暴力便利一遍图，统计一下就好了。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int maxn=5005,maxe=10005;
int n,mdep;
int fir[maxn],nxt[maxe],son[maxe],tot;
int dep[5005],cnt[5005];
LL s1[5005],s2[5005],ans;
void add(int x,int y){
    son[++tot]=y; nxt[tot]=fir[x]; fir[x]=tot;
}
void dfs(int x,int pre){
    mdep=max(dep[x],mdep); cnt[dep[x]]++;
    for(int j=fir[x];j;j=nxt[j]) if(son[j]!=pre) dep[son[j]]=dep[x]+1, dfs(son[j],x);   
}
int main(){
    freopen("bzoj3522.in","r",stdin);
    freopen("_bzoj3522.out","w",stdout); 
    scanf("%d",&n);
    for(int i=1;i<=n-1;i++){
        int x,y; scanf("%d%d",&x,&y);
        add(x,y); add(y,x);
    }
    for(int x=1;x<=n;x++){
        memset(s1,0,sizeof(s1)); memset(s2,0,sizeof(s2));
        for(int j=fir[x];j;j=nxt[j]){
            dep[son[j]]=1; dfs(son[j],x);
            for(int j=1;j<=mdep;j++) ans+=s2[j]*cnt[j], s2[j]+=cnt[j]*s1[j], s1[j]+=cnt[j];
            for(int j=1;j<=mdep;j++) cnt[j]=0;
        }
    }
    printf("%lld\n",ans);
    return 0;
}