没有上司的舞会
题目描述
某大学有 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,且给出的关系一定是一棵树。
思路
树形dp
d
p
[
i
]
[
0
/
1
]
dp[i][0/1]
dp[i][0/1]
i
~~~~i
i 表示
i
i
i 的子树最大快乐指数
0
0
0 表示
i
i
i 来参加,
1
1
1 表示
i
i
i 不来参加
那么状态转移方程推导一下便可出来
d
p
[
n
o
w
]
[
0
]
+
=
m
a
x
(
d
p
[
v
]
[
0
]
,
d
p
[
v
]
[
1
]
)
;
dp[now][0]+=max(dp[v][0],dp[v][1]);
dp[now][0]+=max(dp[v][0],dp[v][1]);
d
p
[
n
o
w
]
[
1
]
+
=
d
p
[
v
]
[
0
]
;
dp[now][1]+=dp[v][0];
dp[now][1]+=dp[v][0];
用dfs解决,根节点需要预先找一下
代码
#include <iostream>
#include <vector>
using namespace std;
const int N=6010;
int n,q,root,a[N];
int dp[N][2];
vector<int> G[N];
bool k[N];
void dfs(int now)
{
dp[now][0]=0;
dp[now][1]=a[now];
for(int i=0;i<G[now].size();i++){
int v=G[now][i];
dfs(v);
dp[now][0]+=max(dp[v][0],dp[v][1]);
dp[now][1]+=dp[v][0];
}
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
cin>>a[i];
for(int i=1;i<=n-1;i++){
int u,v;cin>>v>>u;
G[u].push_back(v);
k[v]=1;
}
for(int i=1;i<=n;i++)
if(!k[i])
root=i;
dfs(root);
cout<<max(dp[root][0],dp[root][1]);
return 0;
}