BZOJ 3522 Hotel

暴力。好像很慢。。。

反正能过。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxv 5050
#define maxe 10050
using namespace std;
int n,x,y,g[maxv],nume=0,dis[maxv],cnt[maxv];
long long val2[maxv],val3[maxv],mx=0,ans=0;
struct edge
{
    int v,nxt;
}e[maxe];
void addedge(int u,int v)
{
    e[++nume].v=v;
    e[nume].nxt=g[u];
    g[u]=nume;
}
void dfs(int x,int fath)
{
    for (int i=g[x];i;i=e[i].nxt)
    {
        int v=e[i].v;
        if (v==fath) continue;
        dis[v]=dis[x]+1;cnt[dis[v]]++;mx=max(mx,1LL*dis[v]);
        dfs(v,x);
    }
}
int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n-1;i++)
    {
        scanf("%d%d",&x,&y);
        addedge(x,y);addedge(y,x);
    }
    for (int x=1;x<=n;x++)
    {
        memset(val2,0,sizeof(val2));
        memset(val3,0,sizeof(val3));
        for (int i=g[x];i;i=e[i].nxt)
        {
            int v=e[i].v;mx=1;dis[v]=1;
            memset(cnt,0,sizeof(cnt));cnt[1]=1;
            dfs(v,x);
            for (int j=1;j<=mx;j++)
            {
                ans+=val2[j]*cnt[j];
                val2[j]+=val3[j]*cnt[j];
                val3[j]+=cnt[j];
            }
        }
    }
    printf("%lld\n",ans);
    return 0;
}