bzoj2152 树形DP

来源：互联网发布：点菜软件免费下载编辑：程序博客网时间：2024/06/06 19:33

听说这题还有点分治做法。。。

传送门：BZOJ2152

用dp[i][j]表示从i的子树中的点到i有多少条路径%3为j

强行树形DP

/**************************************************************    Problem: 2152    User: SmallFat    Language: C++    Time : 232ms   Result: Accepted****************************************************************/ #include<cstdio>#include<cstdlib>#include<cstring>#include<algorithm>#define maxn 101111using namespace std;struct edges{    int r,nxt,w;}e[maxn<<1];int head[maxn],esz;long long ans=0,dp[maxn][3];long long gcd(long long a,long long b){    return b?gcd(b,a%b):a;}void dfs(int n,int fa){    dp[n][0]++;    for(int i=head[n];i;i=e[i].nxt){        int m=e[i].r;        if(m==fa)continue;        dfs(m,n);ans+=dp[n][0]*dp[m][(3-e[i].w)%3];ans+=dp[n][1]*dp[m][(3+2-e[i].w)%3];ans+=dp[n][2]*dp[m][(3+1-e[i].w)%3];        for(int j=0;j<3;++j)            dp[n][(j+e[i].w)%3]+=dp[m][j];            }         }void addedge(int x,int y,int w){    e[++esz].r=y;e[esz].nxt=head[x];head[x]=esz;e[esz].w=w;    e[++esz].r=x;e[esz].nxt=head[y];head[y]=esz;e[esz].w=w;}int main(){    int n;    scanf("%d",&n);    for(int i=1;i<n;++i){        int a,b,w;        scanf("%d%d%d",&a,&b,&w);        addedge(a,b,w%3);    }    dfs(1,0);    ans=ans*2+n;    long long x=n*n;    long long g=gcd(x,ans);    printf("%lld/%lld",ans/g,x/g);}

话说这不一水题吗。。。我竟然调啦好久

0 0