码迷,mamicode.com
首页 > 其他好文 > 详细

多叉树转二叉树+树形dp(codevs 1746 贪吃的九头龙 2002noi)

时间:2017-06-08 21:28:57      阅读:169      评论:0      收藏:0      [点我收藏+]

标签:using   break   lan   付出   递归   imp   树根   roo   target   

题目传送门

看到这个题目我们要先把问题简化了,条件中是多叉树,我们可以把它转换成二叉树,左边是儿子右边是兄弟的储存方式。

首先先判断否的部分,当总的果子小于需求,也就是N-k<M-1时输出-1

我们再判断是的部分

如果没有大头,一定存在难受值为0的方案但是现在题目中有大头,我们就可以按按照小头的个数进行分类

1.有一个小头,我们要考虑小头和大头的难受值之和。

2.有多个小头,因为小头可以在奇偶的进行变换,所以我们只需要考虑大头的难受值。

分析到这里,我们就可以发现是树形dp我们设f[i][j][k]表示以i为根的子树,大头吃j个果子,他父亲节点为kk=1是大头吃的,k=0是小头吃的。

下面分析的部分摘自kiana810的博客:

 

f[i][j][k]=min(f1[i][j][k],f2[i][j][k])

1f1[i][j][k]=f[son][j1][1]+f[brother][j-j1-1][k]+d(k,1)*cost(i,fa[i]);(j1<j)

2f2[i][j][k]=f[son][j1][0]+f[brother][j-j1][k]+d(k,0)*cost(i,fa[i]);(j1j)

其中sonbrother分别表示i的左儿子和右兄弟(对应二叉树左右子树),j1是枚举的变量,costifa[i]的树枝难受值。d的定义如下:

1)若i=1j=1,则大头吃下了相连的果子,需要付出难受值,返回1

2)若i=0j=0,但M=2,只有一个小头,小头吃下了相连的果子,需要付出难受值,返回1

3)其它情况,总有办法分配果子使得九头龙不付出难受值,返回0

 

思考完之后,我们不要忘记考虑边界 f[0][0][k]=0,f[0][j][k]=inf这很好理解,我们假设有一个虚拟节点,他是二叉树根节点的根节点。

分析完之后,放上代码吧,写完之后我感觉有些疲倦,兴许是我太菜了呢~

#include<bits/stdc++.h>
#define inf 129312910
using namespace std;
int N,M,K,root;
vector<int> g[400];//存权值 
vector<int> p[400];//存子节点 
int f[400][400][2];
int fa[400];

struct Point
{
	int l,r,c;//分别为左儿子,右儿子,和儿子到父亲的权值 
	Point(){l=r=c=0;}
}t[400];


void build(int x)
{
	if(p[x].empty()) return ;
	t[x].l=p[x][0];t[p[x][0]].c=g[x][0];
	for(int i=1;i<p[x].size();i++)
	{
		t[p[x][i-1]].r=p[x][i];//只找儿子节点,并且一次在后插入 
		t[p[x][i]].c=g[x][i];
	}
	for(int i=0;i<p[x].size();i++) build(p[x][i]);
}

int d(int x,int y)
{
	return (( x==0&&y==0&&M==2)||(x&&y));
}

int dp(int i,int j,int k)//递归dp的过程
{
	if(f[i][j][k]==-1)//如果之前没有搜索过
	{
		int v,u=inf;
		for(int a=0;a<=j;a++)
		{
			v=dp(t[i].l,a,0)+dp(t[i].r,j-a,k)+d(k,0)*t[i].c;
			u=min(u,v);
			if(a<j)//防止越界important!!!
				v=dp(t[i].l,a,1)+dp(t[i].r,j-a-1,k)+d(k,1)*t[i].c;
			u=min(u,v);
		}
		f[i][j][k]=u;
	}
	return f[i][j][k];
}

int main()
{
	cin>>N>>M>>K;
	if(N-K<M-1)//判断有无解
	{
		cout<<"-1\n";
		return 0;
	}
	int t1,t2,t3;
	int ans=0;
	for(int i=1;i<N;i++)
	{
		cin>>t1>>t2>>t3;
		p[t1].push_back(t2);
		g[t1].push_back(t3);
		fa[t2]=t1;
	}
	for(int i=1;i<=N;i++)
		if(!fa[i])
		{
			root=i;
			build(i);
			break;
		} 
	memset(f,-1,sizeof(f));//初始化为-1,也算是没有遍历过
	f[0][0][1]=f[0][0][0]=0;
	for(int i=1;i<=K;i++)
		f[0][i][0]=f[0][i][1]=inf;
	ans=dp(t[root].l,K-1,1);//从根节点的第一个节点,也就是在二叉树中的左儿子
	cout<<ans<<endl;
	return 0;
}

 

多叉树转二叉树+树形dp(codevs 1746 贪吃的九头龙 2002noi)

标签:using   break   lan   付出   递归   imp   树根   roo   target   

原文地址:http://www.cnblogs.com/foreverpiano/p/6964729.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!