题目描述
输入描述
输出描述
输入样例
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
输出样例
5
数据说明
树形 dp 的板子题,如果从根向子结点进行遍历(因为每个结点只能有一个父亲,所以自上而下可以规避一些麻烦),显然对于每一个结点,选与不选只会影响它的子结点是否能够被选入,因此可用 f ( i , 0 / 1 ) 描述每一个状态,其中第一维代表该结点的编号,第二维表示是否被选入,其中 0 代表不选,1 代表可选。因为只会影响到子结点的状态,由此可得到状态转移方程:
f
[
i
]
[
0
]
=
Σ
m
a
x
(
f
[
i
]
[
0
]
,
f
[
i
]
[
1
]
)
f[i][0]=Σ max(f[i][0],f[i][1])
f[i][0]=Σmax(f[i][0],f[i][1])
f
[
i
]
[
1
]
=
Σ
f
[
i
]
[
0
]
+
a
[
i
]
f[i][1]=Σf[i][0]+a[i]
f[i][1]=Σf[i][0]+a[i]
设根节点的根号为 n ,最后返回的值即为 f [ n ][ 1 ] 及 f [ n ][ 0 ] 之间的最大值,需要注意当仅有一个根节点时的特判。
参考代码
#include <bits/stdc++.h>
using namespace std;
const int N=6005;
struct edge{
int v,next;
}e[N];
int head[N],n,cnt,f[N][2],ans,ru[N],vis[N];
void add(int u,int v){
e[++cnt].v=v;//v记载的是该点本身的编号
e[cnt].next=head[u];
head[u]=cnt;
}
void dfs(int k){
vis[k]=1;
for(int i=head[k];i;i=e[i].next){//枚举该节点的所有孩子
if(vis[e[i].v])
continue;
dfs(e[i].v);
f[k][1]+=f[e[i].v][0];
f[k][0]+=max(f[e[i].v][0],f[e[i].v][1]);//对于其每个孩子是否选择的最大值累加
}
return;
}
int main(){
cin>>n;
if(n==1){//特判仅有根节点情况
int x;
cin>>x;
cout<<x;
return 0;
}
for(int i=1;i<=n;i++)
cin>>f[i][1];//每个点的权值 直接转化为选择第i点的状态方程
for(int i=1;i<=n;i++){
int l,k;
cin>>l>>k;
ru[l]=1;//即是有父亲的节点
add(k,l);
}
for(int i=1;i<=n;i++){
if(ru[i]==0){//从根节点开始
dfs(i);
cout<<max(f[i][1],f[i][0]);
return 0;
}
}
return 0;
}