[BZOJ2152]聪聪可可(点分治)

题目:

我是超链接

题解:

喵喵喵刚交上去就发现当ans=0时自己的输出时0\n0“woc!”,然后刷新了一下刚想重交,A了。。。

树上的路径问题可以通过点分治来解决,这个%3=0怎么办呢?如果还是按照刚才模板题的想法—–两点之间的距离是深度和的话,那么我们求出到各个点的深度的时候答案就是(%3=0的点的个数)^2+2*(%3=1的点的个数)*(%3=2的点的个数)
很好理解?

代码:

#include <cstdio>
#include <cstring>
#include <iostream>
#define INF 1e9
using namespace std;
const int N=20005;
int tot,nxt[N*2],point[N],v[N*2],c[N*2],f[N],size[N],sum,root,one,two,zer,d[N],ans;
bool vis[N];
int gcd(int a,int b){if (!b) return a;else return gcd(b,a%b);}
void addline(int x,int y,int z)
{
    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y; c[tot]=z;
    ++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x; c[tot]=z;
}
void getroot(int x,int fa)
{
    f[x]=0; size[x]=1;
    for (int i=point[x];i;i=nxt[i]) 
      if (v[i]!=fa && !vis[v[i]])
      {
        getroot(v[i],x);
        size[x]+=size[v[i]];
        f[x]=max(f[x],size[v[i]]);
      }
    f[x]=max(f[x],sum-f[x]);
    if (f[x]<f[root]) root=x;
}
void getdeep(int x,int fa)
{
    switch(d[x])
    {
        case 1:one++;break;
        case 2:two++;break;
        case 0:zer++;break;
    }
    for (int i=point[x];i;i=nxt[i]) 
      if (v[i]!=fa && !vis[v[i]])
      {
        d[v[i]]=(d[x]+c[i])%3;
        getdeep(v[i],x);
      }
}
int calc(int x,int now)
{
    d[x]=now;
    one=0; two=0; zer=0;
    getdeep(x,0);
    return zer*zer+2*one*two;
}
void work(int x)
{
    ans+=calc(x,0);
    vis[x]=1;
    for (int i=point[x];i;i=nxt[i])
      if (!vis[v[i]])
      {
        ans-=calc(v[i],c[i]);
        sum=size[v[i]]; root=0; getroot(v[i],x);
        work(root);
      }
}
int main()
{
    int n,i;
    scanf("%d",&n);
    for (i=1;i<n;i++)
    {
        int x,y,z;scanf("%d%d%d",&x,&y,&z); z%=3;
        addline(x,y,z);
    }
    sum=n; f[0]=INF; root=0; getroot(1,0);
    work(root); 
    if (ans==0){printf("0/0"); return 0;}
    int t=gcd(ans,n*n);
    printf("%d/%d",ans/t,n*n/t);
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值