[POI2014]HOT-Hotels

题目

因为三个点肯定会有一个中转点，然后从中将一棵无根树转成有根树，然后再通过乘法原理（c2表示两两的乘积，若有第三个数就可以变成3数相乘，就可以乘法原理了）

为什么要搜索每颗子树而不从从自己的根开始搜了

以为这样就有可能不是中转点了

所以这样的话时间复杂的为O(N^2),但是可以使用长链剖分优化为O(N)，虽然我不知道怎么用

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
inline long long read()
{
    long long f=1,ans=0;char c;
    while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9'){ans=ans*10+c-'0';c=getchar();}
    return f*ans;
}
long long c1[10001],c2[10001],n;
struct node{
    long long u,v,nex;
}x[10001];
long long cnt=0;
long long ans[10001],head[5001],maxn,sum;
void add(long long u,long long v)
{
    x[cnt].u=u,x[cnt].v=v,x[cnt].nex=head[u],head[u]=cnt++;
}
void dfs(long long pos,long long fath,long long anss)
{
    maxn=max(maxn,anss);
    ans[anss]++;
    for(long long i=head[pos];i!=-1;i=x[i].nex)
    {
        if(x[i].v!=fath) dfs(x[i].v,pos,anss+1);
    }
}
int main()
{
    memset(head,-1,sizeof(head));
     n=read();
     for(long long i=1;i<n;i++) 
     {
         long long u=read(),v=read();
         add(u,v),add(v,u);
//         cout<<i<<endl;
     }
     
     for(long long i=1;i<=n;i++)
     {
         memset(c1,0,sizeof(c1));
         memset(c2,0,sizeof(c2));
         for(long long j=head[i];j!=-1;j=x[j].nex)
         {
             maxn=0;
             dfs(x[j].v,i,1);
             for(long long k=1;k<=maxn;k++)
             {
                 sum+=ans[k]*c2[k];
                 c2[k]+=c1[k]*ans[k];
                 c1[k]+=ans[k];
                 ans[k]=0;
             }
        }
     }
     cout<<sum;
}