题目
1015. 摘花生 - AcWing题库
解释
- 第一种算法空间复杂度(N*N)
- w[i][j]存放花生的数量
- f[i][j]存放当前路径下最优的状态,即从(1,1)走到(i,j)的最优路径下的花生数量
- 第二种算法空间复杂度(N)
- 考虑到使用二维数组时,只是使用相邻两行之间的关系进行运算,因此我们可以简化为一维数组来减小空间复杂度
- f[j]相当于从上面过来,f[j-1]相当于从左边过来
代码段
算法一
#include<iostream>
#include<cstring>
using namespace std;
int t,n,m;
const int N=110;
int w[N][N];
int f[N][N];
int main()
{
cin>>t;
while(t--)
{
memset(w,0,sizeof w);
memset(f,0,sizeof f);
cin>>n>>m;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
cin>>w[i][j];
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
f[i][j]=max(f[i][j-1],f[i-1][j])+w[i][j];
}
cout<<f[n][m]<<endl;
}
}
算法二
#include<iostream>
#include<cstring>
using namespace std;
int t,n,m;
const int N=110;
int w[N][N];
int f[N];
int main()
{
cin>>t;
while(t--)
{
memset(w,0,sizeof w);
memset(f,0,sizeof f);
cin>>n>>m;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
cin>>w[i][j];
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
f[j]=max(f[j],f[j-1])+w[i][j];
}
cout<<f[m]<<endl;
}
}