[BZOJ2152] [国家集训队] 聪聪可可 [点分治]

最新推荐文章于 2021-09-22 18:41:38 发布

_er

最新推荐文章于 2021-09-22 18:41:38 发布

阅读量232

点赞数

文章标签：点分治

本文链接：https://blog.csdn.net/estia_/article/details/82902912

版权

题意：
给出一个 $N$ 个点的树，每条边上有权值 $w_i$ 。
求 $\frac{路径上权值和为3的倍数的有序对(u,v)数}{总有序对数}$

分母好搞，考虑怎么求分子。
显然可以把路径上边权和对 $3$ 取模，结果为 $0$ 的就是合法路径。
不过路径的问题得分解为 $d i s (u) + d i s (v) - 2 * d i s (l c a)$
考虑对于某个点，路径过该点的点对数
$cnt(dis\%3==1)*cnt(dis\%3==2)*2+cnt^2(3|dis)$ 。
$lca\ne root$ 的容斥即可，点分。
可以有 $u = v$ 。
$\Theta(Nlog^2N)$

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<cctype>
using namespace std;
#define add_edge(u,v,w) nxt[++tot]=head[u],head[u]=tot,to[tot]=v,val[tot]=w
#define LL long long
int N,tot,root=0,mn,sum;
LL ans=0;
int head[20005]={},nxt[40005]={},to[40005]={},val[40005]={};
int siz[20005]={},mxsiz[20005]={};
LL cnt[3]={};
bool vis[20005]={};
void getroot(int x,int fa)
{
	siz[x]=1; mxsiz[x]=0;
	for(int i=head[x];i;i=nxt[i])
	{
		if(to[i]==fa||vis[to[i]])continue;
		getroot(to[i],x);
		siz[x]+=siz[to[i]];
		mxsiz[x]=max(mxsiz[x],siz[to[i]]);
	}
	mxsiz[x]=max(mxsiz[x],sum-siz[x]);
	if(mxsiz[x]<mn)mn=mxsiz[x],root=x;
}
void query(int x,int fa,int l)
{
	++cnt[l%3];
	for(int i=head[x];i;i=nxt[i])
	{
		if(vis[to[i]]||to[i]==fa)continue;
		query(to[i],x,(l+val[i])%3);
	}
}
LL solve(int x,int add)
{
	cnt[0]=cnt[1]=cnt[2]=0;
	query(x,0,add);
	return 2ll*cnt[1]*cnt[2]+cnt[0]*cnt[0];
}
void divide(int x)
{
	ans+=solve(x,0); vis[x]=1;
	for(int i=head[x];i;i=nxt[i])
	{
		if(vis[to[i]])continue;
		ans-=solve(to[i],val[i]);
		sum=siz[to[i]]; mn=2147483647; getroot(to[i],0);
		divide(root);
	}
}
LL gcd(LL a,LL b){return !b?a:gcd(b,a%b);}
int main()
{
	scanf("%d",&N);
	for(int a,b,c,i=1;i<N;++i)
	{
		scanf("%d%d%d",&a,&b,&c);
		add_edge(a,b,c); add_edge(b,a,c);
	}
	sum=N; mn=2147483647;
	getroot(1,0);
	divide(1);
	LL tmp=N*N,g=gcd(ans,tmp);
	printf("%lld/%lld",ans/g,tmp/g);
	return 0;
}