【BZOJ】【P2152】【聪聪可可】【题解】【点分治】

传送门:http://www.lydsy.com/JudgeOnline/problem.php?id=2152

裸题不解释

Code:

#include<cstdio>
#include<map>
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=20010;
struct edge{int u,v,w;};
vector<edge>G[maxn];
typedef int LL;
int siz[maxn],f[maxn],dep[maxn],n,root,cant[maxn],All;
LL d[maxn],ans,data[maxn];
void makert(int u,int fa){
	siz[u]=1;f[u]=0;
	for(int i=0;i<G[u].size();i++){
		edge e=G[u][i];if(e.v==fa||cant[e.v])continue;
		makert(e.v,u);
		siz[u]+=siz[e.v];
		f[u]=max(f[u],siz[e.v]);
	}f[u]=max(f[u],All-f[u]);
	if(f[root]>f[u])root=u;
}
void dfs(int u,int f){
	data[++data[0]]=d[u];
	for(int i=0;i<G[u].size();i++){
		edge e=G[u][i];
		if(e.v==f||cant[e.v])continue;
		d[e.v]=d[u]+e.w;
		dfs(e.v,u);
	}
}
LL M[3];
void calc(int u){
	M[0]=M[1]=M[2]=0;M[0]++;
	for(int i=0;i<G[u].size();i++){
		edge e=G[u][i];if(cant[e.v])continue;
		data[0]=0;
		d[e.v]=e.w;dfs(e.v,u);
		for(int j=1;j<=data[0];j++)
			ans+=M[(3-data[j]%3)%3];
		for(int j=1;j<=data[0];j++)M[data[j]%3]++;
	}
}
void solve(int u){
 	calc(u);cant[u]=1;
	for(int i=0;i<G[u].size();i++){
		edge e=G[u][i];if(cant[e.v])continue;
		d[e.v]=0;f[root=0]=n+1;
		All=siz[e.v];
		makert(e.v,u);
		solve(root);
	}
}
int main(){
	scanf("%d",&n);All=n;
	for(int i=2;i<=n;i++){
		int u,v,w;scanf("%d%d%d",&u,&v,&w);
		G[u].push_back((edge){u,v,w});
		G[v].push_back((edge){v,u,w});
	}f[root=0]=n+1;
	makert(1,0);
	solve(root);
	ans=ans*2+n;
	LL d=__gcd(ans,(LL)n*n);
	cout<<ans/d<<'/'<<(LL)n*n/d<<endl;
	return 0;
}


你可能感兴趣的:(bzoj)