记录编号 483755 评测结果 AAAAAAAAAAAAAAAAAAAA
题目名称 [国家集训队2011] 聪聪可可 最终得分 100
用户昵称 GravatarShirry 是否通过 通过
代码语言 C++ 运行时间 0.726 s
提交时间 2018-01-19 07:11:45 内存使用 14.42 MiB
显示代码纯文本
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=20010;
struct poi{int v,w;};
vector<poi>p[maxn];
int n,root,sum,ans,f[maxn],d[maxn],t[5],son[maxn],vis[maxn];
int gcd(int a,int b){return !b?a:gcd(b,a%b);}
void getroot(int x,int fa){
	son[x]=1,f[x]=0;
	for(int i=0;i<(int)p[x].size();i++){
		int u=p[x][i].v;
		if(u==fa||vis[u])continue;
		getroot(u,x);
		son[x]+=son[u];
		f[x]=max(f[x],son[u]);
	}
	f[x]=max(f[x],sum-son[x]);
	if(f[x]<f[root])root=x;
}
void getdeep(int x,int fa){
	t[d[x]]++;
	for(int i=0;i<(int)p[x].size();i++){
		int u=p[x][i].v;
		if(u==fa||vis[u])continue;
		d[u]=(d[x]+p[x][i].w)%3;
		getdeep(u,x);
	}
}
int cal(int x,int v){
	t[0]=t[1]=t[2]=0,d[x]=v;
	getdeep(x,0);
	return t[1]*t[2]*2+t[0]*t[0];
}
void solve(int x){
	ans+=cal(x,0),vis[x]=1;
	for(int i=0;i<(int)p[x].size();i++){
		int u=p[x][i].v;
		if(vis[u])continue;
		ans-=cal(u,p[x][i].w);
		root=0,sum=son[u];
		getroot(u,0);
		solve(root);
	}
}
int main(){
	freopen("cckk.in","r",stdin);
	freopen("cckk.out","w",stdout);
	scanf("%d",&n);
	int x,y,v;
	for(int i=1;i<n;i++){
		scanf("%d%d%d",&x,&y,&v);
		p[x].push_back((poi){y,v%3});
		p[y].push_back((poi){x,v%3});
	}
	sum=n;f[0]=n;
	getroot(1,0);
	solve(root);
	int t=gcd(ans,n*n);
	printf("%d/%d\n",ans/t,n*n/t);
	return 0;
}