BZOJ 2152: 聪聪可可点分治

最新推荐文章于 2020-01-23 23:32:49 发布

sillyf

最新推荐文章于 2020-01-23 23:32:49 发布

阅读量356

点赞数

CC 4.0 BY-SA版权

分类专栏：分治文章标签：分治

本文链接：https://blog.youkuaiyun.com/qq_35878547/article/details/72667388

分治专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一种图论中的算法——点分治，并通过一道题目加深理解。文章提供了完整的C++实现代码，包括获取重心、计算子树深度及解决过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

脑子不太清醒，看点分治很久才有点感觉，于是顶着睡意卡了一道题

点分治，算是入个门吧

学习点分治http://www.cnblogs.com/chty/p/5912360.html

相当于照着抄的

#include<cstdio>
#include<algorithm>
#define N 20005
using namespace std;

inline int read()
{
	int a=0,f=1;static char c=getchar();
	while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
	while(c>='0'&&c<='9'){a=a*10+c-'0';c=getchar();}
	return a*f;
}

struct edge{int to,v,next;}e[N*2];
int n,sum,ans=0,root=0,cnt=0;
int t[5],d[N],son[N],mx[N],vis[N],head[N];

int gcd(int a,int b){if(b==0) return a;else return gcd(b,a%b);}

void add(int x,int y,int v)
{
	e[++cnt].next=head[x];
	head[x]=cnt;
	e[cnt].to=y,e[cnt].v=v;
}

void getr(int x,int fa)
{
	son[x]=1,mx[x]=0;
	for(int i=head[x];i;i=e[i].next)
	{
		if(e[i].to==fa||vis[e[i].to]) continue;
		getr(e[i].to,x);
		son[x]+=son[e[i].to];
		mx[x]=max(mx[x],son[e[i].to]);
	}
	mx[x]=max(mx[x],sum-son[x]);
	if(mx[x]<mx[root]) root=x;
}

void getdeep(int x,int fa)
{
	++t[d[x]];
	for(int i=head[x];i;i=e[i].next)
	{
		if(e[i].to==fa||vis[e[i].to]) continue;
		d[e[i].to]=(d[x]+e[i].v)%3;
		getdeep(e[i].to,x);
	}
}

int cal(int x,int v)
{
	t[0]=t[1]=t[2]=0;
	d[x]=v;
	getdeep(x,0);
	return t[1]*t[2]*2+t[0]*t[0];
}

void work(int x)
{
	ans+=cal(x,0);vis[x]=1;
	for(int i=head[x];i;i=e[i].next)
	{
		if(vis[e[i].to]) continue;
		ans-=cal(e[i].to,e[i].v);
		root=0,sum=son[e[i].to];
		getr(e[i].to,0);
		work(root);
	}
}

int main()
{
	n=read();
	register int i;
	for(i=1;i<n;++i)
	{
		int x=read(),y=read(),z=read()%3;
		add(x,y,z);add(y,x,z);
	}
	sum=mx[0]=n;
	getr(1,0);
	work(root);
	int t=gcd(ans,n*n);
	printf("%d/%d\n",ans/t,n*n/t);
	return 0;
}