树形DP

赶着学的树形DP,就先从最最最最最裸的树上DP题开始做起了,
前六道: ye-leng大佬的六道起步题。

一号板子题:

Anniversary party [POJ-2342]
题意:

给出每个点的快乐值当做点权,员工快乐时,他的直接领导不能在场,求最大快乐值。
题解:
f [ i ] [ j ] : f[i][j]: f[i][j]:表示第i个人状态为j(表示选或不选)的最大快乐值

状态转移方程式:
f [ f a ] [ 1 ] + = f [ i ] [ 0 ] f[fa][1]+=f[i][0] f[fa][1]+=f[i][0]//父亲的快乐+=儿子不快乐
f [ f a ] [ 0 ] + = m a x ( f [ i ] [ 0 ] , f [ i ] [ 1 ] ) f[fa][0]+=max(f[i][0],f[i][1]) f[fa][0]+=max(f[i][0],f[i][1])父亲不快乐+=儿子不快乐/快乐

代码:
#include<iostream>
#include<stdio.h>
using namespace std;
const int sea=6060;
inline int read()
{
	int s=0,w=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
	while(ch<='9'&&ch>='0')s=s*10+ch-'0',ch=getchar();
	return s*w;
}
int n,a[sea],x,y,root,fa[sea],v[sea],f[sea][2];//f[i][j]:表示第i个人状态为j(表示选或不选)的最大快乐值 
void dp(int pa)
{
	v[pa]=1;
	for(int i=1;i<=n;i++)
	{
		if(v[i]||fa[i]!=pa) continue;
		dp(i);
		f[pa][1]+=f[i][0];
		f[pa][0]+=max(f[i][0],f[i][1]);
	}
}
int main()
{
	n=read(); 
	for(int i=1;i<=n;i++) f[i][1]=read();
	for(int i=1;i<=n;i++)
	{
		x=read(); y=read();
		if(x==0&&y==0) break;
		fa[x]=y;
	}
	for(int i=1;i<=n;i++)
	{
		if(fa[i]!=0) continue;
		else root=i;
	}
	dp(root);
	int ans=max(f[root][1],f[root][0]);
	printf("%d",ans);
	return 0;
}
二号板子题

Strategic game [POJ-1463]
题意:
给一棵树放士兵,士兵可以管辖此点及此点的儿子,求最少的士兵放置数。
题解:
f [ i ] [ j ] f[i][j] f[i][j]:表示第i个节点的状态为j(有无人看管)

状态转移方程:
f [ f a ] [ 0 ] + = f [ i ] [ 1 ] f[fa][0]+=f[i][1] f[fa][0]+=f[i][1]//父亲有没人管,儿子要有人管
f [ f a ] [ 1 ] + = m i n ( f [ i ] [ 0 ] , f [ i ] [ 1 ] ) f[fa][1]+=min(f[i][0],f[i][1]) f[fa][1]+=min(f[i][0],f[i][1])//父亲有人管,儿子可管可不管

代码:
#include<stdio.h>
#include<iostream>
#include<string.h>
using namespace std;
const int sea=3030;
bool fa[sea];
int n,root,f[sea][2],brother[sea],son[sea];
void dp(int pa)
{
	f[pa][0]=0; f[pa][1]=1;
	int ff=son[pa];
	while(ff!=-1)
	{
		dp(ff);
		f[pa][1]+=min(f[ff][0],f[ff][1]);
		f[pa][0]+=f[ff][1];
		ff=brother[ff];
	}
}
int main()
{
//	freopen("a.txt","r",stdin);
	while(scanf("%d",&n)!=EOF)
	{
		memset(son,-1,sizeof(son));
		memset(fa,0,sizeof(fa));
		for(int i=0;i<n;i++)
		{
			int m,x,y;
			scanf("%d: (%d)",&y,&m);
			for(int i=0;i<m;i++)
			{
				scanf("%d",&x);
				brother[x]=son[y];//这个儿子的兄弟是y的上一个儿子 
				son[y]=x; //这个儿子也是y的儿子 
				fa[x]=1;//这个儿子有父亲 
			}
		}
		for(int i=0;i<n;i++) if(!fa[i]){root=i; break;}
		dp(root);
		int ans=min(f[root][1],f[root][0]);
		printf("%d\n",ans);
	}
	return 0;
}
三号板子题:

