小明有一张藏宝图,上面有m*n个房间,每个房间里面都有一个有一定价值的宝物,小明只能从左上角的房间进入收集宝物,且每次只能向右边或向下边的房间继续寻宝,最终只能从最右下的房间出来。请你帮小明计算下他最多可以收集到多少价值的宝物?
输入格式:
输入第一行给出两个正整数m,n(1=<m,n<=2000),随后给出m行数据,每行都包括n个正整数,中间用空格分割。
输出格式:
输出收集到的最大价值v,题目保证v<10^9。
输入样例:
4 4
1 18 9 3
7 10 6 12
5 13 4 15
2 11 8 16
输出样例:
78
普通DP二维代码,内存超限
#include<iostream>
#include<vector>
using namespace std;
int main(){
int m,n;
cin>>m>>n;
vector < vector <int> >dp(m+1);//dp[i][j]表示在第i行j列找到的宝物最大价值
vector < vector <int> >arr(m+1);
for(int i=0;i<=m;i++){
dp[i].resize(n+1,0);
arr[i].resize(n+1,0);
}
for(int i=1;i<=m;i++){
for(int j=1;j<=n;j++){
cin>>arr[i][j];
}
}
//状态转移方程dp[i][j]=arr[i][j]+max(dp[i-1][j],dp[i][j-1])
//初始化“从上至下边界”
dp[1][1] = arr[1][1];
for(int i=2;i<=m;i++){
dp[i][1]=arr[i][1]+dp[i-1][1];
}
for(int j=2;j<=n;j++){
dp[1][j]=arr[1][j]+dp[1][j-1];
}
//递推
for(int i=2;i<=m;i++){
for(int j=2;j<=n;j++){
dp[i][j]=arr[i][j]+max(dp[i-1][j],dp[i][j-1]);
}
}
cout<<dp[m][n]<<endl;
return 0;
}
滚动数组DP
#include<iostream>
#include<vector>
using namespace std;
int main(){
int m,n;
cin>>m>>n;
vector <int> dp; //由于二维dp内存超了,采用滚动数组
vector < vector <int> >arr(m+1);
dp.resize(n+1);
for(int i=0;i<=m;i++){
arr[i].resize(n+1,0);
}
for(int i=1;i<=m;i++){
for(int j=1;j<=n;j++){
cin>>arr[i][j];
}
}
//状态转移方程dp[i][j]=arr[i][j]+max(dp[i-1][j],dp[i][j-1])
//初始化“从上至下边界”
for(int j=1;j<=n;j++){
dp[j]=arr[1][j]+dp[j-1];
}
//递推
for(int i=2;i<=m;i++){
dp[1]+=arr[i][1];
for(int j=2;j<=n;j++){
dp[j]=arr[i][j]+max(dp[j],dp[j-1]);
}
}
cout<<dp[n]<<endl;
return 0;
}