JZOJ 4814. 【NOIP2016提高A组五校联考2】tree
来源:互联网 发布:单片机c语言指令集 编辑:程序博客网 时间:2024/05/22 09:28
Problem
Description
给一棵n 个结点的有根树,结点由1 到n 标号,根结点的标号为1。每个结点上有一个物品,第i 个结点上的物品价值为vi。
你需要从所有结点中选出若干个结点,使得对于任意一个被选中的结点,其到根的路径上所有的点都被选中,并且选中结点的个数不能超过给定的上限lim。在此前提下,你需要最大化选中结点上物品的价值之和。
求这个最大的价值之和。
Input
第一行为两个整数n; lim
接下来n 行,第i 行包含一个整数vi,表示结点i 上物品的价值。
接下来n- 1 行,每行包含两个整数u; v, 描述一条连接u; v 结点的树边。
Output
输出一行答案。
Sample Input
6 4
-5
4
-6
6
9
6
3 2
3 1
2 4
2 5
1 6
Sample Output
2
Data Constraint
对于前20% 的数据,1<=n; lim<=10
对于前60% 的数据,1<=n; lim<=100
对于100% 的数据,1<=n; lim<=3000; |vi| <=10^5 数据有梯度,保证给出的是合法的树。
Solution
树形DP,
DP方程显然。
Code
#include<iostream>#include<cstdio>#include<cmath>#include<cstring>#define N 3010#define fo(i,a,b) for(i=a;i<=b;i++)using namespace std;int v[N],a[N][N],b[N][N],f[N][N],deep[N];int mx,i,j,n,lim,x,y,tot;bool bz[N];void dg(int k,int fa){ int i; fo(i,1,a[k][0]) { if (a[k][i]==fa) continue; deep[a[k][i]]=deep[k]+1; dg(a[k][i],k); }}void dfs(int k){ int i,j; bz[k]=1; fo(i,1,b[k][0]) { if (bz[b[k][i]]) continue; memcpy(f[b[k][i]],f[k],sizeof(f[k])); dfs(b[k][i]); fo(j,1,lim) f[k][j]=max(f[k][j],f[b[k][i]][j-1]+v[b[k][i]]); }}int main(){ scanf("%d%d",&n,&lim); fo(i,1,n) scanf("%d",&v[i]); fo(i,1,n-1) { scanf("%d%d",&x,&y); a[x][++a[x][0]]=y; a[y][++a[y][0]]=x; } dg(1,0); fo(i,1,n) fo(j,1,a[i][0]) if (deep[i]<deep[a[i][j]]) b[i][++b[i][0]]=a[i][j]; dfs(1); printf("%d",v[1]+f[1][lim-1]);}
3 0
- JZOJ 4814. 【NOIP2016提高A组五校联考2】tree
- 【jzoj 4814】【NOIP2016提高A组五校联考2】tree
- JZOJ 4814 【NOIP2016提高A组五校联考2】tree
- jzoj 4814. 【NOIP2016提高A组五校联考2】tree 树形dp
- 【NOIP2016提高A组五校联考2】tree
- 【NOIP2016提高A组五校联考2】tree
- 【NOIP2016提高A组五校联考2】tree
- tree 【NOIP2016提高A组五校联考2】
- JZOJ 4812 【NOIP2016提高A组五校联考2】string
- [jzoj 4812]【NOIP2016提高A组五校联考2】string
- JZOJ 4813 【NOIP2016提高A组五校联考2】running
- 【JZOJ 4814】【NOIP2016提高组 五校联考2】tree
- 【JZOJ4814】【NOIP2016提高A组五校联考2】tree
- JZOJ4814. 【NOIP2016提高A组五校联考2】tree
- jzoj 4811. 【NOIP2016提高A组五校联考1】排队
- JZOJ 4809 【NOIP2016提高A组五校联考1】挖金矿
- JZOJ 4810 【NOIP2016提高A组五校联考1】道路规划
- JZOJ 4811 【NOIP2016提高A组五校联考1】排队
- 关于Ajax请求跨域问题之个人见解
- (jdbcType)mybatis 查询Oracle Date类型字段只精确到年月日
- 贝叶斯分类器
- [2016-10-05]一些感悟2
- 关于AndroidManifest.xml报Use ' instead of '.' for inner classes....警报的解决办法
- JZOJ 4814. 【NOIP2016提高A组五校联考2】tree
- 天才第一步
- 【NOIP2016提高A组五校联考2】string
- Zookeeper节点类型
- Wireshark数据抓包教程之认识捕获分析数据包
- 最初的时间
- 拷贝构造函数和赋值运算符函数
- JAVA设计模式之单例模式
- 衰老的舌头