Starship Troopers(HDU 1011 树形DP)

题意:

  给定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]=;
int temp=(cav[p].bugs+)/;
for(i=temp;i<=m;i++)
dp[p][i]=cav[p].val;
for(i=;i<node[p].size();i++)
{
int t=node[p][i];
if(vis[t])
continue;
dfs(t);
for(j=m;j>=temp;j--) //回溯时完成对该点的动态规划
for(k=;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==-&&m==-)
break;
memset(vis,,sizeof(vis));
memset(dp,,sizeof(dp));
for(i=;i<=n;i++) node[i].clear();
for(i=;i<=n;i++)
scanf("%d%d",&cav[i].bugs,&cav[i].val);
for(i=;i<n-;i++)
{
scanf("%d%d",&a,&b);
node[a].push_back(b);
node[b].push_back(a);
}
if(m==)
{
printf("0\n");
continue;
}
dfs();
printf("%d\n",dp[][m]);
}
return ;
}
上一篇:致第一次安装RIME的你


下一篇:Java多线程,哲学家就餐问题