AcWing 898. 线性dp：数字三角形    原题链接    简单

作者： 作者的头像   不二_3 ,  2021-04-01 16:55:20 ,  所有人可见 ,  阅读 401

1


正序dp 要考虑好边界 0 和 i+1 的位置也要初始化为负无穷
倒序dp 不用考虑边界问题，代码简单一些

C++ 代码

#include<iostream>
#include<cstring>
#include<algorithm>

using namespace std;

const int N=505; 
const int INF=1e9;

int n;
int w[N][N],f[N][N];

int main()
{
    cin>>n;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=i;j++)
            cin>>w[i][j];
    }
    //初始化边界
    for(int i=0;i<=n;i++){
        for(int j=0;j<=i+1;j++)
            f[i][j]=-INF;
    }
    //状态转移
    f[1][1]=w[1][1];//初始化！！！
    for(int i=2;i<=n;i++)
        for(int j=1;j<=i;j++)
            f[i][j]=max(f[i-1][j-1],f[i-1][j])+w[i][j];

    int res=-INF;
    for(int j=1;j<=n;j++)res=max(res,f[n][j]);

    cout<<res<<"\n";

    return 0;
}

1 评论

用户头像
itdef   2021-04-01 17:10         踩      回复

使用原数组 就没有初始化 也不用考虑边界了

// 11111.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
//

#include <iostream>
#include <algorithm>

using namespace std;


/*
输入样例：
5
7
3 8
8 1 0
2 7 4 4
4 5 2 6 5
输出样例：
30
*/

const int N = 510;
int arr[N][N];
int n;

int main()
{
    cin >> n;

    for (int i = 0; i < n; i++) {
        for (int j = 0; j <= i; j++) {
            cin >> arr[i][j];
        }
    }

    for (int i = n-2; i >=0; i--) {
        for (int j = 0; j <= i; j++) {
            arr[i][j] = max(arr[i][j] + arr[i + 1][j], arr[i][j] + arr[i+1][j + 1]);
        }
    }

    cout << arr[0][0] << endl;


    return 0;
}