POJ Apple Tree 题解(树形dp+01背包)
时间:2021-08-23
本文章向大家介绍POJ Apple Tree 题解(树形dp+01背包),主要包括POJ Apple Tree 题解(树形dp+01背包)使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。
题目链接
题目大意
有一个树,n个节点,第一个节点为根节点,每个点都有一个权值,每个点都可以移动到相邻的点,给你树的连接情况,求出,走k步最多获得多少权值?
题目思路
主要考虑回头的情况,所以不能简单的设置\(dp[i][j]\)
\(dp[i][j][0]\)表示在子树 i中最多走 j步最后还回到 i 能得到的最大苹果数。
\(dp[i][j][1]\) 表示在子树 i中最多走 j步最后不回到 i 能得到的最大苹果数
然后再跑01背包即可
代码
#include<set>
#include<map>
#include<queue>
#include<stack>
#include<cmath>
#include<cstdio>
#include<vector>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#define se second
#define debug cout<<"I AM HERE"<<endl;
using namespace std;
typedef long long ll;
const int maxn=2e2+5,inf=0x3f3f3f3f,mod=1e9+7;
const double eps=1e-6;
int n,k;
int a[maxn];
int dp[maxn][maxn][2];
vector<int> g[maxn];
void dfs(int u,int fa){
for(int i=0;i<=k;i++){
dp[u][i][0]=dp[u][i][1]=a[u];
}
for(int a=0;a<g[u].size();a++){
int x=g[u][a];
if(x==fa) continue;
dfs(x,u);
for(int i=k;i>=1;i--){
for(int j=0;j<=i;j++){
if(i-j-2>=0) dp[u][i][0]=max(dp[u][i][0],dp[x][j][0]+dp[u][i-j-2][0]);
if(i-j-1>=0) dp[u][i][1]=max(dp[u][i][1],dp[x][j][1]+dp[u][i-j-1][0]);
if(i-j-2>=0) dp[u][i][1]=max(dp[u][i][1],dp[x][j][0]+dp[u][i-j-2][1]);
}
}
}
}
signed main(){
while(scanf("%d%d",&n,&k)!=-1){
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
g[i].clear();
for(int j=0;j<=k;j++){
dp[i][j][0]=dp[i][j][1]=0;
}
}
for(int i=1,u,v;i<=n-1;i++){
scanf("%d%d",&u,&v);
g[u].push_back(v);
g[v].push_back(u);
}
dfs(1,-1);
int ans=0;
ans=max(ans,dp[1][k][0]);
ans=max(ans,dp[1][k][1]);
printf("%d\n",ans);
}
return 0;
}
卷也卷不过,躺又躺不平
原文地址:https://www.cnblogs.com/hunxuewangzi/p/15177594.html
- JavaScript 教程
- JavaScript 编辑工具
- JavaScript 与HTML
- JavaScript 与Java
- JavaScript 数据结构
- JavaScript 基本数据类型
- JavaScript 特殊数据类型
- JavaScript 运算符
- JavaScript typeof 运算符
- JavaScript 表达式
- JavaScript 类型转换
- JavaScript 基本语法
- JavaScript 注释
- Javascript 基本处理流程
- Javascript 选择结构
- Javascript if 语句
- Javascript if 语句的嵌套
- Javascript switch 语句
- Javascript 循环结构
- Javascript 循环结构实例
- Javascript 跳转语句
- Javascript 控制语句总结
- Javascript 函数介绍
- Javascript 函数的定义
- Javascript 函数调用
- Javascript 几种特殊的函数
- JavaScript 内置函数简介
- Javascript eval() 函数
- Javascript isFinite() 函数
- Javascript isNaN() 函数
- parseInt() 与 parseFloat()
- escape() 与 unescape()
- Javascript 字符串介绍
- Javascript length属性
- javascript 字符串函数
- Javascript 日期对象简介
- Javascript 日期对象用途
- Date 对象属性和方法
- Javascript 数组是什么
- Javascript 创建数组
- Javascript 数组赋值与取值
- Javascript 数组属性和方法