题目描述
在一个贴有 n*m 瓷砖的房间内,一些瓷砖上有垃圾要拾捡。现在有一个能捡垃圾的机器人从左上瓷砖处出发,每次只能向右或者向下走。每次他完全到达一块瓷砖点上,就会自动把这个瓷砖上的垃圾拾掉。问:最多能拾多少垃圾。
算法描述:
如果这个题是说机器人随机移动,那么就很难求解了。但是机器人的移动受到了限制,只能向下和向右移动。这就和动态规划的dp数组复制方向恰好一直。
由此联想到使用dp动态规划来做。某一个点(i,j)只能通过机器人从上面,或者左面走来。因此假设机器人走到(i-1,j)也就是左边的时候,捡到的垃圾数量是dp[i-1][j]的话,显然走到(i,j)时捡到的垃圾的数量 dp[i][j] 就等于 d[i-1][j] + matrix[i][j] 。
很容易可以想到状态转移方程应该是
输入描述
多组输入,每组第一行两个整数 n 和 m,下面 n 行 m 列表示瓷砖上的垃圾分布情况 (1<=n,m<=200)
输出描述
对于每组输入,输出一个整数,表示所能捡的垃圾最大数量。
样例输入
6 7
0 1 0 1 0 0 0
0 0 0 1 0 1 0
0 0 0 0 0 0 0
0 0 0 1 0 0 1
0 0 0 0 0 0 0
0 0 0 0 0 1 0
注:1 表示有垃圾,0 表示没有垃圾。
样例输出
5
代码如下:
#include<bits/stdc++.h>
using namespace std;
/*
6 7
0 1 0 1 0 0 0
0 0 0 1 0 1 0
0 0 0 0 0 0 0
0 0 0 1 0 0 1
0 0 0 0 0 0 0
0 0 0 0 0 1 0
*/
int row;
int column;
int main()
{
while (cin >> row >> column)
{
vector<vector<int> > matrix(row, vector<int>(column, 0));
for (int i = 0; i < row; i++)
{
for (int j = 0; j < column; j++)
{
cin >> matrix[i][j];
}
}
vector<vector<int> > dp(row, vector<int>(column, 0));
dp[0][0] = matrix[0][0];
//先对第一行 和第一列进行特殊初始化。
for (int i = 0; i < row-1; i++)
{
dp[i + 1][0] = dp[i][0] + matrix[i + 1][0];
}
for (int j = 0; j < column-1; j++)
{
dp[0][j + 1] = dp[0][j] + matrix[0][j + 1];
}
for (int i = 1; i < row; i++)
{
for (int j = 1; j < column; j++)
{
dp[i][j] = max(dp[i - 1][j], dp[i][j - 1]) + matrix[i][j];
}
}
cout << dp[row - 1][column - 1] << endl;
}
return 0;
}