【金牌导航】【洛谷 P2634】【点分治】聪聪可可

【金牌导航】【洛谷 P2634】【点分治】聪聪可可

题目

---

解题思路

用点分治，缩小规模解决问题
因为要距离能整除3
把边距和维护的d都存距离mod 3的值就可以了

---

代码

#include<iostream>
#include<cstdio>
using namespace std;
struct lzf{
	int to,w,nxt;
}f[50000];
bool vis[50000];
int n,t,u,v,w,ans,q[50000],p[5],fa[50000];
int dis[50000],son[50000],head[50000],size[50000];
void add(int x,int y,int w)
{
	 f[++t]=(lzf){y,w%3,head[x]};
	 head[x]=t;
}
int weight(int x)
{
	int ma=n,w;
	q[t=1]=x,fa[x]=0;
	for (int i=1;i<=t;i++)
	{
		son[q[i]]=0,size[i]=1;
		for (int j=head[q[i]];j;j=f[j].nxt)
		    if (!vis[f[j].to]&&fa[q[i]]!=f[i].to)
		    {
		    	fa[f[i].to]=q[i];
		    	q[++t]=f[i].to;
			}
	}
	for (int i=t;i>0;i--)
	{
		int y=q[i];
		if (t-size[y]>son[q[i]]) son[y]=t-size[y]; 
		if (ma>son[y]) ma=son[y],w=y;
		size[fa[y]]+=size[y];
		if (son[fa[y]]<size[y]) son[fa[y]]=size[y];
	}
	return w;
}
int calc(int x,int v)
{
	dis[x]=v,p[0]=p[1]=p[2]=0;
	q[t=1]=x,fa[x]=0;
	int sum=0;
	for (int i=1;i<=t;i++)
	{
		p[dis[q[i]]]++;
		for (int j=head[q[i]];j;j=f[j].nxt)
		    if (!vis[f[j].to]&&fa[q[i]]!=f[j].to)
			   dis[f[j].to]=(dis[q[i]]+f[j].w)%3,q[++t]=f[j].to;
	}
	sum=p[1]*p[2];
	if (p[0]>1) sum+=p[0]*(p[0]-1);
	return sum;
}
void solve(int x)
{
	int G=weight(x);
	vis[G]=1;
	ans+=calc(G,0);
	for (int i=head[G];i;i=f[i].nxt)
	    if (!vis[f[i].to]) ans-=calc(f[i].to,f[i].w);
	for (int i=head[G];i;i=f[i].nxt)
	    if (!vis[f[i].to]) solve(f[i].to);
}
int gcd(int x,int y)
{
	if (x%y==0) return x;
	return gcd(y,x%y);
}
int mian()
{
	scanf("%d",&n);
	for (int i=1;i<=n;i++)
	{
		scanf("%d%d%d",&u,&v,&w);
		add(u,v,w);
		add(v,u,w);
	}
	solve(1);
	u=gcd(ans,n*n);
	printf("%d/%d\n",ans/u,n*n/u);
	return 0;
}