记录编号 446315 评测结果 AAAAAAAAAA
题目名称 不平凡的许愿树 最终得分 100
用户昵称 GravatarShirry 是否通过 通过
代码语言 C++ 运行时间 1.137 s
提交时间 2017-09-07 21:33:30 内存使用 0.59 MiB
显示代码纯文本
#include<cstdio>
#include<vector>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn=5010;
vector<int>p[maxn];
int n,md;
long long ans,h[maxn],f[maxn],g[maxn],t[maxn],dep[maxn],vis[maxn];
void Dep(int x,int d){
	md=max(md,d),vis[x]=1,f[d]++;
	for(int i=0;i<(int)p[x].size();i++){
		int u=p[x][i];
		if(!vis[u])Dep(u,d+1);
		if(d==0){
			for(int i=1;i<=md;i++){
				ans+=(f[i]*h[i]);
				h[i]+=(f[i]*g[i]);
				g[i]+=f[i];
			}
			md=0;memset(f,0,sizeof(f));
		}
	}
}
int main(){
	freopen("hopetree.in","r",stdin);
	freopen("hopetree.out","w",stdout);
	scanf("%d",&n);
	int u,v;
	for(int i=1;i<n;i++)scanf("%d%d",&u,&v),p[u].push_back(v),p[v].push_back(u);
	for(int i=1;i<=n;i++){
		memset(g,0,sizeof(g));
		memset(f,0,sizeof(f));
		memset(h,0,sizeof(h));
		memset(vis,0,sizeof(vis));
		Dep(i,0);
	}
	printf("%lld %lld",ans%338+1,(ans+233)%338+1);
	return 0;
}