您的位置:首页 > 其它

hdu1561 zoj3201

2014-11-09 14:02 393 查看
本题同样是一道树形dp的题,即在树上进行背包操作,中文题意。解题思路:将有依赖关系的链接起来形成一棵树,把所有没有依赖关系的点都和0点相连,几多出一个点0,即需要攻破m+1个城堡获得的宝物,这样就可已在树上进行被曝操作了。有当前节点能被进攻的条件是他的亲节点已经被攻击了,因此dp[root][j+1],表示当前攻击了root节点,一共攻击了j+1各节点,为什么不是j,而是j+1;+1是为了保证他的父亲节点也被攻击,这就是他与zoj3201 的区别,zoj3201同样是求节点数为m个的字数的点权最大,但在写状态转移方程是,只需用dp[root][j],表示;两题有相同之处,也有不同,将两题比较求解分析,更能让我体会书上背包问题的求解:

hdu 1561代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 205

int n,m;
int dp

,aa
,dp1

;
struct node
{
int st,en,next;
}e[2*N];
int p
,num;
void init()
{
memset(p,-1,sizeof(p));
num=0;
}
void add(int st,int en)
{
e[num].st=st;
e[num].en=en;
e[num].next=p[st];
p[st]=num++;
}
void dfs(int root,int fa)
{
dp[root][1]=aa[root];
for(int i=p[root];i+1;i=e[i].next)
{
int son=e[i].en;
if(son==fa)continue;
dfs(son,root);
for(int j=m+1;j>=1;j--)
{
for(int k=1;k<=j;k++)
{
dp[root][j+1]=max(dp[root][j+1],dp[son][j+1-k]+dp[root][k]);
}
}
}
}
int main()
{
int a,b;
while(scanf("%d%d",&n,&m)!=EOF)
{
if(m==0&&n==0)break;
init();
memset(dp,0,sizeof(dp));
aa[0]=0;
for(int i=1;i<=n;i++)
{
scanf("%d%d",&a,&b);
add(a,i);
add(i,a);
aa[i]=b;
}
dfs(0,-1);
printf("%d\n",dp[0][m+1]);
}
return 0;
}


zoj3201代码:
#include<cstdio>
#include<queue>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 105
int a
,k;
int dp

;

struct node
{
int st,en,next;
}e[N*2];
int num,p
;
void init()
{
memset(p,-1,sizeof(p));
num=0;
}
void add(int st,int en)
{
e[num].st=st;
e[num].en=en;
e[num].next=p[st];
p[st]=num++;
}
int dfs(int root,int fa)
{
dp[root][1]=a[root];
for(int i=p[root];i+1;i=e[i].next)
{
int son=e[i].en;
if(son==fa)continue;
dfs(son,root);
for(int j=k;j>0;j--)
{
for(int tt=1;tt<=j;tt++)
dp[root][j]=max(dp[root][tt]+dp[son][j-tt],dp[root][j]);
}
}
}
int main()
{
int n,s,ee;
while(scanf("%d%d",&n,&k)!=EOF)
{
for(int i=0;i<=n;i++)
for(int j=0;j<=k;j++)
dp[i][j]=0;
init();
for(int i=0;i<n;i++)
{
scanf("%d",&a[i]);
}
for(int i=1;i<n;i++)
{
scanf("%d%d",&s,&ee);
add(s,ee);
add(ee,s);
}
dfs(0,-1);
int ans=-1;
for(int i=0;i<n;i++)
{
ans=max(ans,dp[i][k]);
}
printf("%d\n",ans);
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  树形dp