题目链接:Tree Cutting [POJ 2378]
题意:一棵树删点,使删过几个点之后的联通块总点数<=n / 2,求点
题解:
这个题其实跟DP没有什么关系,,,,
枚举每个点,判断如果它的子树大小 <= n/2&&n – 它的子树大小<=n/2
就可删。

代码:
#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;
inline int read()
{
	int s=0,w=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
	while(ch<='9'&&ch>='0')s=s*10+ch-'0',ch=getchar();
	return s*w;
}
const int sea=10100l;
struct see{int x,ver,next;}e[sea<<1];
int f[sea],n,v[sea],last[sea],tot,cnt=0;
void add(int x,int y){e[++tot].x=x;e[tot].ver=y;e[tot].next=last[x];last[x]=tot;}
void dfs(int x,int fa)
{
	bool flag=1;
	for(int i=last[x];i!=-1;i=e[i].next)
	{
		int y=e[i].ver; if(y==fa) continue;
		dfs(y,x);
		f[x]+=f[y];
		if(f[y]>n/2) flag=0;
	}
	if(n-f[x]>n/2) flag=0;
	if(flag) v[x]=1; 
}
int main()
{
	while(scanf("%d",&n)!=EOF)//记得是多组数据哦
	{
		memset(last,-1,sizeof(last));
		for(int i=1;i<=n;i++) f[i]=1;
		for(int i=1;i<n;i++)
		{
			int x,y;
			x=read(); y=read();
			add(x,y); add(y,x);
		}
		dfs(1,-1); int flag=0;
		for(int i=1;i<=n;i++) if(v[i]) printf("%d\n",i),flag=1;
		if(!flag) puts("NONE");
	}
	return 0;
}
四号板子题:

题目链接:Tree of Tree [ZOJ-3201]
题意:
求一棵树大小为k的子树最大的权值和。
题解:

f [ i ] [ j ] f[i][j] f[i][j]:表示的是第i个节点体积为j时的最大子树的总权重(有点类似于01背包)

状态转移方程:
f [ x ] [ j ] = m a x ( f [ x ] [ j ] , f [ x ] [ j − k ] + f [ y ] [ k ] ) f[x][j]=max(f[x][j],f[x][j-k]+f[y][k]) f[x][j]=max(f[x][j],f[x][jk]+f[y][k])

代码:
#include<bits/stdc++.h>
using namespace std;
inline int read()
{
	int s=0,w=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
	while(ch<='9'&&ch>='0')s=s*10+ch-'0',ch=getchar();
	return s*w;
}
const int sea=110;
struct see{int x,ver,next;}e[sea<<1];
int n,m,ans,a[sea],last[sea],tot;
int f[sea][sea];//f[i][j]表示的是第i个节点体积为j时的最大子树的总权重,类似于01背包 
void add(int x,int y){e[++tot].x=x;e[tot].ver=y;e[tot].next=last[x];last[x]=tot;}
void dfs(int x,int fa)
{
	f[x][1]=a[x];
	for(int i=last[x];i!=-1;i=e[i].next)
	{
		int y=e[i].ver;if(y==fa) continue;
		dfs(y,x);
		for(int j=m;j>0;j--) for(int k=0;k<j;k++)
		f[x][j]=max(f[x][j],f[x][j-k]+f[y][k]);
	}
	ans=max(ans,f[x][m]);
	return ;
}
int main()
{
	int x,y;
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		memset(f,0,sizeof(f));
		memset(last,-1,sizeof(last));
		for(int i=0;i<n;i++) a[i]=read();	
		for(int i=1;i<n;i++)
		{
			x=read(); y=read(); 
			add(x,y); add(y,x);
		}
		ans=0; dfs(0,-1);
		printf("%d\n",ans);
	}
	return 0;
}
五号板子题:

