~~~~~ P3565 [POI2014] HOT-Hotels ~~~~~ 总题单链接
~~~~~ 不会长链剖分和空间共享的同学可以先看看我的长链剖分讲解。
~~~~~ ps:本文主要讲优化,数组和变量也用的 O ( N 2 ) O(N^2) O(N2) 版本的,所以建议先看完 O ( N 2 ) O(N^2) O(N2)版本的题解。
思路
~~~~~ 发现 d p dp dp 和 g g g 都会继承自己的长儿子的值,所以可以考虑长链剖分优化 D P DP DP。
~~~~~ D P DP DP 优化部分和模板题几乎一样,只是 D P DP DP 数组要反过来分配空间。
~~~~~ 注意:每条长链分配双倍空间,不然会因为一些神秘的东西 W A WA WA 掉。
代码
#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;
}