【题解】
基础的点分治题
还是分为路径是否经过root来讨论,计算点到根的距离时,一边计算,一边统计距离对3取模等于0,1,2的点的个数注意:统计答案时(a,a)算1次,(a,b)算2次
【代码】
#include<stdio.h> #include<stdlib.h> int v[40005]={0},w[40005]={0},first[20005]={0},next[40005]={0}; int size[20005]={0},maxsize[20005]={0},vis[20005]={0},d[20005]={0},cnt[5]; int e=0,root=0,sum,ans=0; int max(int a,int b) { if(a>b) return a; return b; } void tj(int x,int y,int z) { v[++e]=y; w[e]=z; next[e]=first[x]; first[x]=e; } void getroot(int x,int fa) { int i; size[x]=1; maxsize[x]=0; for(i=first[x];i!=0;i=next[i]) if(v[i]!=fa&&vis[v[i]]==0) { getroot(v[i],x); size[x]+=size[v[i]]; maxsize[x]=max(maxsize[x],size[v[i]]); } maxsize[x]=max(maxsize[x],sum-size[x]); if(root==0||maxsize[root]>maxsize[x]) root=x; } void getd(int x,int fa) { int i; cnt[d[x]%3]++; for(i=first[x];i!=0;i=next[i]) if(v[i]!=fa&&vis[v[i]]==0) { d[v[i]]=d[x]+w[i]; getd(v[i],x); } } int getcnt(int x,int now) { int i; d[x]=now; cnt[0]=cnt[1]=cnt[2]=0; getd(x,0); return cnt[0]*(cnt[0]-1)+cnt[1]*cnt[2]*2; } void work(int x) { int i; vis[x]=1; ans+=getcnt(x,0); for(i=first[x];i!=0;i=next[i]) if(vis[v[i]]==0) { ans-=getcnt(v[i],w[i]); root=0; sum=size[v[i]]; getroot(v[i],x); work(root); } } int gcd(int a,int b) { if(b==0) return a; return gcd(b,a%b); } int main() { int n,i,x,y,z; scanf("%d",&n); for(i=1;i<n;i++) { scanf("%d%d%d",&x,&y,&z); tj(x,y,z); tj(y,x,z); } sum=n; getroot(1,0); work(root); ans+=n; printf("%d/%d",ans/gcd(ans,n*n),n*n/gcd(ans,n*n)); return 0; }