Cell Phone Network [POJ 3659]
题意:求一棵树的最小支配集
题解:
f [ i ] [ 2 ] f[i][2] f[i][2]在此节点的子节点建塔
f [ i ] [ 0 ] f[i][0] f[i][0]在此节点的父亲点建塔
f [ i ] [ 1 ] f[i][1] f[i][1]在此点自身建塔
状态转移方程:
f [ i ] [ 2 ] = m i n ( f [ i ] [ 0 ] , f [ i ] [ 1 ] , f [ i ] [ 2 ] ) f[i][2]=min(f[i][0],f[i][1],f[i][2]) f[i][2]=min(f[i][0],f[i][1],f[i][2])
f [ i ] [ 1 ] = m i n ( f [ i ] [ 0 ] , f [ i ] [ 1 ] ) f[i][1]=min(f[i][0],f[i][1]) f[i][1]=min(f[i][0],f[i][1])
f [ i ] [ 0 ] = m i n ( f [ i ] [ 0 ] , f [ i ] [ 1 ] ) f[i][0]=min(f[i][0],f[i][1]) f[i][0]=min(f[i][0],f[i][1])
对于f[i][1]的情况还是需要特别处理一下的,对于遇到像下面这样的点(当我们在g位置放过了时,就会发现s1或s2点不能被覆盖)时,我们就特别取一下,(就如下面代码)
在这里插入图片描述

	......
	if(f[y][2]<f[y][1])
	f[x][2]+=f[y][2],ccc=min(ccc,f[y][1]-f[y][2]);//直接计算差值,就是特例的个数
	else f[x][2]+=f[y][1],flag=1;//打个标记记录一下这个
	......
	if(!flag) f[x][2]+=ccc;//判断标记,加上差值
代码:
#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;
inline int read()
{
	int s=0,w=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
	while(ch<='9'&&ch>='0')s=s*10+ch-'0',ch=getchar();
	return s*w;
}
const int sea=10100l;
const int inf=999999;
struct see{int x,ver,next;}e[sea<<1];
int f[sea][3],n,last[sea],tot,ans;
void add(int x,int y){e[++tot].x=x;e[tot].ver=y;e[tot].next=last[x];last[x]=tot;}
void dfs(int x,int fa)
{
	int ccc=inf;
	bool flag=0; 
	for(int i=last[x];i;i=e[i].next)
	{
		int y=e[i].ver;if(y==fa) continue;
		dfs(y,x);
		f[x][1]+=min(f[y][1],min(f[y][0],f[y][2]));
		f[x][0]+=min(f[y][1],f[y][2]);
		if(f[y][2]<f[y][1])
		f[x][2]+=f[y][2],ccc=min(ccc,f[y][1]-f[y][2]);
		else f[x][2]+=f[y][1],flag=1;
	}
	if(!flag) f[x][2]+=ccc;
}
int main()
{
	while(scanf("%d",&n)!=EOF)
	{
		for(int i=1;i<=n;i++) f[i][1]=1,f[1][0]=0,f[2][0];
		for(int i=1;i<n;i++)
		{
			int x=read(),y=read(); 
			add(x,y); add(y,x);
		}
		dfs(1,0);
		printf("%d\n",min(f[1][1],f[1][2]));
	}	
	return 0;
}
六号板子题:

题目链接:Computer [HDU-2196]
题意:求树上的每个点到达整棵树对于这个顶点的最远距离。
题解:
(这个题啊,告诉我们不要在不常用的OJ上交题,否则就光CE就能CE好长时间,,o(╥﹏╥)o)
(正经的题解:)
对于一个节点,要想找他的最远距离,就只能从两个方向,一个是从他的父亲节点的方向,第二个就是从他的儿子节点的方向找,但是,但是,当你在向他的父亲方向找距离的时候,又时候会找到一条路径,这条路径是最大的,而且这条路径包括了此点,这样子就多找了一个点,所以,就可以在维护最大值得时候,顺便再维护一个次大值,这样的话就解决了这类问题,最重要的部分就是这样了,其他的就在注释里吧。

