题解: $ n<=5000 $ 那我们考虑暴力枚举根
$ dp[x] $表示任意两个属于不同根儿子子树且深度为x个数的乘积和
$ sum[x] $表示不同根儿子子树 深度为x的个数和
然后统计方案就行了
#include <algorithm> #include <iostream> #include <cstring> #include <cstdio> #include <vector> #include <stack> #include <queue> #include <cmath> #include <set> #include <map> #define mp make_pair #define pb push_back #define pii pair<int,int> #define link(x) for(edge *j=h[x];j;j=j->next) #define inc(i,l,r) for(int i=l;i<=r;i++) #define dec(i,r,l) for(int i=r;i>=l;i--) const int MAXN=5e3+10; const double eps=1e-8; #define ll long long using namespace std; struct edge{int t;edge*next;}e[MAXN<<1],*h[MAXN],*o=e; void add(int x,int y){o->t=y;o->next=h[x];h[x]=o++;} ll read(){ ll x=0,f=1;char ch=getchar(); while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();} while(isdigit(ch))x=x*10+ch-'0',ch=getchar(); return x*f; } int n; ll dp[MAXN],sum[MAXN],num[MAXN],ans; int maxx; bool flag; void dfs(int x,int pre,int deep){ maxx=max(maxx,deep+1); num[deep+1]++; if(flag)ans+=dp[deep+1]; link(x){ if(j->t==pre)continue; dfs(j->t,x,deep+1); } } int main(){ n=read(); int x,y; inc(i,2,n)x=read(),y=read(),add(x,y),add(y,x); inc(i,1,n){ int cnt=0; link(i){ cnt++; if(cnt>=3)flag=1; maxx=0; dfs(j->t,i,1); inc(j,1,maxx)dp[j]+=sum[j]*num[j],sum[j]+=num[j],num[j]=0; } inc(j,1,n)dp[j]=sum[j]=num[j]=0; } printf("%lld\n",ans); }
3522: [Poi2014]Hotel
Time Limit: 20 Sec Memory Limit: 128 MBSubmit: 1250 Solved: 690
[Submit][Status][Discuss]
Description
有一个树形结构的宾馆,n个房间,n-1条无向边,每条边的长度相同,任意两个房间可以相互到达。吉丽要给他的三个妹子各开(一个)房(间)。三个妹子住的房间要互不相同(否则要打起来了),为了让吉丽满意,你需要让三个房间两两距离相同。
有多少种方案能让吉丽满意?
Input
第一行一个数n。
接下来n-1行,每行两个数x,y,表示x和y之间有一条边相连。
Output
让吉丽满意的方案数。
Sample Input
71 2
5 7
2 5
2 3
5 6
4 5