聪聪可可-点分治P2634

https://www.luogu.org/problemnew/show/P2634

思路

1. 点分治
2. 考虑solve，发现我们可以记录一个cnt[0],cnt[1],cnt[2],分别代表经过根的dis值%3余0、1、2。那么对答案的贡献就是cnt[0]^2+2(1->2,2->1算两个）*cnt[1]*cnt[2]。

code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<cmath>
#include<cstdlib>
#include<ctime>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
inline int read(){
	char ch=' ';int f=1;int x=0;
	while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
	return x*f;
}
const int N=20100;
struct node
{
	int v,nxt,w;
}edge[N<<1];
int head[N],cnt;
void add(int u,int v,int w)
{
	cnt++;
	edge[cnt].v=v;
	edge[cnt].w=w;
	edge[cnt].nxt=head[u];
	head[u]=cnt;
}
int sim[N],mxson[N];
int MX,root,smer;
int fz;
bool vis[N];
int dis[N],cc,sum[3];
void getroot(int u,int fa)
{
	sim[u]=1;mxson[u]=0;
	for(int i=head[u];i;i=edge[i].nxt)
	{
		int v=edge[i].v;
		if(vis[v]||v!=fa) continue;
		getroot(v,u);
		sim[u]+=sim[v];mxson[u]=max(mxson[u],sim[v]);
	}
	mxson[u]=max(mxson[u],smer-sim[u]);
	if(mxson[u]<MX){
		root=u;MX=mxson[u];
	}
}
void dfs(int u,int fa,int dist)
{
	dis[u]=dist;sum[dis[u]%3]++;
	for(int i=head[u];i;i=edge[i].nxt)
	{
		int v=edge[i].v;
		int w=edge[i].w;
		if(vis[v]||v==fa) continue;
		dfs(v,u,dist+w);
	}
}
int solve(int u,int w)
{
	memset(sum,0,sizeof(sum));
	dfs(u,0,w);
	return sum[0]*sum[0]+2*sum[1]*sum[2];
}
void divide(int u)
{
	fz+=solve(u,0);
	vis[u]=true;
	for(int i=head[u];i;i=edge[i].nxt)
	{
		int v=edge[i].v;
		int w=edge[i].w;
		if(vis[v]) continue;
		fz-=solve(v,w);
		MX=inf;smer=sim[v];getroot(v,0);
		divide(root);
	}
}
int gcd(int a,int b)
{
	return b==0?a:gcd(b,a%b);
}
int main()
{
	int n;
	n=read();
	int i,j;
	for(i=1;i<n;i++)
	{
		int u,v,w;
		u=read();v=read();w=read();
		add(u,v,w);
		add(v,u,w);
	} 
	fz=0;int fm=n*n;smer=n;MX=inf;getroot(1,0);divide(root);
	int x=gcd(fz,fm);
	cout<<fz/x<<"/"<<fm/x<<endl;
	return 0;
}