【bzoj4033】 T1 树形dp
2015-07-07 22:01
302 查看
表示长这么大第一次写树形背包,感觉有些难理解。
f[i][j]表示考虑以i为根这棵子树,有j个黑色结点的最大贡献值为多少。
则每次转移的时候我们需要考虑每条边对答案作出的贡献,为((k-j)*j+(size[i]-j)*(n-k+j-size[i]))*w。
利用那种类似树形背包的dp,O(n^2)解决。
f[i][j]表示考虑以i为根这棵子树,有j个黑色结点的最大贡献值为多少。
则每次转移的时候我们需要考虑每条边对答案作出的贡献,为((k-j)*j+(size[i]-j)*(n-k+j-size[i]))*w。
利用那种类似树形背包的dp,O(n^2)解决。
#include<cstdio> #include<cstring> #include<cstdlib> #include<cmath> #include<iostream> #include<algorithm> #define maxn 2010 using namespace std; int n,m,k,num; int head[maxn],to[2*maxn],next[2*maxn],v[2*maxn],size[maxn]; long long tmp[maxn]; long long f[maxn][maxn]; void addedge(int x,int y,int z) { num++;to[num]=y;v[num]=z;next[num]=head[x];head[x]=num; } long long calc(int x,int y) { return (long long)y*(x-y); } void dfs(int x,int fa,int w) { size[x]=1; for (int p=head[x];p;p=next[p]) if (to[p]!=fa) { dfs(to[p],x,v[p]); memcpy(tmp,f[x],sizeof(tmp)); for (int i=0;i<=min(size[x],k);i++) for (int j=0;j<=min(size[to[p]],k-i);j++) tmp[i+j]=max(tmp[i+j],f[x][i]+f[to[p]][j]); memcpy(f[x],tmp,sizeof(tmp)); size[x]+=size[to[p]]; } for (int i=0;i<=min(size[x],k);i++) f[x][i]+=(calc(k,i)+calc(n-k,size[x]-i))*w; } int main() { scanf("%d%d",&n,&k); for (int i=1;i<n;i++) { int x,y,z; scanf("%d%d%d",&x,&y,&z); addedge(x,y,z); addedge(y,x,z); } dfs(1,0,0); printf("%lld\n",f[1][k]); return 0; }
相关文章推荐
- eclipse构建maven的web项目
- 常用财务软件:用友,金蝶,新中大,速达,管家婆,金算盘,远方,远光,金钥匙,润衡,浪潮,上海博科,易商,任我行,千方百剂,智管,小蜜蜂,SAP,ORACLE,SSA,QAD,MAPICS,JDE。
- bootstrap-table(一)
- IOS中NSUserDefaults的用法(轻量级本地数据存储)
- table插入标签form标记怪现象
- 常用财务软件:用友,金蝶,新中大,速达,管家婆,金算盘,远方,远光,金钥匙,润衡,浪潮,上海博科,易商,任我行,千方百剂,智管,小蜜蜂,SAP,ORACLE,SSA,QAD,MAPICS,JDE。
- Find Peak Element
- 黑马程序员---Java概述与jdk的安装和配置(一)
- BZOJ3456【分治FFT】
- 查找总结
- 查看mysql数据文件存放路径
- SpringMVC 使用JSR-303进行校验 @Valid
- centos7初步命令
- leetcode 010 —— Regular Expression Matching(hard)
- 【2-3】Java 中的原生数据类型
- 非线性结构之二叉树
- [华为机试真题]72.操作系统任务调度问题
- 虚幻4蓝图:流程控制(一)
- 面试常问的40个问题 附带经典答案!
- 【Java】若MxN矩阵中某个元素为0, 则将其所在的行与列清零