您的位置:首页 > 运维架构

Starship Troopers(HDU 1011 树形DP)

2016-03-01 19:42 495 查看
题意:

  给定n个定点和m个士兵,n个定点最终构成一棵树,每个定点有一定x个bugs和y个value,每20个bug需要消耗一个士兵,不足20也消耗一个,然后最终收获y个value,只有父节点被占领后子节点才有被占领的可能。

DP状态转移方程:

  dp[p][j]=max(dp[p][j],dp[p][j-k]+dp[t][k]);

看的王大神的代码,DFS写的,先从叶子节点开始向上遍历进行动态规划,自己看了dp方程也没写出来。。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
#define Max 102
struct node
{
int bugs,val;
}cav[Max];
vector<int> node[Max];
int n,m;
int vis[Max],dp[Max][Max];
int a,b;
void dfs(int p)
{
int i,j,k;
vis[p]=1;
int temp=(cav[p].bugs+19)/20;
for(i=temp;i<=m;i++)
dp[p][i]=cav[p].val;
for(i=0;i<node[p].size();i++)
{
int t=node[p][i];
if(vis[t])
continue;
dfs(t);
for(j=m;j>=temp;j--)            //回溯时完成对该点的动态规划
for(k=1;k<=j-temp;k++)
dp[p][j]=max(dp[p][j],dp[p][j-k]+dp[t][k]);        //保证父节点能够占领
}
return;
}
int main()
{
int i,j;
freopen("in.txt","r",stdin);
while(scanf("%d%d",&n,&m)!=EOF)
{
if(n==-1&&m==-1)
break;
memset(vis,0,sizeof(vis));
memset(dp,0,sizeof(dp));
for(i=0;i<=n;i++)    node[i].clear();
for(i=1;i<=n;i++)
scanf("%d%d",&cav[i].bugs,&cav[i].val);
for(i=0;i<n-1;i++)
{
scanf("%d%d",&a,&b);
node[a].push_back(b);
node[b].push_back(a);
}
if(m==0)
{
printf("0\n");
continue;
}
dfs(1);
printf("%d\n",dp[1][m]);
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: