树形dp经典换根法——cf1187E
Posted zsben991126
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了树形dp经典换根法——cf1187E相关的知识,希望对你有一定的参考价值。
假设以u为根时结果是tot,现在转换到了以u的儿子v为根,那么结果变成了tot-size[v]+(sizetot-size[v])
根据这个转移方程,先求出以1为根的tot,然后dfs一次转移即可
#include<iostream> #include<cstring> #include<cstdio> #include<algorithm> using namespace std; #define ll long long #define N 200005 int a[N],n; vector<int>e[N]; ll sum[N],ans=0,sig=0; inline void dfs1(int p,int fa,int dep) sum[p]=a[p],sig+=(ll)a[p]*dep; for(int i=0;i<e[p].size();++i) if(e[p][i]!=fa) dfs1(e[p][i],p,dep+1),sum[p]+=sum[e[p][i]]; inline void dfs2(int p,int fa,ll Sum) ans=max(ans,Sum); for(int i=0;i<e[p].size();++i) if(e[p][i]!=fa) dfs2(e[p][i],p,Sum-2ll*sum[e[p][i]]+sum[1]); int main() cin>>n; for(int i=1;i<=n;++i) a[i]=1; for(int i=1,u,v;i<n;++i) cin>>u>>v,e[u].push_back(v),e[v].push_back(u); dfs1(1,0,0); dfs2(1,0,sig); cout<<ans+n;
以上是关于树形dp经典换根法——cf1187E的主要内容,如果未能解决你的问题,请参考以下文章
题解 poj3585 Accumulation Degree (树形dp)(二次扫描和换根法)
[CF1187E] Tree Painting - 树形dp
POJ3585 Accumulation Degree 树形dp