java---线性dp---数字三角形（每日一道算法2022.10.11）

最新推荐文章于 2023-03-20 20:18:19 发布

SRestia

最新推荐文章于 2023-03-20 20:18:19 发布

阅读量209

点赞数 1

分类专栏：算法文章标签： java 算法动态规划

本文链接：https://blog.csdn.net/SRestia/article/details/127272748

版权

算法专栏收录该内容

157 篇文章 5 订阅

订阅专栏

堪称经典的一道题，其在线性dp中的地位就如同hello world一般~

题目：
给定一个如下图所示的数字三角形，从顶部出发，在每一结点可以选择移动至其左下方的结点或移动至其右下方的结点，一直走到底层，要求找出一条路径，使路径上的数字的和最大

        7
      3   8
    8   1   0
  2   7   4   4
4   5   2   6   5

第一行包含整数 n，表示数字三角形的层数
接下来 n 行，每行包含若干整数，其中第 i 行表示数字三角形第 i 层包含的整数

输出一个整数，表示最大的路径数字和

1 ≤ n ≤ 500
−10000 ≤ 三角形中的整数 ≤ 10000

输出：
30

public class 线性dp_数字三角形 {
	//a存储三角形中的数据，f存状态
    public static int N = 510, INF = -1000000009, n;
    public static int[][] a = new int[N][N], f = new int[N][N];

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        n = in.nextInt();

        //将三角形的数据导入
        for (int i = 1; i<=n; i++) {
            for (int j = 1; j<=i; j++) {
                a[i][j] = in.nextInt();
            }
        }

        base();
    }

    //基础线性dp写法
    public static void base() {
        //这里先将f中的数据预处理为负无穷
        for (int i = 1; i<=n; i++) {
            for (int j = 0; j<=i+1; j++) {  //切记两侧也要处理，因为三角内数据会有负数的情况，和边界外的0一比较就完蛋啦!
                f[i][j] = INF;
            }
        }

        //状态计算，注意从第二行开始计算状态，第一行f[1][1]直接赋值为初始值了
        f[1][1] = a[1][1];
        for (int i = 2; i<=n; i++) {
            for (int j = 1; j<=i; j++) {
                f[i][j] = Math.max(f[i-1][j-1], f[i-1][j]) + a[i][j];
            }
        }

        //找到此时所有最终状态中，值最大的那个，即为结果
        int res = INF;
        for (int i = 1; i<=n; i++) res = Math.max(res, f[n][i]);

        System.out.println(res);
    }

    //倒序dp优化，用这种方法甚至不需要初始化~因为压根就不会选到除了三角外的任何数据
    public static void reverse_op() {
        for (int i = n; i>=1; i--) {
            for (int j = i; j>=1; j--) {
                f[i][j] = Math.max(f[i+1][j+1] + a[i][j], f[i+1][j] + a[i][j]);
            }
        }

        System.out.println(f[1][1]);
    }
}