洛谷P2634 聪聪可可 [国家集训队] 点分治/dp

正解:点分治/dp

解题报告:

传送门!

这题有两个做法,都是我不擅长的就都说下好了QAQ

首先这题一看到就会想到点分治?

也确实可以用点分治,那就直接用点分治鸭

每次求出到当前根距离余数为0,1,2的点的数量x,y,z

然后它的贡献就是x*x+2*y*z

然后就做完了,,,

晚上放代码QAQ

然后港下dp方法,,,

就直接树形dp,记录f[i][0/1/2]:i的子树内距离为0/1/2的个数

over

同晚上放代码QAQ

然后说一下这题有一个要注意的就是可以是选的两个点可以是同一个点

 

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define rg register
#define il inline
#define mp make_pair
#define rp(i,x,y) for(rg ll i=x;i<=y;++i)

const ll N=200000+10;
ll n,m,head[N],cnt,f[N][3],g[3],as1,as2;
struct ed{ll to,nxt,wei;}edge[N<<2];
bool vis[N];

il ll read()
{
    rg char ch=getchar();rg ll x=0;rg bool y=1;
    while(ch!='-' && (ch>'9' || ch<'0'))ch=getchar();
    if(ch=='-')ch=getchar(),y=0;
    while(ch>='0' && ch<='9')x=(x<<1)+(x<<3)+(ch^'0'),ch=getchar();
    return y?x:-x;
}
il ll gcd(ll x,ll y){return y==0?x:gcd(y,x%y);}
il void ad(ll x,ll y,ll z){edge[++cnt].to=y;edge[cnt].nxt=head[x];edge[cnt].wei=z;head[x]=cnt;}
void dfs(ll nw)
{
    vis[nw]=1;f[nw][0]=1;
    for(rg ll i=head[nw];i;i=edge[i].nxt)
    {
        if(vis[edge[i].to])continue;dfs(edge[i].to);
        rp(j,0,2)g[(edge[i].wei+j)%3]=f[edge[i].to][j];
        rp(j,0,2)as1+=(ll)g[j]*f[nw][(3-j)%3];
        rp(j,0,2)f[nw][j]+=g[j];
    }
}

int main()
{
    n=read();rp(i,1,n-1){ll x=read(),y=read(),z=read();ad(x,y,z);ad(y,x,z);}
    dfs(1);as1<<=1;as1+=n;as2=(ll)n*n;ll tmp=gcd(as1,as2);as1/=tmp;as2/=tmp;printf("%lld/%lld\n",as1,as2);
    return 0;
}

这是树形dp的代码!