题意:
给你一颗树,你选择一个节点当根,求所有点的深度和最少为多少?
思路:
树状dp+换根
一开始随便选一个点当根计算结果。
dp[i]表示以i为根的子树节点的深度和。
se[i]表示以i为根的子树的节点数目。
换根:
dp[v]=dp[u]+n-2*se[v];(v为u的子节点,根从u转向v时,以v为根的子树中的节点深度全减一,其余节点深度加一)
代码:
#include <bits/stdc++.h>
typedef long long ll;
using namespace std;
const ll inf=1000000007;
inline int read()
{
int x=0,f=1;
char ch=getchar();
while(ch<'0'||ch>'9')
{
if(ch=='-')
f=-1;
ch=getchar();
}
while(ch>='0'&&ch<='9')
{
x=(x<<1)+(x<<3)+(ch^48);
ch=getchar();
}
return x*f;
}
vector<int> g[1000005];
ll dp[1000005], se[1000005];
int n;
void dfs(int v,int f,int d)
{
dp[v]=d;
se[v]=1;
for(int i=0;i<g[v].size();i++)
{
if(g[v][i]!=f)
{
dfs(g[v][i],v,d+1);
dp[v]+=dp[g[v][i]];
se[v]+=se[g[v][i]];
}
}
}
void dfs1(int v,int f)
{
if(f!=-1)
{
dp[v]=dp[f]+n-2*se[v];
}
for(int i=0;i<g[v].size();i++)
{
if(g[v][i]!=f)
{
dfs1(g[v][i],v);
}
}
}
int main()
{
n=read();
for(int i=1;i<n;i++)
{
int u=read(), v=read();
g[u].push_back(v);
g[v].push_back(u);
}
dfs(1,-1,0);
dfs1(1,-1);
ll ans=inf*inf;
for(int i=1;i<=n;i++)
{
ans=min(ans,dp[i]);
}
cout << ans << endl;
return 0;
}

京公网安备 11010502036488号