POJ 2486 树形DP
2016-01-02 13:02
239 查看
有一颗苹果树,每一个节点上面有非常多苹果,从一个节点到另外一个能够到达的节点花费1步。求k步最多能吃到多少苹果,起始点为1,能够不回到起始点。
这是典型的回溯型树状dp。
dp[i][j][0]代表以i为根节点的子树最多j步后回到i能吃到的最多的苹果,
dp[i][j][1]代表以i为根节点的子树最多j步后不回到i节点最多能吃到的子树。那么状态转移就分三步了。
(1)dp[i][j+2][0] = max(dp[i][j+2][0], dp[i][j-k][0]+dp[son][k][0]);
(2)dp[i][j+1][1] = max(dp[i][j+1][1], dp[i][j-k][0]+dp[son][k][1]); 人留在i的子节点son的子树中
(3)dp[i][j+2][1] = max(dp[i][j+2][1], dp[i][j-k][1]+dp[son][k][0]); 人留在不是son的i的子节点的子树中
这是典型的回溯型树状dp。
dp[i][j][0]代表以i为根节点的子树最多j步后回到i能吃到的最多的苹果,
dp[i][j][1]代表以i为根节点的子树最多j步后不回到i节点最多能吃到的子树。那么状态转移就分三步了。
(1)dp[i][j+2][0] = max(dp[i][j+2][0], dp[i][j-k][0]+dp[son][k][0]);
(2)dp[i][j+1][1] = max(dp[i][j+1][1], dp[i][j-k][0]+dp[son][k][1]); 人留在i的子节点son的子树中
(3)dp[i][j+2][1] = max(dp[i][j+2][1], dp[i][j-k][1]+dp[son][k][0]); 人留在不是son的i的子节点的子树中
#include "stdio.h" #include "string.h" #include "vector" using namespace std; vector< vector<int> >data(110); int dp[110][210][3],vis[110],v[110]; int n,m; int Max(int a,int b) { if (a<b) return b;else return a; } void dfs(int cur) { int i,next,j,k; vis[cur]=1; for (i=0;i<=m;i++) dp[cur][i][0]=dp[cur][i][1]=v[cur]; for (i=0;i<data[cur].size();i++) { next=data[cur][i]; if (vis[next]==0) { dfs(next); for (j=m;j>=0;j--) for (k=0;k<=j;k++) { dp[cur][j+2][0]=Max(dp[cur][j+2][0],dp[cur][j-k][0]+dp[next][k][0]); dp[cur][j+1][1]=Max(dp[cur][j+1][1],dp[cur][j-k][0]+dp[next][k][1]); dp[cur][j+2][1]=Max(dp[cur][j+2][1],dp[cur][j-k][1]+dp[next][k][0]); } } } } int main() { int i,x,y; while (scanf("%d%d",&n,&m)!=EOF) { for (i=1;i<=n;i++) { data[i].clear(); scanf("%d",&v[i]); } for (i=2;i<=n;i++) { scanf("%d%d",&x,&y); data[x].push_back(y); data[y].push_back(x); } memset(vis,0,sizeof(vis)); memset(dp,0,sizeof(dp)); dfs(1); printf("%d\n",dp[1][m][1]); } return 0; }
相关文章推荐
- 电脑爱好者GHOSTWIN7纯净版v2.0
- C# HEX字符串转HEX值的方法
- centos7搭建LAMP平台环境
- SVM学习笔记(二)----手写数字识别
- 关于C++的const
- c#小结(4)---面向对象
- W5500EVB TCP Client模式设置说明
- Spring 学习笔记 2. 尚硅谷_佟刚_Spring_IOC&DI概述
- VS2015代码自动对齐
- 获取手机上已安装应用信息
- 如何获取Java层的函数在Dalvik中对应的Method?
- Android登陆界面实现-支持输入框清楚和震动效果功能
- cocos2d-x 新建项目 Cannot open include file: ‘cocos2d.h’
- apache httpclient CacheStorage的一个自定义实现
- ssh整合(http://blog.csdn.net/songanling/article/details/22454973)
- [Canvas前端游戏开发]——FlappyBird详解
- C语言怎么获得进程的PE文件信息
- 未名湖边的烦恼
- Camera Calibration 相机标定
- 监督学习之再聊支持向量机——Andrew Ng机器学习笔记(六)