题目链接acwing1027. 方格取数
题目描述
原本的思路是分别找两次,把第一遍找到的数删除掉,后面发现这样不行,像上面的输入,就不能分别找两次。
这里考虑同时走两次
f[i1][j1][i2][j2]表示走到(i1,j1),(i2,j2)取得的数字最大和
其中i1+j1=i2+j2
我们可以考虑优化一下,令k=i1+j1=i2+j2
状态表示为f[k][i1][i2]
考虑状态转移f[k][i1][i2]=max(max(f[k-1][i1-1][i2],f[k-1][i1-1][i2-1]),max(f[k-1][i1][i2],f[k-1][i1][i2-1]))+w[k][i1][i2];
w[k][i1][i2]为预先处理出走到点(i1,k-i1),(i2,k-i2)这两个点的和,如果i1==i2只算一次
代码如下:
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=15;
int w[2*N][N][N];
int f[2*N][N][N];
int st[N][N];
int main(){
int n;
int x,y,z;
cin>>n;
while(scanf("%d%d%d",&x,&y,&z),x||y||z){
st[x][y]=z;
}
for(int k=2;k<=2*n;k++)
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
if(k-i>=1&&k-i<=n&&k-j>=1&&k-j<=n){
if(i==j) w[k][i][i]=st[i][k-i];
else w[k][i][j]=st[i][k-i]+st[j][k-j];
}
for(int k=2;k<=2*n;k++)
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
if(k-i>=1&&k-i<=n&&k-j>=1&&k-j<=n){
f[k][i][j]=max(max(f[k-1][i-1][j],f[k-1][i-1][j-1]),max(f[k-1][i][j],f[k-1][i][j-1]))+w[k][i][j];
}
cout<<f[2*n][n][n];
return 0;
}