奇数塔问题

最新推荐文章于 2024-03-30 13:41:11 发布

欧耶2023

最新推荐文章于 2024-03-30 13:41:11 发布

阅读量766

点赞数 24

文章标签：动态规划算法

本文链接：https://blog.csdn.net/zc2000911/article/details/135209005

版权

文章讨论了如何使用动态规划方法解决一个数塔问题，其中目标是从顶层到底层找到一条奇数长度的最大路径。通过定义状态转移方程，计算从每个节点出发的奇偶长度路径的最大值。

摘要由CSDN通过智能技术生成

问题描述

有如下所示的数塔，要求从顶层走到底层，若每一步只能走到相邻的结点，最终答案必须在最后一行，且为奇数，问答案最大是多少？

分析

考虑用 dp 来做。

状态：

$dp_{i,j,0}$ 为走到点 $i, j$ 的偶数长度最大路径， $dp_{i,j,1}$ 为走到点 $i, j$ 的奇数长度最大路径。

状态转移方程：

如果 $a_{i,j}$ 是偶数： $dp_{i,j,1} = \max(dp_{i+1,j,1},dp_{i+1,j+1,1})+a_{i,j}$ ， $dp_{i,j,0} = \max(dp_{i+1,j,0},dp_{i+1,j+1,0})+a_{i,j}$ 。
如果 $a_{i,j}$ 是奇数： $dp_{i,j,1} = \max(dp_{i+1,j,0},dp_{i+1,j+1,0})+a_{i,j}$ ， $dp_{i,j,0} = \max(dp_{i+1,j,1},dp_{i+1,j+1,1})+a_{i,j}$ 。

初始状态

$dp_{i,j,0}=dp_{i,j,1}=\infty$ 。

如果 $a_{n,i}$ 是偶数 $dp_{n,i,0} = a_{n,i},dp_{n,i,1}=0$ 。
如果 $a_{n,i}$ 是奇数 $dp_{n,i,1} = a_{n,i},dp_{n,i,0}=0$ 。

答案输出 $dp_{1,1,1}$ 即可。

代码

#include <bits/stdc++.h>

using namespace std;

const int N = 105;
int n, a[N][N], dp[N][N][2];

int main(){
	cin >> n;
	for(int i = 1; i <= n; i ++){
		for(int j = 1; j <= i; j ++){
			cin >> a[i][j];
			dp[i][j][0] = dp[i][j][1] = -1e9;
		}
	}
	for(int i = 1; i <= n; i ++){
		if(a[n][i] % 2 == 0){
			dp[n][i][0] = a[n][i];
			dp[n][i][1] = 0;
		}else{
			dp[n][i][1] = a[n][i];
			dp[n][i][0] = 0;
		}
	}
	for(int i = n - 1; i >= 1; i --){
		for(int j = 1; j <= i; j ++){
			if(a[i][j] % 2 == 0){//偶数 
				dp[i][j][1] = max(dp[i + 1][j][1], dp[i + 1][j + 1][1]) + a[i][j];
				dp[i][j][0] = max(dp[i + 1][j][0], dp[i + 1][j + 1][0]) + a[i][j];
			}else{//奇数 
				dp[i][j][1] = max(dp[i + 1][j][0], dp[i + 1][j + 1][0]) + a[i][j];
				dp[i][j][0] = max(dp[i + 1][j][1], dp[i + 1][j + 1][1]) + a[i][j];
			}
		}
	}
	cout << dp[1][1][1];
	return 0;
}