换根DP,顾名思义就是换一个根进行DP(废话),一般出现在树形结构中,需要定一个结点为根(题意可能是中心或其他的),然后求总值。
先看一道例题https://www.luogu.com.cn/problem/P2986
题目不难理解需要求从各个结点的奶牛走到中心的花费总值,如果用暴力解题思路也不难理解,
用一次DFS求出一个结点的所有子节点的花费值,再加上子树牛的数量乘以根到子节点的道路长
图示:
然后每个节点进行一次DP就行了。时间复杂度是O(n2)
但是问题是数据量是100000,这个方法明显超时了,因此要用到换根DP。
我们看一下可以发现所有节点似乎和父节点有关系,因此可以尝试再用一次DFS来求解
设u为父节点,v为子节点,w为u到v的道路长度,s[v]为以v为子树的总牛数
dp[v]=dp[v]+(dp[u]-w*s[v]-dp[v])+w*(s[1]-s[v])
我们分开来看这个式子
dp[v] 当前节点子树总值(其实用不到的)
dp[u]-w*s[v]-dp[v] 父节点总值减去子节点总值再减去牛牛聚集在子节点进入父节点乘以路长
w*(s[1]-s[v]) 不是该节点的其他牛牛聚集在父节点再进入该子节点的花费
我们看见dp[v]没了!!
所以再经过一次DFS就可将时间复杂度将为O(n)
AC代码如下
#include <iostream>
#include <cstdio>
#include <algorithm>
#define N 100010
using namespace std;
typedef long long ll;
ll n;
ll p[N],s[N];//子树总牛数
ll head[N],nxt[N<<1],we[N<<1],vi[N<<1];//数组模拟链表存图
ll fa[N];//父亲节点
ll cnt;
ll dp[N];
ll result;
void add(int from,int to,int weight)
{
cnt++;
vi[cnt]=to;
we[cnt]=weight;
nxt[cnt]=head[from];
head[from]=cnt;
}
void dfs(int k)
{
s[k]=p[k];
for(int i=head[k];i;i=nxt[i])
{
if(vi[i]!=fa[k])
{
fa[vi[i]]=k;
dfs(vi[i]);
s[k]+=s[vi[i]];
dp[k]+=dp[vi[i]]+s[vi[i]]*we[i];
}
}
}
void dfs2(int k)
{
for(int i=head[k];i;i=nxt[i])
{
if(fa[vi[i]]==k)
{
dp[vi[i]]=dp[k]-2*we[i]*s[vi[i]]+we[i]*s[1];
result=min(result,dp[vi[i]]);
dfs2(vi[i]);
}
}
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&p[i]);
for(int i=1;i<n;i++)
{
int a,b,w;
scanf("%d%d%d",&a,&b,&w);
add(a,b,w);
add(b,a,w);
}
dfs(1);
result=dp[1];
dfs2(1);
printf("%lld",result);
return 0;
}
总结换根DP套路
换根dp一般分为三个步骤
1、先指定一个根节点
2、一次DFS统计子树内的节点对当前节点的贡献
3、一次DFS统计父亲节点对当前节点的贡献并合并统计最终答案