【算法基础24】树形DP与记忆化搜索（没有上司的舞会、滑雪）

本文链接：https://blog.csdn.net/qq_52622503/article/details/127972071

一、没有上司的舞会（树形DP）

问题描述：

问题分析： 用f [ u , 0 ]表示从以u为根的子树中选择不含u的方案，f[ u , 1 ]表示从以u为根的子树中选择含u的方案。用 si 表示 u 的第 i 个子树，则 f[ u , 0 ] = $\sum_{}^{}$ max( f[ si , 0 ] , f[ si , 1 ])，

f[ u , 1] = $\sum_{}^{}$ f[ si , 0 ]。

代码：

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;

const int N=20;
int happy[N];
int h[N],ne[N],e[N],idx;//数组模拟链表
int f[N][2];
bool has_father[N];

void add(int a,int b){//添加从a到b的路径
	e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

void dfs(int u){
	f[u][1]=happy[u];//含u先加上u所有的快乐值
	
	for(int i=h[u];i!=-1;i=ne[i]){//遍历u的子树
		int j=e[i];
		dfs(j);//递归调用求子树的最大快乐值
		
		f[u][0]+=max(f[j][0],f[j][1]);//代入公式由子树推整体
		f[u][1]+=f[j][0];
	}
}

int main(){
	int n;
	cin>>n;
	
	for(int i=1;i<=n;i++) cin>>happy[i];//存储各点的快乐值
	
	memset(h,-1,sizeof h);
	for(int i=0;i<n-1;i++){//添加边，即上司关系
		int a,b;
		cin>>a>>b;
		has_father[a]=true;
		add(b,a);
	}
	
	int root=1;
	while(has_father[root]) root++;//找所有点的根节点
	
	dfs(root);
	
	printf("%d\n",max(f[root][0],f[root][1]));
	
	return 0;
}

输出示例：

二、滑雪（记忆化搜索）

题目描述：

题目分析： 用f[ i , j ]表示从点( i , j )开始滑雪的最长轨迹，可以根据上、下、左、右分为四种情况。当其中一个方向的高度低于当前点时，当前点的最长轨迹就是该点的最长轨迹再加一，在四种情况种选出最大的一种情况作为f[ i , j ]的值。

代码：

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;

const int N=20;
int h[N][N],f[N][N];
int dx[4]={0,0,-1,1},dy[4]={1,-1,0,0};//上下左右方向向量
int n,m;

int dp(int x,int y){
	int &v=f[x][y];
	if(v!=-1)	return v;//减少不必要的计算
	
	v=1;//最差情况为轨迹只有当前点
	for(int i=0;i<4;i++){
		int a=x+dx[i],b=y+dy[i];
		if(1<=a&&a<=n&&1<=b&&b<=m&&h[a][b]<h[x][y])//该方向上的点符合条件
			v=max(v,dp(a,b)+1);		
	}
	return v;
}

int main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
			cin>>h[i][j];
			
	memset(f,-1,sizeof f);
	
	int res=0;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
			res=max(res,dp(i,j));
			
	cout<<res<<endl;
	
	return 0;
}

输出示例：