hdu 1561 The more,The better

樹形依賴揹包,用多叉轉換成二叉的形式解決

一個節點兩個孩子,左孩子是它的孩子中的第一個,右孩子是它的第一個右兄弟,這樣一直遞歸下去,所以只有左孩子是節點的真正孩子

dp[i][j]=max{dp[r[i]][j], dp[l[i][k]] + dp[r[i]][j-1-k] +w[i] },0<=k<j

表示要麼全部分配給兄弟節點,要麼分給孩子一部分,分給兄弟一部分,既然要分給孩子,它自己就必須被選


代碼:

include<iostream>
#include<memory.h>
#include<string>
#include<cstdio>
#include<algorithm>
#include<math.h>
#include<stack>
#include<queue>
#include<vector>
#include<map>
using namespace std;
const int MAX=205;
struct node 
{
	int v,next;
}g[MAX*MAX];
int adj[MAX],dp[MAX][MAX],val[MAX],e,n,m,l[MAX],r[MAX];
void add(int u,int v)
{
	g[e].v=v; g[e].next=adj[u]; adj[u]=e++;
}
void Tree_add(int u,int v)
{
	if(l[u]==-1)
	{
		l[u]=v;
	}
	else
	{
		u=l[u];
		while(r[u]!=-1)
			u=r[u];
		r[u]=v;
	}
}
void dfs(int u,int k)
{
	int i,j,v; 
	for(i=adj[u];i!=-1;i=g[i].next)
	{
		v=g[i].v;
		memcpy(dp[v],dp[u],sizeof(dp[u]));
		dfs(g[i].v,k-1);
		for(j=1;j<=k;j++)
		{
			dp[u][j]=max(dp[u][j],dp[v][j-1]+val[v]);
		}
	}
}
int dfs2(int u,int k)
{
	int i,v;
	int a1,a2,a3;
	if(u==-1||k==0)
		return 0;
	if(dp[u][k]!=-1)
		return dp[u][k];
	a1=dfs2(r[u],k);
	dp[u][k]=a1;
	for(i=0;i<k;i++)
	{
		a2=dfs2(l[u],i);
		a3=dfs2(r[u],k-1-i);
		//dp[u][k]=max(dp[u][k],a1+a2+val[u]);
		dp[u][k]=max(dp[u][k],a2+a3+val[u]);
	}
	return dp[u][k];
}
int main()
{
	int i,j,k,a,b;
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		if(m==0&&n==0)
			break;
		memset(adj,-1,sizeof(adj));
		memset(r,-1,sizeof(r));
		memset(l,-1,sizeof(l));
		e=0;
		memset(dp,-1,sizeof(dp));
		for(i=1;i<=n;i++)
		{
			scanf("%d%d",&a,&val[i]);
			add(a,i);
			Tree_add(a,i); 
		}
		//dfs(0,m);
		dfs2(0,m+1);
		printf("%d\n",dp[0][m+1]);
	}
	return 0;
}


發佈了242 篇原創文章 · 獲贊 5 · 訪問量 23萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章