树形dp
1. what is 树形dp?
树形dp其实简单来讲就是以树这种结构来进行动态规划。
具体地,给定一棵有N个节点的树(一般为无根树,也就是有N-1条无向边),我们可以任选一个节点作为根结点,从而定义出每个节点的深度和每棵子树的根。
2. 实操
没有上司的舞会
题目描述
某大学有 n n n 个职员,编号为 1 … n 1\ldots n 1…n。
他们之间有从属关系,也就是说他们的关系就像一棵以校长为根的树,父结点就是子结点的直接上司。
现在有个周年庆宴会,宴会每邀请来一个职员都会增加一定的快乐指数 r i r_i ri,但是呢,如果某个职员的直接上司来参加舞会了,那么这个职员就无论如何也不肯来参加舞会了。
所以,请你编程计算,邀请哪些职员可以使快乐指数最大,求最大的快乐指数。
输入格式
输入的第一行是一个整数 n n n。
第 2 2 2 到第 ( n + 1 ) (n + 1) (n+1) 行,每行一个整数,第 ( i + 1 ) (i+1) (i+1) 行的整数表示 i i i 号职员的快乐指数 r i r_i ri。
第 ( n + 2 ) (n + 2) (n+2) 到第 2 n 2n 2n 行,每行输入一对整数 l , k l, k l,k,代表 k k k 是 l l l 的直接上司。
输出格式
输出一行一个整数代表最大的快乐指数。
样例 #1
样例输入 #1
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
样例输出 #1
5
提示
数据规模与约定
对于 100 % 100\% 100% 的数据,保证 1 ≤ n ≤ 6 × 1 0 3 1\leq n \leq 6 \times 10^3 1≤n≤6×103, − 128 ≤ r i ≤ 127 -128 \leq r_i\leq 127 −128≤ri≤127, 1 ≤ l , k ≤ n 1 \leq l, k \leq n 1≤l,k≤n,且给出的关系一定是一棵树。
设定f的二维数组
1.f [x][0]:
表示为:以x为根的子树中,邀请一部分职员参加舞会,且x不参加舞会时的快乐指数最大值,此时他的子节点都可以参加舞会。如下:
2.f[x][1]:
表示为:以x为根的子树中,邀请一部分职员参加舞会,且x参加舞会时的时的快乐指数最大值,此时他的子节点都不能参加舞会。如下:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
ll read(){
ll ans = 0 , f = 1;
char ch = getchar();
while(!isdigit(ch)){
f *= (ch == '-') ? -1 : 1;
ch = getchar();
}
while(isdigit(ch)){
ans = (ans << 3) + (ans << 1) + (ch ^ 48);
ch = getchar();
}
return ans * f;
}
#define in(a) (a) = read();
#define out(a) printf("%d" , (a));
#define space() printf(" ");
#define enter() printf("\n");
const int MAXN = 1e5 + 5;
int m[MAXN];
int h[MAXN] , p[MAXN] , e[MAXN] , idx;
bool vis[MAXN];
int f[MAXN][2];
void add(int a , int b){
e[idx] = b;
p[idx] = h[a];
h[a] = idx;
idx ++;
}
void dfs(int a){
f[a][1] = m[a];
for(int i = h[a];i != -1;i = p[i]){
int j = e[i];
dfs(j);
f[a][0] += max(f[j][0] , f[j][1]);
f[a][1] += f[j][0];
}
}
int ans , n , r = 1;
signed main(){
in(n);
memset(h , -1 , sizeof(h));
for(int i = 1;i <= n; i ++)
in(m[i]);
for(int i = 1;i < n; i ++){
int a , b;
in(a); in(b);
vis[a] = 1;
add(b , a);
}
while(vis[r]) r ++;
dfs(r);
ans = max(f[r][0] , f[r][1]);
out(ans);
return 0;
}
//7
//1
//1
//1
//1
//1
//1
//1
//1 3
//2 3
//6 4
//7 4
//4 5
//3 5