bzoj 4033: [HAOI2015]T1(树形DP)
2016-05-11 10:47
399 查看
4033: [HAOI2015]T1
Time Limit: 10 Sec Memory Limit: 256 MBSubmit: 819 Solved: 375
[Submit][Status][Discuss]
Description
有一棵点数为 N 的树,树边有边权。给你一个在 0~ N 之内的正整数 K ,你要在这棵树中选择 K个点,将其染成黑色,并将其他 的
N-K个点染成白色 。 将所有点染色后,你会获得黑点两两之间的距
离加上白点两两之间的距离的和的受益。问受益最大值是多少。
Input
第一行包含两个整数 N, K 。接下来 N-1 行每行三个正整数 fr, to, dis , 表示该树中存在一条长度
为 dis 的边 (fr, to) 。输入保证所有点之间是联通的。
Output
输出一个正整数,表示收益的最大值。Sample Input
3 11 2 1
1 3 2
Sample Output
3HINT
对于 100% 的数据, 0<=K<=N <=2000Source
鸣谢bhiaibogf提供[Submit][Status][Discuss]
题解:树形dp
f[i][j]表示从i的子树中选j个黑点的最大收益值。
对于一条边c[i],他的贡献就是(他子树内的黑点个数×子树外的黑点个数+子树内白点的个数×子树外白点的个数)×c[i],因为是黑点和白点之间两两连边,要到达这棵子树,那么就必须经过这条边,所以这条边被经过的次数是一定的。
那么如果转移呢?
f[x][j]=max(f[x][j],f[x][j-l]+f[v[i]][l](当前子树的贡献)+当前边的贡献)
#include<iostream> #include<cstdio> #include<algorithm> #include<cstring> #include<cmath> #define LL long long using namespace std; LL f[2003][2003]; int tot,next[4003],point[4003],size[2003],v[4003],c[4003]; int n,k; void add(int x,int y,int z) { tot++; next[tot]=point[x]; point[x]=tot; v[tot]=y; c[tot]=z; tot++; next[tot]=point[y]; point[y]=tot; v[tot]=x; c[tot]=z; } void dfs(int x,int fa) { f[x][0]=f[x][1]=0; size[x]=1; for (int i=point[x];i;i=next[i]) if (v[i]!=fa) { dfs(v[i],x); size[x]+=size[v[i]]; for (int j=min(k,size[x]);j>=0;j--) for (int l=0;l<=min(j,size[v[i]]);l++) { LL ans1=(LL)(l*(k-l))*c[i]; LL ans2=(LL)(size[v[i]]-l)*(n-k-(size[v[i]]-l))*c[i]; f[x][j]=max(f[x][j],f[x][j-l]+f[v[i]][l]+ans1+ans2); } } } int main() { scanf("%d%d",&n,&k); memset(f,128,sizeof(f)); for (int i=1;i<n;i++) { int x,y,z; scanf("%d%d%d",&x,&y,&z); add(x,y,z); } dfs(1,0); printf("%lld\n",f[1][k]); }
相关文章推荐
- 安卓动画
- 图片轮播所遇到的问题
- android 网络加载图片,对图片资源进行优化,并且实现内存双缓存 + 磁盘缓存
- 设置时间 new Date
- IOS开发中如何使用通知NSNotification传值
- TCP/IP详解学习笔记(4)-ICMP协议,ping和Traceroute
- RxJava_ _学了下RxJava
- 配置类库dll所在的目录下的相对路径
- sql server 定期自动清理日志
- auto_ptr作为vector的元素会出现什么情况
- 使用Java管理Azure(1):基础配置
- JavaScript从数组的indexOf()深入之Object的Property机制
- Android中Fragmen首选项使用自定义的ListPreference的方法
- Python开发工具PyCharm个性化设置
- [BZOJ4033][HAOI2015]T1(树形dp)
- MySQL存储引擎总结
- Spring4新特性:Groovy Bean定义DSL
- Activity(一)
- LeetCode 012: Integer to Roman
- 单例模式 Singleton Pattern