普通树形dp
树形dp通常围绕根节点来写状态转移方程
用一道基础的树形dp例题来具体分析:
没有上司的舞会
Ural 大学有 N 名职员,编号为 1∼N。
他们的关系就像一棵以校长为根的树,父节点就是子节点的直接上司。
每个职员有一个快乐指数,用整数 Hi 给出,其中 1≤i≤N。
现在要召开一场周年庆宴会,不过,没有职员愿意和直接上司一起参会。
在满足这个条件的前提下,主办方希望邀请一部分职员参会,使得所有参会职员的快乐指数总和最大,求这个最大值。
输入格式
第一行一个整数 N。
接下来 N 行,第 i 行表示 i 号职员的快乐指数 Hi。
接下来 N−1 行,每行输入一对整数 L,K,表示 K 是 L 的直接上司。
输出格式
输出最大的快乐指数。
数据范围
1 ≤ N ≤ 6000
128 ≤ Hi ≤127
输入样例:
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
输出样例:
5
状态表示:f [ u , 0 ]表示所有从以 u 为根的子树中选择,并且不选 u 这个点的方案
f [ u , 1 ]表示所有从以 u 为根的子树中选择,并且选 u 这个点的方案
f [ u , 0 ] = f [ u , 0 ] + max( f [ son , 0 ] ,f [ son , 1 ] );
f [ u , 1 ] =∑ f [ son , 0 ] ;
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=6010;
int n;
int happy[N];
int h[N],e[N],ne[N],idx;
int f[N][2];
bool has_father[N];
void add(int a,int b)
{
e[idx]=b;
ne[idx]=h[a];
h[a]=idx++;
}
void dfs(int u)
{
f[u][1]=happy[u];
for(int i=h[u];i!=-1;i=ne[i])
{
int j=e[i];
dfs(j);
f[u][0]+=max(f[j][0],f[j][1]);
f[u][1]=f[u][1]+f[j][0];
}
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&happy[i]);
memset(h,-1,sizeof h);
for(int i=0;i<n-1;i++)
{
int a,b;
scanf("%d %d",&a,&b);
has_father[a]=true;
add(b,a);
}
int root=1;
while(has_father[root])//找到头结点
root++;
dfs(root);
printf("%d\n",max(f[root][1],f[root][0]));
}
换根dp
我也不知道换根dp到底是啥意思,可能是假设一下每个点为根节点来计算信息
树的中心(本题来自acwing算法提高课)
给定一棵树,树中包含 n 个结点(编号1~n)和 n−1 条无向边,每条边都有一个权值。
请你在树中找到一个点,使得该点到树中其他结点的最远距离最近。
输入格式
第一行包含整数 n。
接下来 n−1 行,每行包含三个整数 ai,bi,ci,表示点 ai 和 bi 之间存在一条权值为 ci 的边。
输出格式
输出一个整数,表示所求点到树中其他结点的最远距离。
数据范围
1 ≤ n ≤ 10000
1 ≤ ai,bi ≤ n
1 ≤ ci ≤ 105
输入样例:
5
2 1 1
3 2 1
4 3 1
5 1 1
输出样例:
2
思路:分别求出每个点到其他点的最远距离,求这些距离可以将路径分为两类,一类是从当前点往子节点走,另一类是向父节点走。
第一类:直接求最大距离即可
第二类: 最大距离 = 当前点走到父节点(固定)+父节点且不走回当前点的最远距离
父节点且不走回当前点的最远距离= max(从父节点往上走 和 从从父节点往下走)
但是这里有种可能是,父节点往下走的最远距离刚好经过当前点……那么如何解决捏?
只需要记录每个点往下走的次大值,那么遇到上述情况判断最大值经过内个点时,直接启用次大值就好啦!
#include<cstring>
#include<iostream>
using namespace std;
const int N=10010,M=N*2,INF=0x3f3f3f3f;
int n;
int h[N],e[M],w[M],ne[M],idx;
int d1[N],d2[N],p1[N],up[N];
bool is_leaf[N];
void add(int a,int b,int c)
{
e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
int dfs_d(int u,int father)
{
d1[u]=d2[u]=-INF;
for(int i=h[u];~i;i=ne[i])
{
int j=e[i];
if(j==father)
continue;
int d=dfs_d(j,u)+w[i];
if(d>=d1[u])
{
d2[u]=d1[u],d1[u]=d;
p1[u]=j;
}
else if(d>d2[u])
d2[u]=d;
}
if(d1[u]==-INF)
{
d1[u]=d2[u]=0;
is_leaf[u]=true;
}
return d1[u];
}
void dfs_u(int u,int father)
{
for(int i=h[u];~i;i=ne[i])
{
int j=e[i];
if(j==father)
continue;
//往下走的两种情况
if(p1[u]==j)
up[j]=max(up[u],d2[u])+w[i];
else
up[j]=max(up[u],d1[u])+w[i];
dfs_u(j,u);
}
}
int main()
{
cin>>n;
memset(h,-1,sizeof h);
for(int i=0;i<n-1;i++)
{
int a,b,c;
cin>>a>>b>>c;
add(a,b,c),add(b,a,c);
}
dfs_d(1,-1);//往下走
dfs_u(1,-1);//往上走
int res=d1[1];
for(int i=2;i<=n;i++)
if(is_leaf[i])
res=min(res,up[i]);
else
res=min(res,max(d1[i],up[i]));
printf("%d\n",res);
}