3522: [Poi2014]Hotel
Time Limit: 20 Sec Memory Limit: 128 MB
Submit: 278 Solved: 136
[Submit][Status][Discuss]
Description
有一个树形结构的宾馆,n个房间,n-1条无向边,每条边的长度相同,任意两个房间可以相互到达。吉丽要给他的三个妹子各开(一个)房(间)。三个妹子住的房间要互不相同(否则要打起来了),为了让吉丽满意,你需要让三个房间两两距离相同。
有多少种方案能让吉丽满意?
Input
第一行一个数n。
接下来n-1行,每行两个数x,y,表示x和y之间有一条边相连。
Output
让吉丽满意的方案数。
Sample Input
7
1 2
5 7
2 5
2 3
5 6
4 5
Sample Output
5
HINT
【样例解释】
{1,3,5},{2,4,6},{2,4,7},{2,6,7},{4,6,7}
【数据范围】
n≤5000
Source
暴力枚举中点,乘法原理。t1,t2过渡一下。
#include<iostream> #include<cstdio> #include<cstdlib> #include<cstring> #include<algorithm> #include<cmath> #include<vector> #include<queue> #define ll long long using namespace std; int next[10005],list[10005],head[5005],num[5005]; ll sum,t1[5005],t2[5005]; int tot,n,x,y,mx; void insert(int x,int y) { next[++tot]=head[x]; head[x]=tot; list[tot]=y; } void dfs(int x,int fa,int deep) { mx=max(mx,deep); num[deep]++; for (int i=head[x];i;i=next[i]) if (list[i]!=fa) dfs(list[i],x,deep+1); } int main() { scanf("%d",&n); for (int i=1;i<n;i++) { scanf("%d%d",&x,&y); insert(x,y); insert(y,x); } for (int i=1;i<=n;i++) { memset(t1,0,sizeof(t1)); memset(t2,0,sizeof(t2)); for (int j=head[i];j;j=next[j]) { memset(num,0,sizeof(num)); dfs(list[j],i,1); for (int k=1;k<=mx;k++) { sum+=t2[k]*num[k]; t2[k]+=num[k]*t1[k]; t1[k]+=num[k]; } } } printf("%lld",sum); return 0; }
时间: 2024-11-10 15:53:25