bzoj 3522: [Poi2014]Hotel

呵呵，一开始天真的我以为求个 西格玛 C（？？，3）就好了。。

（题解：比枚举2个数的再多一个，，一样搞）

 #include <bits/stdc++.h>
 #define LL long long
 #define lowbit(x) x&(-x)
 #define inf 1e15
 using namespace std;
 inline int ra()
 {
     int x=,f=; char ch=getchar();
     while (ch<'' || ch>'') {if (ch=='-') f=-; ch=getchar();}
     while (ch>='' && ch<='') {x=x*+ch-''; ch=getchar();}
     return x*f;
 }
 struct edge{int next,to;}e[];
 int tot[],mx,n,head[],cnt,s1[],s2[];
 LL ans;
 void insert(int x, int y){e[++cnt].next=head[x]; e[cnt].to=y; head[x]=cnt;}
 void dfs(int x, int fa, int deep)
 {
     tot[deep]++;
     mx=max(mx,deep);
     for (int i=head[x];i;i=e[i].next)
     {
         if (e[i].to==fa) continue;
         dfs(e[i].to,x,deep+);
     }
 }
 int main(int argc, char const *argv[])
 {
     n=ra();
     for (int i=; i<n; i++)
     {
         int x=ra(),y=ra();
         insert(x,y); insert(y,x);
     }
     for (int x=; x<=n; x++)
     {
         memset(s1,,sizeof(s1));
         memset(s2,,sizeof(s2));
         for (int i=head[x];i;i=e[i].next)
         {
             dfs(e[i].to,x,);
             for (int j=; j<=mx; j++)
             {
                 ans+=s2[j]*tot[j];
                 s2[j]+=s1[j]*tot[j];
                 s1[j]+=tot[j];
             }
             for (int j=; j<=mx; j++) tot[j]=;
         }
     }
     cout<<ans;
     return ;
 }