给定一棵有N个节点的树(通常是无根树,也就是有N-1条无向边),我们可以任选一个节点为根节点,从而定义出每个节点的深度和每棵子树的根。 在树上设计动态规划算法时,一般就以节点从深到浅(子树从小到大)的顺序作为DP的“阶段”。DP的状态表示中,第一维通常是节点编号(代表以该节点为根的子树)。大多数时候,我们采用递归的方式实现树形动态规划。对于每个节点x,先递归在它的每个子节点上进行DP,在回溯时,从子节点向节点x进行状态转移。
简单点说,其实,就是dfs,但这个dfs,时间复杂度要小的很多,并且,具体操作使用数组来模拟不同情况 并且,很多时候,线性dp的可以演变成树形DP,如背包
1.最大独立子集 2.树的重心 3.树的直径
首先拿到这个题,就会先把根给求出来 然后,分析题意,其实,如果选fa,那,son一定不行,但不选fa,son可选可不选,这时,我们就会想到,打dfs,或dp,但又想想,dfsO(2^n),肯定不好,dp,虽然好像可以,但,好像没有阶段,于是,就要用树形DP dp[x][0] =max(dp[s][0],dp[s][1]) dp[x][1] = H[x] + dp[son][0](son∈Son(x) 比如dp[i][1]表示以i号节点为子树的根,当它参会时整棵子树的快乐指数和;dp[i][0]表示以i号节点为子树的根,当它不参会时整棵子树的快乐指数和 很明显,这就是由叶节点,一点点的传到根
#include<bits/stdc++.h> using namespace std; int n; int a[100050]; int x,y; vector<int>g[100005]; int dp[10005][2]; int fa[100005]; int root; void dfs(int x) { dp[x][1]=a[x]; dp[x][0]=0; for(int i=0;i<g[x].size();i++) { int v=g[x][i]; dfs(v); dp[x][1]+=dp[v][0]; dp[x][0]+=max(dp[v][0],dp[v][1]); } } int main() { scanf("%d",&n); for(int i=1;i<=n;i++) { scanf("%d",&a[i]); } for(int i=1;i<=n-1;i++) { scanf("%d %d",&x,&y); g[y].push_back(x); fa[x]=1; } for(int i=1;i<=n;i++) { if(!fa[i]) { root=i; break; } } dfs(root); printf("%d",max(dp[root][1],dp[root][0])); }首先,我们要知道,什么是重心 树的重心定义为,当把节点x去掉后,其最大子树的节点个数最少(或者说成最大连通块的节点数最少),那么节点x就是树的重心。 他有以下几个性质 1、删除重心后所得的所有子树,节点数不超过原树的1/2,一棵树最多有两个重心,且相邻; 2、树中所有节点到重心的距离之和最小,如果有两个重心,那么它们距离之和相等; 3、两个树通过一条边合并,新的重心在原树两个重心的路径上; 4、树删除或添加一个叶子节点,重心最多只移动一条边。 还有,一棵树,重心最多2个,n%2= =0有2个,n%2= =1有一个 运用第一个性质,可以解决大多数问题
运用第一个性质,我们可以排除那些不是重心的,然后,输出剩下的 可以统计任选一个节点为根,把无根树变成有根树,然后dp[i]表示以i为根的子树的节点个数 dp[i]=∑dp[son]+1 这只是部分子树,还有当前节点的长辈,节点总数为n-dp[now],然后在做比较
#include<bits/stdc++.h> using namespace std; int n; int x,y; int dp[100054]; vector<int>g[10005]; vector<int>ans; int vis[10005]; void dfs(int x) { bool flag=1; dp[x]=1; for(int i=0;i<g[x].size();i++) { int v=g[x][i]; if(vis[v]) { continue; } vis[v]=1; dfs(v); dp[x]+=dp[v]; if(dp[v]>n/2) { flag=0; } } if(n-dp[x]>n/2) { flag=0; } if(flag) { ans.push_back(x); } } int main() { scanf("%d",&n); for(int i=1;i<n;i++) { scanf("%d %d",&x,&y); g[x].push_back(y); g[y].push_back(x); } vis[1]=1; dfs(1); printf("%d\n",ans.size()); if(ans.size()==1) { printf("%d",ans[0]); } else { printf("%d\n%d",min(ans[0],ans[1]),max(ans[0],ans[1])); } }给定一棵树,树中每条边都有一个权值,树中两点之间的距离定义为连接两点的路径边权之和。树中最远的两个节点(两个节点肯定都是叶子节点)之间的距离被称为树的直径,连接这两点的路径被称为树的最长链。后者通常也可称为直径。
跑2个dfs,第一个求以根为起点的最短距离,第二个求一最长距离终点为起点的最长距离,ans为第二个
#include <bits/stdc++.h> using namespace std; int n; int x, y; vector<int> g[100005]; int dp[1000005]; int ans; int jl; void dfs(int x, int fa) { if (ans < dp[x]) { ans = dp[x]; jl = x; } for (int i = 0; i < g[x].size(); i++) { int v = g[x][i]; if (v == fa) { continue; } dp[v] = dp[x] + 1; dfs(v, x); } } int main() { scanf("%d", &n); for (int i = 1; i < n; i++) { scanf("%d %d", &x, &y); g[x].push_back(y); g[y].push_back(x); } ans = 0; dp[1] = 0; dfs(1, 0); ans = 0; dp[jl] = 0; dfs(jl, 0); printf("%d", ans); }