动态规划-数塔问题

图片来自百度搜索
图片来自百度搜索

问题描述:
从数塔的最高层出发到达最底层存在多种路径,求出经过的数字之和最大的路径。

解释:

如本图所示,最高点13可以经过11,也可以经过8,二11又可以经过40或者7。枚举方法明显时间复杂度过大而行不通。
仔细观察,11点的最大路径会用到7的最大路径,而8也会用到7的最大路径,碰到这种重叠子问题,所以要用到动态规划。
观察可发现11点的最大路径=max(40的最大路径,7的最大路径)+11,7点的最大路径=max(14的最大路径,15的最大路径)+7。
用dp[i][j]表示第i行第j列元素的最大路径,f[i][j]表示第i行第j列元素的值。
所以dp[i][j] = max(dp[i+1][j],dp[i+1][j+1])+f[i][j]
边界处,即最底层的最大路径等于权值本身。

#include<bits/stdc++.h>
using namespace std;
#define maxn 105
int f[maxn][maxn];
int dp[maxn][maxn];
int main(){
	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		for(int j=1;j<=i;j++){//注意是i不是n,观察形状即可知
			scanf("%d",&f[i][j]);
		}
	}
	for(int j=1;j<=n;j++){
		dp[n][j] = f[n][j];
	}
	for(int i=n-1;i>=1;i--){
		for(int j=1;j<=i;j++){//<=号,不是<号,在i+1层,有i+1列,可以取到dp[i+1][j+1]
			dp[i][j] = max(dp[i+1][j],dp[i+1][j+1])+f[i][j];
		}
	}
	printf("%d\n",dp[1][1]);
}

猜你喜欢

转载自blog.csdn.net/weixin_37762592/article/details/88600778
今日推荐