直接树形DP搞一搞,也可以点分治,然而有
O(n)
复杂度的树形DP为什么还要点分治呢QAQ
dis[x][i]
表示在以
x
为根的子树中,到
发现距离直接对
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<iostream>
#include<algorithm>
#define N 400022
#define M 1000002
using namespace std;
int sc()
{
int i=0,f=1; char c=getchar();
while(c>'9'||c<'0'){if(c=='-')f=-1;c=getchar();}
while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();
return i*f;
}
int head[N],nxt[N],lst[N],v[N];
int n,dis[N][3],sum,tot;
int gcd(int x,int y){return x==0?y:gcd(y%x,x);}
void insert(int x,int y,int z)
{
lst[++tot]=y;nxt[tot]=head[x];head[x]=tot;v[tot]=z;
lst[++tot]=x;nxt[tot]=head[y];head[y]=tot;v[tot]=z;
}
void dfs(int x,int f)
{
dis[x][0]=1;
for(int i=head[x];i;i=nxt[i])
if(lst[i]!=f)
{
dfs(lst[i],x);
for(int j=0;j<3;j++)
dis[x][(j+v[i])%3]+=dis[lst[i]][j];
}
sum+=dis[x][0];
for(int i=head[x];i;i=nxt[i])
if(lst[i]!=f)
for(int j=0;j<3;j++)
sum+=(dis[x][j]-dis[lst[i]][(j-v[i]+6)%3])*dis[lst[i]][(9-j-v[i])%3];
}
int main()
{
n=sc();
for(int i=1;i<n;i++)
{
int x=sc(),y=sc(),z=sc()%3;
insert(x,y,z);
}
dfs(1,0);
int ans=n*n,g=gcd(sum,ans);
printf("%d/%d",sum/g,ans/g);
return 0;
}