绕来绕去,又绕回这道经典的题了。
从洛谷上找到了这道题:cgg最帅了!
点开它,传说中的数字三角形就出现在你的面前。
因为有题目我就不罗列了。
看到题目后有什么想法?
很难?很简单?
都不是!
你想到的应该是我们做动态规划最重要的三步中的两步(第一步好像我帮你做掉了)。
那就是设计状态和转态转移方程。
那么怎么设计状态呢?
还记得我们求数列最大值的时候怎么做的吗?
当前的部分的最大值,等于两部分的最大值的最大值。
那么我们这里也可以借鉴这个思路去解决这个问题。
我们就可以设计这样的状态:
f[i][j]表示由点(i,j)向下走能得到的最大值。
那么,我们的状态转移方程怎么写呢?其实很简单:
f[i][j]=max(f[i+1][j],f[i+1][j+1])+a[i][j];
a[][]表示原数字三角形。
这样代码也就出来了:
#include<bits/stdc++.h>
using namespace std;
int a[1005][1005];
int f[1005][1005];
int n;
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++){
for(int j=1;j<=i;j++){
scanf("%d",&a[i][j]);
}
}
for(int i=1;i<=n;i++){//这里有一个初始化
f[n][i]=a[n][i];
}
for(int i=n-1;i>=1;i--){
for(int j=1;j<=i;j++){
f[i][j]=max(f[i+1][j],f[i+1][j+1])+a[i][j];
}
}
printf("%d",f[1][1]);
return 0;
}
有的初学者可能比较蒙:为什么一切都显得那么自然?
实际上如果你到了我这个角度也会发现这个题目很简单。
但是我毕竟是要将给初学者听的,所以我们来想一下怎么想,怎么思考。
首先二维是很显然的(虽然我们可以用一维来解,这个过会在讲),因为数字三角形就是个二维的,而数列是一维的。所以用二维去描述很显然。
那么我们就会想到用二维坐标去描述每一个点,那么我们记录什么东西呢?
从上面到这个点得到的最大值?还是由这个点往下走的最大值?
很明显后者更优,为什么?
因为前者的状态转移方程并不好写,考虑边界的问题。
那么我们就采取后者,然后就可以很自然地写出状态转移方程,然后就得到程序了。
这就是我们的思考过程。
从这个过程,我们可以看出选择状态是个动态的过程。
状态决定状态转移方程,而状态转移方程又反过来影响状态。
因为你绝不想写一个特别复杂的状态转移方程,甚至写不出方程。
所以需要不断调整状态。
那么我们再来讲讲优化,怎么做成一维。
我们来看一下这行代码:
f[i][j]=max(f[i+1][j],f[i+1][j+1])+a[i][j];
有什么发现?
我们每次涉及的只有当前行的下面一行的最优值,而之前的根本没有任何作用,所以我们完全能再计算后,用新的值去覆盖以前的值,这样只要一维就行了。
于是,状态转移方程就变成了:
f[j]=max(f[j],f[j+1])+a[i][j];
所以程序就变成了:
#include<bits/stdc++.h>
using namespace std;
int a[1005][1005];
int f[1005];
int n;
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++){
for(int j=1;j<=i;j++){
scanf("%d",&a[i][j]);
}
}
for(int i=1;i<=n;i++){
f[i]=a[n][i];
}
for(int i=n-1;i>=1;i--){
for(int j=1;j<=i;j++){
f[j]=max(f[j],f[j+1])+a[i][j];
}
}
printf("%d",f[1]);
return 0;
}
当然这个数组有个美妙的名字叫做滚动数组。
这就是我今天要讲的。