BZOJ #4543. [POI2014]Hotel加强版

参考blog
题目连接
长链剖分真是优雅的暴力!

如果是找中间点的话,只能做到$O(n^2)$,
显然两个点到一个点的距离等距是很好维护的,但是多一个点就不太行了.
既然这种情况麻烦,我们不妨把它记录在状态内,这样找第3个点就很方便啦~~

定义$f[i][j]$表示子树$i$内深度为$j$的点数,$g[i][j]$表示子树$i$内钦定第三个点在子树外且距离$i$ $j$个单位时,能够形成的合法三元组数量,$le{n}_i$表示子树$i$内最大深度.

转移和贡献如下:
首先贡献答案$ans+={\sum }_y{\sum }_{i=1}^{le{n}_y}g[x][i]\ast f[y][i-1]+g[y][i]\ast f[x][i-1]$.(第三个点距离$y$ $i$个单位且在子树$y$外,那么显然距离$x$就是$i-1$啊…

然后考虑转移:
$g[x][i-1]+=g[y][i]$,
$g[x][i]+=f[x][i]\ast f[y][i-1]$,
$f[x][i]+=f[y][i-1]$.
注意先贡献后转移,同时注意$g$是往前贡献,所以我们长链剖分的时候记得开够空间.

int n,len[N],son[N],fa[N];
ll _f[N<<2],*p=_f,*f[N],*g[N],ans;

void New(int x) {
	f[x]=p; p += 2*len[x];
	g[x]=p; p += 2*len[x];
}

V<int> e[N];
void ins(int x,int y) {e[x].pb(y);}

void dfs(int x) {
	for(int y:e[x]) if(y ^ fa[x]) {
		fa[y]=x; dfs(y);
		if(len[son[x]] < len[y]) son[x]=y;
	}
	len[x]=len[son[x]]+1;
}

void DFS(int x) {
	f[x][0]=1;
	if(son[x]) f[son[x]]=f[x]+1,g[son[x]]=g[x]-1,DFS(son[x]),ans += g[x][0];
	else return ;
	for(int y:e[x]) if(y ^ son[x] && y ^ fa[x]) {
		New(y); DFS(y);
		FOR(i,len[y]) ans += g[x][i]*f[y][i-1] + f[x][i-1]*g[y][i];
		FOR(i,len[y]) g[x][i-1] += g[y][i], g[x][i] += f[x][i] * f[y][i-1], f[x][i] += f[y][i-1];
	}
}

int main() {
	qr(n);  int x,y;
	rep(i,2,n) qr(x),qr(y),ins(x,y),ins(y,x);
	dfs(1); New(1); DFS(1); pr2(ans); return 0;
}