P3565 [POI2014] HOT-Hotels【长链剖分】

OMG_NOIP

于 2024-09-10 11:45:10 发布

阅读量598

点赞数 9

分类专栏：题解省选复习文章标签：算法 c++

本文链接：https://blog.csdn.net/MC_wansui/article/details/142094137

版权

题解同时被 2 个专栏收录

32 篇文章 0 订阅

订阅专栏

省选复习

31 篇文章 0 订阅

订阅专栏

P3565 [POI2014] HOT-Hotels 总题单链接

不会长链剖分和空间共享的同学可以先看看我的长链剖分讲解。

ps：本文主要讲优化，数组和变量也用的 $O(N^2)$ 版本的，所以建议先看完 $O(N^2)$ 版本的题解。

思路

发现 $d p$ 和 $g$ 都会继承自己的长儿子的值，所以可以考虑长链剖分优化 $D P$ 。

$D P$ 优化部分和模板题几乎一样，只是 $D P$ 数组要反过来分配空间。

注意：每条长链分配双倍空间，不然会因为一些神秘的东西 $W A$ 掉。

代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;

ll n,ans;
vector<ll>eg[5005];
ll len[5005],son[5005];
ll buf1[10005],*dp[5005],*t1=buf1;
ll buf2[10005],*g[5005],*t2=buf2;

void dfs_build(ll fa,ll p){
	for(ll v:eg[p]){
		if(v==fa)continue;
		dfs_build(p,v);
		if(len[son[p]]<len[v])son[p]=v;
	}
	len[p]=len[son[p]]+1;
}

void dfs_dp(ll fa,ll p){
	g[p][0]=1;
	if(son[p]){
		dp[son[p]]=dp[p]-1;
		g[son[p]]=g[p]+1;
		dfs_dp(p,son[p]);
		ans+=dp[p][0];
	}
	for(ll v:eg[p]){
		if(v==fa||v==son[p])continue;
		dp[v]=t1+len[v],t1+=len[v]<<1;
		g[v]=t2,t2+=len[v]<<1;
		dfs_dp(p,v);
		for(ll i=1;i<=len[v];i++)ans+=dp[p][i]*g[v][i-1];
		for(ll i=1;i<len[v];i++)ans+=g[p][i-1]*dp[v][i];
		for(ll i=1;i<=len[v];i++)dp[p][i]+=g[p][i]*g[v][i-1];
		for(ll i=1;i<len[v];i++)dp[p][i-1]+=dp[v][i];
		for(ll i=1;i<=len[v];i++)g[p][i]+=g[v][i-1];
	}
}

signed main(){
	ios::sync_with_stdio(false);
	
	cin>>n;
	for(ll i=1;i<n;i++){
		ll x,y;cin>>x>>y;
		eg[x].push_back(y);
		eg[y].push_back(x);
	}
	
	dfs_build(0,1);
	
	dp[1]=t1+len[1],t1+=len[1]<<1;
	g[1]=t2,t2+=len[1]<<1;
	dfs_dp(0,1);
	
	cout<<ans;
	
	return 0;
}