代码:
#include<iostream>
#include<cstdio>
#include<string.h> 
#include<algorithm>
#define inf 1000000000
using namespace std;
inline int read()
{
	int s=0,w=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
	while(ch<='9'&&ch>='0')s=s*10+ch-'0',ch=getchar();
	return s*w;
}
const int sea=10005;
struct see{int ver,next,edge;}e[20005];
int f[sea],g[sea],n,last[20005],tot,ans[sea];
//f[i]存的是第i个点父亲节点的最大距离 
//g[i]存的是第i个点儿子节点的最大距离 
void add(int x,int y,int z){e[++tot].edge=z;e[tot].ver=y;e[tot].next=last[x];last[x]=tot;}
void dp(int x,int fa)
{
	for(int i=last[x];i;i=e[i].next)
	{
		int y=e[i].ver,z=e[i].edge;	
		if(y!=fa) dp(y,x),f[x]=max(f[x],f[y]+z); 
	}
}
void so(int x,int fa)
{
	int p=0,q=0,p1=0,q1=0;
	for(int i=last[x];i;i=e[i].next)//最大值
	{
		int y=e[i].ver,z=e[i].edge;
		if(y!=fa&&f[y]+z>=f[p]+p1)	p=y,p1=z;
	}
	for(int i=last[x];i;i=e[i].next)//次大值
	{
		int y=e[i].ver,z=e[i].edge;
		if(y!=fa&&y!=p)
		{
			g[y]=ans[y]=f[p]+z+p1;
			if(f[y]+z>=f[q]+q1) q=y,q1=z;
		} 
	}
	g[p]=ans[p]=f[q]+p1+q1;
	for(int i=last[x];i;i=e[i].next)//更新子节点
	{
		int y=e[i].ver,z=e[i].edge;
		if(y!=fa) g[y]=max(g[y],g[x]+z);
	}
	for(int i=last[x];i;i=e[i].next)//更新答案
	{
		int y=e[i].ver,z=e[i].edge;
		if(y!=fa) ans[y]=max(ans[y],g[x]+z),so(y,x);
	}
}
int main()
{
	while(scanf("%d",&n)!=EOF)
	{
		tot=0;
		memset(last,0,sizeof(last));
		memset(f,0,sizeof(f));
		memset(g,0,sizeof(g));
		for(int i=2;i<=n;i++)
		{
			int x=read(),z=read(); 
			add(x,i,z); add(i,x,z); 		
		}
		f[0]=g[0]=-inf;
		dp(1,0);so(1,0);
		for(int i=1;i<=n;i++) printf("%d\n",max(f[i],ans[i]));
	}
	return 0;
} 

这样的话,树形DP入门的六道题就可以完结了,,,(但是树形DP这一章还是不可能就这样翻过去的,,(先发在说,草稿箱里太多了))
好的来补道题吧,,

七号板子题

一道刚考的题:
题目链接:树上距离
题解:
(这个题,我30行+的树形DP直接击溃100+行的点分治(但是不代表不能写点分治))
(一个写DP题的套路:先定义f[]再试状态转移方程)
f [ i ] [ j ] f[i][j] f[i][j]:以i作为长度为j的路径其中一个点的方案数
状态转移方程:
f [ x ] [ j ] + = f [ y ] [ j − 1 ] f[x][j]+=f[y][j-1] f[x][j]+=f[y][j1]
A n s + = d p [ y ] [ j ] ∗ d p [ x ] [ m − j − 1 ] ; Ans+=dp[y][j]*dp[x][m-j-1]; Ans+=dp[y][j]dp[x][mj1];
(这里的m就是题目中的k)

代码:
#include<bits/stdc++.h>
using namespace std;
inline int read()
{
	int s=0,w=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
	while(ch<='9'&&ch>='0')s=s*10+ch-'0',ch=getchar();
	return s*w;
}
const int sea=50005;
struct see{int x,ver,next;}e[100005];
int n,m,last[100005],tot,ans,f[sea][520];
void add(int x,int y){e[++tot].x=x;e[tot].ver=y;e[tot].next=last[x];last[x]=tot;}
void dfs(int x,int fa)
{
	f[x][0]=1;
	for(int i=last[x];i;i=e[i].next)
	{
		int y=e[i].ver; if(y==fa) continue;
		dfs(y,x);
		for(int j=0;j<m;j++) ans+=f[y][j]*f[x][m-j-1]; 
		for(int j=1;j<=m;j++) f[x][j]+=f[y][j-1];
	}
}
int main()
{
	n=read(); m=read();
	for(int i=1;i<n;i++)
	{
		int x,y;
		x=read(); y=read(); 
		add(x,y); add(y,x);	
	}
	dfs(1,0);
	printf("%d\n",ans);	
	return 0;
} 

Continue……

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值