数字三角形—方格取数

题目链接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;
}