bzoj3522 [Poi2014]Hotel

题目链接

在黄学长博客上居然分类树形DP？

暴力，以每个点为根计算子树内答案，推推式子就维护两个S就可以啦

#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<string>
#include<cmath>
#include<ctime>
#include<queue>
#include<stack>
#include<map>
#include<set>
#define rre(i,r,l) for(int i=(r);i>=(l);i--)
#define re(i,l,r) for(int i=(l);i<=(r);i++)
#define Clear(a,b) memset(a,b,sizeof(a))
#define inout(x) printf("%d",(x))
#define douin(x) scanf("%lf",&x)
#define strin(x) scanf("%s",(x))
#define LLin(x) scanf("%lld",&x)
#define op operator
#define CSC main
typedef unsigned long long ULL;
typedef const int cint;
typedef long long LL;
using namespace std;
void inin(int &ret)
{
    ret=0;int f=0;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=1;ch=getchar();}
    while(ch>='0'&&ch<='9')ret*=10,ret+=ch-'0',ch=getchar();
    ret=f?-ret:ret;
}
int n,head[5050],next[10010],zhi[10010],ed,shen[5050],sum[5050],Max;
LL s1[5050],s2[5050],ans;
void add(int a,int b)
{
    next[++ed]=head[a],head[a]=ed,zhi[ed]=b;
    next[++ed]=head[b],head[b]=ed,zhi[ed]=a;
}
void dfs(int x,int fa)
{
    sum[shen[x]]++;Max=max(Max,shen[x]);
    for(int i=head[x];i;i=next[i])if(zhi[i]!=fa)
        shen[zhi[i]]=shen[x]+1,dfs(zhi[i],x);
}
int main()
{
    inin(n);
    re(i,2,n)
    {
        int q,w;
        inin(q),inin(w);
        add(q,w);
    }
    re(i,1,n)
    {
        Clear(s1,0),Clear(s2,0);Max=0;
        for(int j=head[i];j;j=next[j])
        {
            Clear(sum,0);
            shen[zhi[j]]=1;
            dfs(zhi[j],i);
            re(k,1,Max)
            {
                ans+=(LL)sum[k]*s1[k];
                s1[k]+=sum[k]*s2[k];
                s2[k]+=sum[k];
            }
        }
    }
    printf("%lld",ans);
     return 0;
}