算法提高 金属采集 树形DP
来源:互联网 发布:凶间疑影 知乎 编辑:程序博客网 时间:2024/04/27 18:17
题目链接:金属采集
思路:d(i, j)表示在以i为根结点的子树中使用j个机器人的最小花费。设v为u的一个子节点,从节点i使用k个机器人收集以v为根结点的能量,状态转移方程为d(u, i) = min(d(u, i - k) + d(v, k) + cost * k) 1 <= k <= i. 注意d(u, i - k)表示用i - k个机器人去收集其他子树的能量的最小花费,在遍历所有子节点之后,d(u, i)的值才会固定。
AC代码:
#include <cstdio>#include <algorithm>#include <cstring>#include <utility>#include <string>#include <iostream>#include <map>#include <set>#include <vector>#include <queue>#include <stack>using namespace std;#define eps 1e-10#define inf 0x3f3f3f3f#define PI pair<int, int> const int maxn = 1e5 + 5;int dp[maxn][12]; struct node{int son, cost; node() {}node(int son, int cost):son(son), cost(cost) {}};vector<node>road[maxn];int n, s, cnt;void dfs(int u, int pre) { //u-当前节点,pre-它的父节点 int n = road[u].size();for(int i = 0; i < n; ++i) {int v= road[u][i].son, cost = road[u][i].cost;if(v == pre) continue;dfs(v, u);// 只考虑当前节点和已经发现的节点 for(int k = cnt; k >= 0; --k) { //该子树停留的机器人总数//逆序枚举--如果d(u, k)更新,那么不能影响d(u,k+1),如果顺序枚举会影响 dp[u][k] += dp[v][0] + cost * 2; //让一个机器人遍历该子树所有子节点并返回的花费 for(int j = 1; j <= k; ++j) { //至少一个机器人进入子树 dp[u][k] = min(dp[u][k], dp[u][k-j] + dp[v][j] + j * cost);}} } }int main() {while(scanf("%d%d%d", &n, &s, &cnt) == 3) {memset(dp, 0, sizeof(dp));for(int i = 0; i <= n; ++i) road[i].clear();int x, y, cost;for(int i = 1; i < n; ++i) {scanf("%d%d%d", &x, &y, &cost);road[x-1].push_back(node(y-1, cost));road[y-1].push_back(node(x-1, cost));}dfs(s - 1, -1);printf("%d\n", dp[s - 1][cnt]); }return 0;}
如有不当之处欢迎指出!
0 0
- 算法提高 金属采集 树形DP
- hdu4003&&蓝桥杯, 算法提高 金属采集 (树形DP,经典。。。)
- 算法提高 金属采集
- 蓝桥杯 算法提高 金属采集 (树形动态规划)
- 蓝桥杯_算法提高_金属采集(树形动态规划)
- 蓝桥杯 算法提高-金属采集
- 蓝桥杯 金属采集(树形dp)
- 树形动态规划-金属采集
- 蓝桥杯 金属采集 树形动态规划
- 采集金属
- 蓝桥杯 算法提高 结点选择(树形DP)
- 树状-金属采集
- ADV-3-金属采集
- CJOJ 2366 机器人采集金属
- 算法提高 拿糖果 DP
- 算法提高 求最大值 DP
- tarjan算法求缩点+树形DP求直径
- 算法训练 结点选择 【树形dp】
- Java - 反射机制(Reflection)
- 内部类
- 最长公共子序列
- 使用 position-sticky 实现粘性布局
- 快速搭建虚拟机集群
- 算法提高 金属采集 树形DP
- java 服务端对接微信支付 Demo
- 使用 Builder 模式构造 JavaBean 的好处
- dva修改端口
- Android 小功能
- js循环语句,for-in,with,函数
- 影响JavaScript应用可扩展性因素
- Java - 网络编程(NetWork)
- consul学习笔记