bzoj2152: 聪聪可可

这题显然点分治DP？我不会告诉你我还不会点分治）

先O(n)求Dfs序，然后倒序dp[i][j]表示i及i的子树中的点到i距离%3为j的方案数。

最后对每个点，O(1)计算经过它的路径为3的倍数的方案数，得到Ans。

求Ans与n*n的Gcd，约分即可。//一开始写的有点乱，只好各种*2+n 。。。

#include<iostream> 
#include<cstdio>
#define N 80000
using namespace std;
int n,a,b,c,fa[N],L[N],dp[N][3],first[N],next[N],v[N],to[N],l,Ans,order[N];
void link(int a,int b,int c)
{
	v[++l]=c;to[l]=b;next[l]=first[a];first[a]=l;
	v[++l]=c;to[l]=a;next[l]=first[b];first[b]=l;
}
void dfs(int t)
{
	order[++l]=t;
	for (int i=first[t];i;i=next[i])
		if (to[i]!=fa[t])
			fa[to[i]]=t,L[to[i]]=v[i],dfs(to[i]);
}
int Gcd(int a,int b){if (b) return Gcd(b,a%b);return a;}
int main()
{
	scanf("%d",&n);
	for (int i=1;i<n;i++)
	{
		scanf("%d%d%d",&a,&b,&c);
		link(a,b,c%3);
	}
	l=0;dfs(1);
	for (int i=n;i>=1;i--)
	{
		int k=order[i];
		dp[k][0]++;
		for (int j=0;j<=2;j++)
			dp[fa[k]][(j+L[k])%3]+=dp[k][j];
	}
	for (int i=1;i<=n;i++)
	{
		int now[3]={0,0,0};
		for (int j=first[i];j;j=next[j])
		if (to[j]!=fa[i])
		{
			Ans+=now[(3-v[j])%3]*dp[to[j]][0]
				+now[(4-v[j])%3]*dp[to[j]][2]
				+now[(5-v[j])%3]*dp[to[j]][1];
			now[(0+v[j])%3]+=dp[to[j]][0];
			now[(1+v[j])%3]+=dp[to[j]][1];
			now[(2+v[j])%3]+=dp[to[j]][2];
		}
		Ans+=now[0];
	}
	Ans=Ans*2+n;n=n*n;
	int gcd=Gcd(Ans,n);
	printf("%d/%d\n",Ans/gcd,n/gcd);
}