2019.01.10【BZOJ4543】【POI2014】Hotel加强版（长链剖分）

传送门

---

解析：

长链剖分裸题。

我们考虑如果树上三个点距离相等，那么必然存在一个点$u$，它到这三个点的距离全部相等。

那么分类讨论两种情况，三个点在$u$节点的不同子树内部，这个很好统计，随手就转移了。

那么还有一个情况，其中一个节点在子树外部。
换句话说，就是在$u$的某一个祖先的其他子树当中。

这个不好直接在$u$处统计啊。

换一个思路，我们考虑在这个上级祖先处统计这个三元组。

我们需要对节点$u$记录数组$g_u$，其中$g_{u,i}$表示需要向子树外延伸距离为$i$来寻找第三个点的二元组。

这个东西每条长链可以$O(dep)$处理，总的复杂度$O(n)$。

然后就是长链剖分的套路了，指针申请空间。其中$g$数组需要开两倍空间，因为转移方向是$-1$，所以向上一倍合并给父亲，向下一倍合并其他子树

---

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define re register
#define gc get_char
#define pc put_char
#define cs const

namespace IO{
	namespace IOONLY{
		cs int Rlen=1<<16|1;
		char buf[Rlen],*p1,*p2;
		char obuf[Rlen],*p3=obuf;
		char ch[23];
	}
	inline char get_char(){
		using namespace IOONLY;
		return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,Rlen,stdin),p1==p2)?EOF:*p1++;
	}
	inline void put_char(char c){
		using namespace IOONLY;
		*p3++=c;
		if(p3==obuf+Rlen)fwrite(obuf,1,Rlen,stdout),p3=obuf;
	}
	inline void FLUSH(){
		using namespace IOONLY;
		fwrite(obuf,1,p3-obuf,stdout),p3=obuf;
	}
	
	inline int getint(){
		re int num;
		re char c;
		while(!isdigit(c=gc()));num=c^48;
		while(isdigit(c=gc()))num=(num+(num<<2)<<1)+(c^48);
		return num;
	}
	inline void outint(ll a){
		using namespace IOONLY;
		if(a==0)pc('0');
		while(a)ch[++ch[0]]=a-a/10*10,a/=10;
		while(ch[0])pc(ch[ch[0]--]^48);
	}
}
using namespace IO;

cs int N=100004;
vector<int> edge[N];
inline void addedge(int u,int v){
	edge[u].push_back(v);
	edge[v].push_back(u);
}

int n,fcnt,gcnt;ll ans;
ll *f[N],*g[N],farr[N],garr[N<<1];
int mxdep[N],son[N];

inline void dfs1(int u,int fa){
	mxdep[u]=1;
	for(int re e=0;e<edge[u].size();++e){
		re int v=edge[u][e];
		if(v==fa)continue;
		dfs1(v,u);
		if(mxdep[v]>mxdep[son[u]])son[u]=v;
		mxdep[u]=max(mxdep[u],mxdep[v]+1);
	}
}

inline void dfs2(int u,int fa){
	++f[u][0];
	if(!son[u])return ;
	else{
		re int v=son[u];
		f[v]=f[u]+1;
		g[v]=g[u]-1;
		dfs2(v,u);
		ans+=f[u][0]*g[u][0];
	}
	for(int re e=0;e<edge[u].size();++e){
		re int v=edge[u][e];
		if(v==fa||v==son[u])continue;
		f[v]=farr+fcnt;fcnt+=mxdep[v];
		gcnt+=mxdep[v];g[v]=garr+gcnt;gcnt+=mxdep[v];
		dfs2(v,u);
		for(int re i=0;i+1<mxdep[v];++i)ans+=f[u][i]*g[v][i+1];
		for(int re i=1;i-1<mxdep[v];++i)ans+=g[u][i]*f[v][i-1];
		for(int re i=0;i-1<mxdep[v];++i){
			g[u][i]+=g[v][i+1];
			if(i>=1)g[u][i]+=f[v][i-1]*f[u][i];
		}
		for(int re i=1;i-1<mxdep[v];++i)f[u][i]+=f[v][i-1];
	}
}

signed main(){
	n=getint();
	for(int re i=1;i<n;++i)addedge(getint(),getint());
	dfs1(1,0);
	f[1]=farr+fcnt;fcnt+=mxdep[1];
	gcnt+=mxdep[1];g[1]=garr+gcnt;gcnt+=mxdep[1];
	dfs2(1,0);
	outint(ans);pc('\n');
	FLUSH();
	return 0;
}