这道题主要是要学习下这个树形dp的写法。
我一开始的想法是全部转移完再来统计贡献。然而,这就造成一个麻烦。
我们要统计的贡献是树中是权值和为3的倍数的路径个数。如果按照暴力思想,肯定是枚举两个点之后check这两点之间的距离是不是3的倍数。
然而,2e4的数据量不允许我们这样做。因此,考虑到这道题是树上问题,再联想到树上求距离还有一个重要的对象就是lca。所以,我们的状态设计其实就是:dp[u][i]表示到点u的距离模3为i的点的个数,相当于枚举了lca。
那么,考虑统计贡献:显然,贡献来自dp[u][0]以及dp[u][1]+dp[u][2]这两个相加起来等于3的路径。但是注意到,这个dp[u][1]和dp[u][2]必须来自不同的儿子,不然他们取得lca就不是u。
因此,我们不能全部转移完再来统计。考虑在转移途中统计贡献。
AC代码
#include <bits/stdc++.h>
using namespace std;
template<typename T>
void read(T&x){
x=0;
char ch=getchar();
int f=1;
while(!isdigit(ch)){
if(ch=='-')f*=-1;
ch=getchar();
}
while(isdigit(ch)){
x=x*10+(ch-'0');
ch=getchar();
}x*=f;
}
template<typename T>
void write(T x){
if(x<0)putchar('-'),x=-x;
if(x>9)write(x/10);
putchar(x%10+'0');
}
//=============================================
typedef long long ll;
#define int ll
const int maxn=2e4+10;
int n;
struct node{
int to,next,w;
}e[maxn<<1];
int head[maxn],cnt;
void add(int x,int y,int w){
e[cnt].to=y;
e[cnt].w=w;
e[cnt].next=head[x];
head[x]=cnt++;
}
bool vis[maxn];
int dp[maxn][3];
int sum=0;
void solve(int u,int fa){
dp[u][0]=1;
for(int i=head[u];~i;i=e[i].next){
int v=e[i].to;
if(v==fa)continue;
solve(v,u);
for(int j=0;j<3;++j){
sum+=2*dp[u][j]*dp[v][((3-e[i].w-j)%3+3)%3];
}
for(int j=0;j<3;++j){
dp[u][(j+e[i].w)%3]+=dp[v][j];
}
}
}
signed main(){
//freopen("in.txt","r",stdin);
memset(head,-1,sizeof(head));
read(n);
for(int i=1;i<n;++i){
int x,y,z;
read(x),read(y),read(z);
add(x,y,z);add(y,x,z);
}
int tot=n*n;
solve(1,-1);
sum+=n;
int gg=__gcd(sum,tot);
cout<<(sum/gg)<<"/"<<(tot/gg)<<endl;
return 0;
}