luogu2634 聪聪可可 (树形dp)

要求出两点间距离==0(mod3) 的数量，然后除以(n*n）

设f[i][j]为i的子树到i的距离==j(mod3）的数量，然后做树形dp即可

因为要最简，所以要求一下gcd，然后除下去

 #include<cstdio>
 #include<algorithm>
 #include<cstring>
 using namespace std;
 const int maxn=;

 int rd(){
     int x=;char c=getchar();
     while(c<''||c>'') c=getchar();
     while(c>=''&&c<='') x=x*+c-'',c=getchar();
     return x;
 }

 struct Edge{
     int b,l,ne;
 }eg[maxn*],son[maxn];
 int N;
 int f[maxn][],f2[maxn][],egh[maxn],sonh[maxn],ect,sct;
 int ansp;

 inline void adeg(int a,int b,int c){
     eg[ect].b=b;eg[ect].l=c;eg[ect].ne=egh[a];egh[a]=ect++;
 }

 void build(int x,int fa){
     for(int i=egh[x];i!=-;i=eg[i].ne){
         int b=eg[i].b;if(b==fa) continue;
         son[sct].b=b;son[sct].l=eg[i].l;son[sct].ne=sonh[x];sonh[x]=sct++;
         build(b,x);
     }
 }

 void dfs(int x){
     f[x][]=;
     for(int i=sonh[x];i!=-;i=son[i].ne) dfs(son[i].b);
     for(int i=sonh[x];i!=-;i=son[i].ne){
         int b=son[i].b,l=son[i].l;
         for(int j=;j<=;j++) f[x][(j+l)%]+=f[b][j],f2[b][(j+l)%]=f[b][j];
     }ansp+=f[x][];
     for(int i=sonh[x];i!=-;i=son[i].ne){
         int b=son[i].b,l=son[i].l;
         for(int j=;j<=;j++) ansp+=f2[b][(-j%)%]*(f[x][j]-f2[b][j]);
     }
 }

 int gcd(int a,int b){
     if(!b) return a;
     else return gcd(b,a%b);
 }

 int main(){
     int i,j,k;
     N=rd();
     memset(egh,-,sizeof(egh));memset(sonh,-,sizeof(sonh));
     for(i=;i<N;i++){
         int a=rd(),b=rd(),c=rd();
         adeg(a,b,c);adeg(b,a,c);
     }build(,);dfs();
     i=gcd(ansp,N*N);
     printf("%d/%d\n",ansp/i,N*N/i);
 }