003-4算法笔记【动态规划】习题3-15-样本收集问题

题目

机器人 Rob 个有 nxn 个方格的方形区域 中收集样本。 (i,j)方格中样本的价值为 v(i, j), 如图所示 Rob 从方形区域 的左上角A点出发，向下或向右行走，直到右下角B点，在走过的路上，收集方格中的样本，Rob从A点到B点共走2次。 试找出 2条行走路径，使其取得的样本总价值最大。

输入

第 1 行有 1 个正整数 n，表示方形区域 F 有 n*n 个方格。 接下来每行有 3 个整数，前 2 个表示方格位置，第 3 个数为该位置样本价值。最后一行是 3 个 0。

8
2 3 13
2 6 6
3 5 7
4 4 14
5 2 21
5 6 4
6 3 15
7 2 14
0 0 0

输出

最大值

67

分析

给定一个由n*n个方格组成的地图,每个方格都有一个价值,每一步只能朝右或者朝下走.从左上角到右下角走两次,一个方格内的价值只能获取一次,求可获取的最大总价值.

思路：走一遍，记下路径，然后沿着路径把走过的格子清空，再走第二遍，输出两次价值的和

递推式：

#define up 1//向上
#define left 2//向左
dp[i][j]=max(dp[i-1][j],dp[i][j-1])+v[i][j];
path[i][j]=dp[i-1][j]>dp[i][j-1]?up:left;
v[i][j]=0;

i=1,	dp[1][j]=dp[i][j-1]+v[i][j]
		path[1][j]=left;
j=1,	dp[i][1]=dp[i-1][1]+v[i][1]
		path[i][1]=up;

代码

/*
给定一个由n*n个方格组成的地图,每个方格都有一个价值,每一步只能朝右或者朝下走,
从左上角到右下角走两次,一个方格内的价值只能获取一次,求可获取的最大总价值.
*/
#include <iostream>
#include <cstring>
using namespace std;

#define UP 1
#define LEFT 2

#define N 50
int dp[N][N];//到达(i,j)时的最大值
int path[N][N];//前一个格子在哪，i-1行(UP)还是j-1列（LEFT）
int v[N][N];

//打印路径，并清空走过的格子
void trace(int i,int j,int n){
    if(i<1||j<1)return;
    if(i==1&&j==1){
        cout<<"A -> ";
        return;
    }
    
    if(path[i][j]==UP)
        trace(i-1,j,n);
    else if(path[i][j]==LEFT)
        trace(i,j-1,n);
    else
        return;
    
    if(i==n&&j==n)
        cout<<"B"<<endl<<endl;
    else
        cout<<"("<<i<<","<<j<<") -> ";
    // else if(v[i][j]!=0)
    //     cout<<"("<<i<<","<<j<<") -> ";

    v[i][j]=0;//清空走过的格子
}
int fun(int n)
{
    memset(dp, 0, sizeof(dp));
    memset(path, 0, sizeof(path));
    //初始化第一列
    for (int i = 1; i <= n; i++)
    {
        dp[i][1] = dp[i - 1][1] + v[i][1];
        path[i][1] = UP;
    }
    //初始化第一行
    for (int j = 1; j <= n; j++)
    {
        dp[1][j] = dp[1][j - 1] + v[1][j];
        path[1][j] = LEFT;
    }

    for (int i = 2; i <= n; i++)
    {
        for (int j = 2; j <= n; j++)
        {
            dp[i][j] = max(dp[i - 1][j], dp[i][j - 1]) + v[i][j];
            path[i][j] = dp[i - 1][j] > dp[i][j - 1] ? UP : LEFT;
        }
    }
    int res=dp[n][n];
    trace(n,n,n);//清除收集过的格子

    return res;
}
int main()
{
    int n;
    cin >> n;
    int x, y;
    while (cin >> x >> y >> v[x][y])
    {
        if (x == 0)
            break;
    }
    int res1=fun(n);
    int res2=fun(n);
    cout<<res1+res2<<endl;

    return 0;
}

评价

走一遍O(n²)，清空O(n)，总共O(n²)

其他解法

https://blog.csdn.net/m0_37809890/article/details/79865018?utm_medium=distribute.pc_relevant.none-task-blog-2_defaultbaidujs_title~default-0.pc_relevant_default&spm=1001.2101.3001.4242.1&utm_relevant_index=3

https://blog.csdn.net/weixin_43958091/article/details/102332184?spm=1001.2101.3001.6650.3&utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromBaidu%7Edefault-3.pc_relevant_default&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromBaidu%7Edefault-3.pc_relevant_default&utm_relevant_index=6